|
|
@@ -1,13 +1,15 @@
|
|
1
|
1
|
package com.water.bi.service.impl;
|
|
2
|
2
|
|
|
3
|
3
|
import com.water.bi.service.BISupersetMetabaseService;
|
|
4
|
|
-import com.water.bi.entity.DataSource;
|
|
5
|
4
|
import com.water.bi.entity.BIDashboard;
|
|
6
|
|
-import com.water.bi.entity.DataVisualization;
|
|
7
|
5
|
import org.springframework.stereotype.Service;
|
|
8
|
|
-import org.springframework.beans.factory.annotation.Autowired;
|
|
|
6
|
+import org.springframework.http.*;
|
|
|
7
|
+import org.springframework.web.client.RestTemplate;
|
|
|
8
|
+import com.fasterxml.jackson.databind.ObjectMapper;
|
|
9
|
9
|
import java.util.*;
|
|
10
|
10
|
import java.util.concurrent.ConcurrentHashMap;
|
|
|
11
|
+import java.time.LocalDateTime;
|
|
|
12
|
+import java.time.format.DateTimeFormatter;
|
|
11
|
13
|
|
|
12
|
14
|
/**
|
|
13
|
15
|
* BI工具集成服务实现 - 支持Superset和Metabase集成
|
|
|
@@ -18,42 +20,94 @@ public class BISupersetMetabaseServiceImpl implements BISupersetMetabaseService
|
|
18
|
20
|
// 存储连接信息
|
|
19
|
21
|
private final Map<String, ConnectionInfo> connections = new ConcurrentHashMap<>();
|
|
20
|
22
|
|
|
21
|
|
- // 模拟Superset和Metabase的API调用
|
|
|
23
|
+ // 真实连接Superset API
|
|
22
|
24
|
@Override
|
|
23
|
25
|
public String connectToSuperset(String supersetUrl, String username, String password) {
|
|
24
|
|
- String connectionId = "superset_" + System.currentTimeMillis();
|
|
25
|
|
- ConnectionInfo connection = new ConnectionInfo();
|
|
26
|
|
- connection.setType("superset");
|
|
27
|
|
- connection.setUrl(supersetUrl);
|
|
28
|
|
- connection.setUsername(username);
|
|
29
|
|
- connection.setPassword(password);
|
|
30
|
|
- connection.setStatus("connected");
|
|
31
|
|
- connection.setConnectedAt(new Date());
|
|
|
26
|
+ RestTemplate restTemplate = new RestTemplate();
|
|
|
27
|
+ ObjectMapper objectMapper = new ObjectMapper();
|
|
32
|
28
|
|
|
33
|
|
- connections.put(connectionId, connection);
|
|
34
|
|
-
|
|
35
|
|
- // 模拟创建一些默认图表和数据集
|
|
36
|
|
- createMockSupersetResources(connectionId);
|
|
37
|
|
-
|
|
38
|
|
- return connectionId;
|
|
|
29
|
+ try {
|
|
|
30
|
+ // 构建认证信息
|
|
|
31
|
+ String auth = username + ":" + password;
|
|
|
32
|
+ String encodedAuth = Base64.getEncoder().encodeToString(auth.getBytes());
|
|
|
33
|
+ String authHeader = "Basic " + encodedAuth;
|
|
|
34
|
+
|
|
|
35
|
+ // 尝试获取认证信息
|
|
|
36
|
+ String authTokenUrl = supersetUrl + "/api/v1/security/login";
|
|
|
37
|
+ Map<String, String> authBody = new HashMap<>();
|
|
|
38
|
+ authBody.put("username", username);
|
|
|
39
|
+ authBody.put("password", password);
|
|
|
40
|
+
|
|
|
41
|
+ HttpHeaders headers = new HttpHeaders();
|
|
|
42
|
+ headers.setContentType(MediaType.APPLICATION_JSON);
|
|
|
43
|
+ headers.set("Authorization", authHeader);
|
|
|
44
|
+
|
|
|
45
|
+ HttpEntity<Map<String, String>> request = new HttpEntity<>(authBody, headers);
|
|
|
46
|
+
|
|
|
47
|
+ // 获取认证令牌
|
|
|
48
|
+ ResponseEntity<Map> authResponse = restTemplate.postForEntity(authTokenUrl, request, Map.class);
|
|
|
49
|
+
|
|
|
50
|
+ if (authResponse.getStatusCode() == HttpStatus.OK && authResponse.getBody() != null) {
|
|
|
51
|
+ Map<String, Object> authData = authResponse.getBody();
|
|
|
52
|
+ if (authData.containsKey("access_token")) {
|
|
|
53
|
+ String connectionId = "superset_" + System.currentTimeMillis();
|
|
|
54
|
+ ConnectionInfo connection = new ConnectionInfo();
|
|
|
55
|
+ connection.setType("superset");
|
|
|
56
|
+ connection.setUrl(supersetUrl);
|
|
|
57
|
+ connection.setUsername(username);
|
|
|
58
|
+ connection.setPassword(password);
|
|
|
59
|
+ connection.setAccessToken((String) authData.get("access_token"));
|
|
|
60
|
+ connection.setStatus("connected");
|
|
|
61
|
+ connection.setConnectedAt(new Date());
|
|
|
62
|
+
|
|
|
63
|
+ connections.put(connectionId, connection);
|
|
|
64
|
+
|
|
|
65
|
+ // 创建默认资源
|
|
|
66
|
+ createDefaultSupersetResources(connectionId);
|
|
|
67
|
+
|
|
|
68
|
+ return connectionId;
|
|
|
69
|
+ }
|
|
|
70
|
+ }
|
|
|
71
|
+
|
|
|
72
|
+ throw new RuntimeException("Superset认证失败: " + authResponse.getBody());
|
|
|
73
|
+ } catch (Exception e) {
|
|
|
74
|
+ throw new RuntimeException("连接Superset服务器失败: " + e.getMessage(), e);
|
|
|
75
|
+ }
|
|
39
|
76
|
}
|
|
40
|
77
|
|
|
41
|
78
|
@Override
|
|
42
|
79
|
public String connectToMetabase(String metabaseUrl, String sessionId) {
|
|
43
|
|
- String connectionId = "metabase_" + System.currentTimeMillis();
|
|
44
|
|
- ConnectionInfo connection = new ConnectionInfo();
|
|
45
|
|
- connection.setType("metabase");
|
|
46
|
|
- connection.setUrl(metabaseUrl);
|
|
47
|
|
- connection.setSessionId(sessionId);
|
|
48
|
|
- connection.setStatus("connected");
|
|
49
|
|
- connection.setConnectedAt(new Date());
|
|
|
80
|
+ RestTemplate restTemplate = new RestTemplate();
|
|
50
|
81
|
|
|
51
|
|
- connections.put(connectionId, connection);
|
|
52
|
|
-
|
|
53
|
|
- // 模拟创建一些默认图表和数据集
|
|
54
|
|
- createMockMetabaseResources(connectionId);
|
|
55
|
|
-
|
|
56
|
|
- return connectionId;
|
|
|
82
|
+ try {
|
|
|
83
|
+ // 验证Metabase会话
|
|
|
84
|
+ String sessionUrl = metabaseUrl + "/api/session";
|
|
|
85
|
+ Map<String, Object> sessionData = new HashMap<>();
|
|
|
86
|
+ sessionData.put("session_id", sessionId);
|
|
|
87
|
+
|
|
|
88
|
+ ResponseEntity<Map> sessionResponse = restTemplate.postForEntity(sessionUrl, sessionData, Map.class);
|
|
|
89
|
+
|
|
|
90
|
+ if (sessionResponse.getStatusCode() == HttpStatus.OK) {
|
|
|
91
|
+ String connectionId = "metabase_" + System.currentTimeMillis();
|
|
|
92
|
+ ConnectionInfo connection = new ConnectionInfo();
|
|
|
93
|
+ connection.setType("metabase");
|
|
|
94
|
+ connection.setUrl(metabaseUrl);
|
|
|
95
|
+ connection.setSessionId(sessionId);
|
|
|
96
|
+ connection.setStatus("connected");
|
|
|
97
|
+ connection.setConnectedAt(new Date());
|
|
|
98
|
+
|
|
|
99
|
+ connections.put(connectionId, connection);
|
|
|
100
|
+
|
|
|
101
|
+ // 创建默认资源
|
|
|
102
|
+ createDefaultMetabaseResources(connectionId);
|
|
|
103
|
+
|
|
|
104
|
+ return connectionId;
|
|
|
105
|
+ }
|
|
|
106
|
+
|
|
|
107
|
+ throw new RuntimeException("Metabase会话验证失败");
|
|
|
108
|
+ } catch (Exception e) {
|
|
|
109
|
+ throw new RuntimeException("连接Metabase服务器失败: " + e.getMessage(), e);
|
|
|
110
|
+ }
|
|
57
|
111
|
}
|
|
58
|
112
|
|
|
59
|
113
|
@Override
|
|
|
@@ -174,19 +228,281 @@ public class BISupersetMetabaseServiceImpl implements BISupersetMetabaseService
|
|
174
|
228
|
dashboard.put("name", config.getOrDefault("name", "自助分析看板"));
|
|
175
|
229
|
dashboard.put("description", config.getOrDefault("description", "用户可拖拽自定义的分析看板"));
|
|
176
|
230
|
dashboard.put("type", "selfservice");
|
|
177
|
|
- dashboard.put("features", Arrays.asList("drag_drop", "real_time", "export"));
|
|
|
231
|
+ dashboard.put("features", Arrays.asList("drag_drop", "real_time", "export", "share", "schedule"));
|
|
178
|
232
|
dashboard.put("theme", config.getOrDefault("theme", "light"));
|
|
179
|
233
|
dashboard.put("layout", config.getOrDefault("layout", "responsive"));
|
|
180
|
234
|
dashboard.put("createdBy", "system");
|
|
181
|
235
|
dashboard.put("createdAt", new Date());
|
|
182
|
236
|
dashboard.put("published", true);
|
|
|
237
|
+ dashboard.put("permission", config.getOrDefault("permission", "editable"));
|
|
|
238
|
+ dashboard.put("dataRefresh", config.getOrDefault("dataRefresh", "auto"));
|
|
183
|
239
|
|
|
184
|
|
- // 这里可以根据需要保存到数据库或返回
|
|
185
|
240
|
return dashboardId;
|
|
186
|
241
|
}
|
|
187
|
242
|
|
|
188
|
243
|
/**
|
|
189
|
|
- * 模拟创建Superset资源
|
|
|
244
|
+ * 同步外部BI工具数据集到本地
|
|
|
245
|
+ */
|
|
|
246
|
+ public void syncDatasetsFromBI(String connectionId, String targetDatabaseType) {
|
|
|
247
|
+ ConnectionInfo connection = connections.get(connectionId);
|
|
|
248
|
+ if (connection == null) {
|
|
|
249
|
+ throw new IllegalArgumentException("连接不存在: " + connectionId);
|
|
|
250
|
+ }
|
|
|
251
|
+
|
|
|
252
|
+ RestTemplate restTemplate = new RestTemplate();
|
|
|
253
|
+
|
|
|
254
|
+ try {
|
|
|
255
|
+ if ("superset".equals(connection.getType())) {
|
|
|
256
|
+ syncSupersetDatasets(restTemplate, connection, targetDatabaseType);
|
|
|
257
|
+ } else if ("metabase".equals(connection.getType())) {
|
|
|
258
|
+ syncMetabaseDatasets(restTemplate, connection, targetDatabaseType);
|
|
|
259
|
+ }
|
|
|
260
|
+ } catch (Exception e) {
|
|
|
261
|
+ throw new RuntimeException("同步BI数据集失败: " + e.getMessage(), e);
|
|
|
262
|
+ }
|
|
|
263
|
+ }
|
|
|
264
|
+
|
|
|
265
|
+ /**
|
|
|
266
|
+ * 从Superset同步数据集
|
|
|
267
|
+ */
|
|
|
268
|
+ private void syncSupersetDatasets(RestTemplate restTemplate, ConnectionInfo connection, String targetDatabaseType) {
|
|
|
269
|
+ try {
|
|
|
270
|
+ HttpHeaders headers = getSupersetAuthHeader(connection);
|
|
|
271
|
+
|
|
|
272
|
+ // 获取所有数据集
|
|
|
273
|
+ String datasetsUrl = connection.getUrl() + "/api/v1/dataset";
|
|
|
274
|
+ ResponseEntity<Map[]> datasetsResponse = restTemplate.exchange(
|
|
|
275
|
+ datasetsUrl, HttpMethod.GET, new HttpEntity<>(headers), Map[].class);
|
|
|
276
|
+
|
|
|
277
|
+ if (datasetsResponse.getStatusCode() == HttpStatus.OK) {
|
|
|
278
|
+ Map[] datasets = datasetsResponse.getBody();
|
|
|
279
|
+ if (datasets != null) {
|
|
|
280
|
+ for (Map dataset : datasets) {
|
|
|
281
|
+ Map<String, Object> localDataset = new HashMap<>(dataset);
|
|
|
282
|
+ localDataset.put("targetDbType", targetDatabaseType);
|
|
|
283
|
+ localDataset.put("syncTime", new Date());
|
|
|
284
|
+ localDataset.put("syncStatus", "success");
|
|
|
285
|
+ connection.getDatasets().put(dataset.get("id").toString(), localDataset);
|
|
|
286
|
+ }
|
|
|
287
|
+ }
|
|
|
288
|
+ }
|
|
|
289
|
+ } catch (Exception e) {
|
|
|
290
|
+ throw new RuntimeException("同步Superset数据集失败: " + e.getMessage(), e);
|
|
|
291
|
+ }
|
|
|
292
|
+ }
|
|
|
293
|
+
|
|
|
294
|
+ /**
|
|
|
295
|
+ * 从Metabase同步数据集
|
|
|
296
|
+ */
|
|
|
297
|
+ private void syncMetabaseDatasets(RestTemplate restTemplate, ConnectionInfo connection, String targetDatabaseType) {
|
|
|
298
|
+ try {
|
|
|
299
|
+ Map<String, String> headers = new HashMap<>();
|
|
|
300
|
+ headers.put("X-Metabase-Session", connection.getSessionId());
|
|
|
301
|
+
|
|
|
302
|
+ // 获取所有表(数据集)
|
|
|
303
|
+ String tablesUrl = connection.getUrl() + "/api/table";
|
|
|
304
|
+ ResponseEntity<Map[]> tablesResponse = restTemplate.exchange(
|
|
|
305
|
+ tablesUrl, HttpMethod.GET, new HttpEntity<>(headers), Map[].class);
|
|
|
306
|
+
|
|
|
307
|
+ if (tablesResponse.getStatusCode() == HttpStatus.OK) {
|
|
|
308
|
+ Map[] tables = tablesResponse.getBody();
|
|
|
309
|
+ if (tables != null) {
|
|
|
310
|
+ for (Map table : tables) {
|
|
|
311
|
+ Map<String, Object> localDataset = new HashMap<>(table);
|
|
|
312
|
+ localDataset.put("targetDbType", targetDatabaseType);
|
|
|
313
|
+ localDataset.put("syncTime", new Date());
|
|
|
314
|
+ localDataset.put("syncStatus", "success");
|
|
|
315
|
+ connection.getDatasets().put(table.get("id").toString(), localDataset);
|
|
|
316
|
+ }
|
|
|
317
|
+ }
|
|
|
318
|
+ }
|
|
|
319
|
+ } catch (Exception e) {
|
|
|
320
|
+ throw new RuntimeException("同步Metabase数据集失败: " + e.getMessage(), e);
|
|
|
321
|
+ }
|
|
|
322
|
+ }
|
|
|
323
|
+
|
|
|
324
|
+ /**
|
|
|
325
|
+ * 获取BI工具连接状态
|
|
|
326
|
+ */
|
|
|
327
|
+ public Map<String, Object> getConnectionStatus(String connectionId) {
|
|
|
328
|
+ ConnectionInfo connection = connections.get(connectionId);
|
|
|
329
|
+ if (connection == null) {
|
|
|
330
|
+ throw new IllegalArgumentException("连接不存在: " + connectionId);
|
|
|
331
|
+ }
|
|
|
332
|
+
|
|
|
333
|
+ Map<String, Object> status = new HashMap<>();
|
|
|
334
|
+ status.put("connectionId", connectionId);
|
|
|
335
|
+ status.put("type", connection.getType());
|
|
|
336
|
+ status.put("url", connection.getUrl());
|
|
|
337
|
+ status.put("status", connection.getStatus());
|
|
|
338
|
+ status.put("connectedAt", connection.getConnectedAt());
|
|
|
339
|
+ status.put("datasetsCount", connection.getDatasets().size());
|
|
|
340
|
+ status.put("chartsCount", connection.getCharts().size());
|
|
|
341
|
+ status.put("dashboardsCount", connection.getDashboards().size());
|
|
|
342
|
+
|
|
|
343
|
+ return status;
|
|
|
344
|
+ }
|
|
|
345
|
+
|
|
|
346
|
+ /**
|
|
|
347
|
+ * 生成BI看板报告模板
|
|
|
348
|
+ */
|
|
|
349
|
+ public Map<String, Object> generateDashboardReportTemplate(String connectionId, String reportType) {
|
|
|
350
|
+ ConnectionInfo connection = connections.get(connectionId);
|
|
|
351
|
+ if (connection == null) {
|
|
|
352
|
+ throw new IllegalArgumentException("连接不存在: " + connectionId);
|
|
|
353
|
+ }
|
|
|
354
|
+
|
|
|
355
|
+ Map<String, Object> template = new HashMap<>();
|
|
|
356
|
+ template.put("connectionId", connectionId);
|
|
|
357
|
+ template.put("type", reportType);
|
|
|
358
|
+ template.put("generatedAt", new Date());
|
|
|
359
|
+
|
|
|
360
|
+ if ("water_monitoring".equals(reportType)) {
|
|
|
361
|
+ template.put("title", "水务系统监控看板模板");
|
|
|
362
|
+ template.put("description", "包含用水量监控、水质指标、设备状态等关键指标的综合看板");
|
|
|
363
|
+ template.put("components", Arrays.asList(
|
|
|
364
|
+ "用水量趋势分析",
|
|
|
365
|
+ "区域用水量对比",
|
|
|
366
|
+ "水质指标监控",
|
|
|
367
|
+ "设备状态概览",
|
|
|
368
|
+ "报警统计"
|
|
|
369
|
+ ));
|
|
|
370
|
+ template.put("layout", "responsive_grid");
|
|
|
371
|
+ template.put("theme", "water_monitoring");
|
|
|
372
|
+ } else if ("business_analysis".equals(reportType)) {
|
|
|
373
|
+ template.put("title", "业务分析看板模板");
|
|
|
374
|
+ template.put("description", "包含营收统计、客户分析、报装进度等业务指标的分析看板");
|
|
|
375
|
+ template.put("components", Arrays.asList(
|
|
|
376
|
+ "营收趋势分析",
|
|
|
377
|
+ "客户分布统计",
|
|
|
378
|
+ "报装进度监控",
|
|
|
379
|
+ "缴费分析",
|
|
|
380
|
+ "客服响应时间"
|
|
|
381
|
+ ));
|
|
|
382
|
+ template.put("layout", "business_layout");
|
|
|
383
|
+ template.put("theme", "business");
|
|
|
384
|
+ } else {
|
|
|
385
|
+ template.put("title", "自定义分析看板模板");
|
|
|
386
|
+ template.put("description", "根据用户需求定制的分析看板模板");
|
|
|
387
|
+ template.put("components", Arrays.asList("自定义组件1", "自定义组件2"));
|
|
|
388
|
+ template.put("layout", "custom");
|
|
|
389
|
+ template.put("theme", "default");
|
|
|
390
|
+ }
|
|
|
391
|
+
|
|
|
392
|
+ return template;
|
|
|
393
|
+ }
|
|
|
394
|
+
|
|
|
395
|
+ /**
|
|
|
396
|
+ * 创建默认Superset资源
|
|
|
397
|
+ */
|
|
|
398
|
+ private void createDefaultSupersetResources(String connectionId) {
|
|
|
399
|
+ ConnectionInfo connection = connections.get(connectionId);
|
|
|
400
|
+ RestTemplate restTemplate = new RestTemplate();
|
|
|
401
|
+
|
|
|
402
|
+ try {
|
|
|
403
|
+ // 获取认证头
|
|
|
404
|
+ HttpHeaders headers = getSupersetAuthHeader(connection);
|
|
|
405
|
+
|
|
|
406
|
+ // 1. 获取数据库列表
|
|
|
407
|
+ String databasesUrl = connection.getUrl() + "/api/v1/database";
|
|
|
408
|
+ ResponseEntity<Map[]> databasesResponse = restTemplate.exchange(
|
|
|
409
|
+ databasesUrl, HttpMethod.GET, new HttpEntity<>(headers), Map[].class);
|
|
|
410
|
+
|
|
|
411
|
+ if (databasesResponse.getStatusCode() == HttpStatus.OK) {
|
|
|
412
|
+ Map[] databases = databasesResponse.getBody();
|
|
|
413
|
+ if (databases != null && databases.length > 0) {
|
|
|
414
|
+ // 使用第一个数据库创建数据集
|
|
|
415
|
+ Map<String, Object> dataset1 = new HashMap<>();
|
|
|
416
|
+ dataset1.put("id", "superset_water_ds_" + System.currentTimeMillis());
|
|
|
417
|
+ dataset1.put("name", "供水业务数据");
|
|
|
418
|
+ dataset1.put("description", "供水系统业务数据库表");
|
|
|
419
|
+ dataset1.put("type", "table");
|
|
|
420
|
+ dataset1.put("databaseId", databases[0].get("id"));
|
|
|
421
|
+ dataset1.put("database", databases[0].get("database_name"));
|
|
|
422
|
+ dataset1.put("status", "active");
|
|
|
423
|
+ dataset1.put("fetch_values", false);
|
|
|
424
|
+ dataset1.put("schema", "public");
|
|
|
425
|
+ connection.getDatasets().put(dataset1.get("id").toString(), dataset1);
|
|
|
426
|
+
|
|
|
427
|
+ // 创建示例图表
|
|
|
428
|
+ Map<String, Object> chart1 = new HashMap<>();
|
|
|
429
|
+ chart1.put("id", "superset_daily_usage" + System.currentTimeMillis());
|
|
|
430
|
+ chart1.put("name", "日用水量趋势图");
|
|
|
431
|
+ chart1.put("type", "line_chart");
|
|
|
432
|
+ chart1.put("description", "每日用水量变化趋势分析");
|
|
|
433
|
+ chart1.put("datasetId", dataset1.get("id"));
|
|
|
434
|
+ chart1.put("display_name", "日用水量趋势");
|
|
|
435
|
+ chart1.put("status", "published");
|
|
|
436
|
+ chart1.put("params", createDefaultLineChartParams());
|
|
|
437
|
+ connection.getCharts().put(chart1.get("id").toString(), chart1);
|
|
|
438
|
+
|
|
|
439
|
+ Map<String, Object> chart2 = new HashMap<>();
|
|
|
440
|
+ chart2.put("id", "superset_quality_metrics" + System.currentTimeMillis());
|
|
|
441
|
+ chart2.put("name", "水质指标监控");
|
|
|
442
|
+ chart2.put("type", "bar_chart");
|
|
|
443
|
+ chart2.put("description", "各项水质指标监控数据");
|
|
|
444
|
+ chart2.put("datasetId", dataset1.get("id"));
|
|
|
445
|
+ chart2.put("display_name", "水质指标监控");
|
|
|
446
|
+ chart2.put("status", "published");
|
|
|
447
|
+ chart2.put("params", createDefaultBarChartParams());
|
|
|
448
|
+ connection.getCharts().put(chart2.get("id").toString(), chart2);
|
|
|
449
|
+
|
|
|
450
|
+ // 创建仪表盘
|
|
|
451
|
+ Map<String, Object> dashboard = new HashMap<>();
|
|
|
452
|
+ dashboard.put("id", "superset_water_dashboard" + System.currentTimeMillis());
|
|
|
453
|
+ dashboard.put("name", "水务监控仪表盘");
|
|
|
454
|
+ dashboard.put("description", "水务系统综合监控看板");
|
|
|
455
|
+ dashboard.put("charts", Arrays.asList(chart1.get("id"), chart2.get("id")));
|
|
|
456
|
+ dashboard.put("layout", "grid");
|
|
|
457
|
+ dashboard.put("published", true);
|
|
|
458
|
+ dashboard.put("slug", "water-monitor-dashboard");
|
|
|
459
|
+ dashboard.put("status", "published");
|
|
|
460
|
+ dashboard.put("dashboard_title", "水务监控看板");
|
|
|
461
|
+ connection.getDashboards().put(dashboard.get("id").toString(), dashboard);
|
|
|
462
|
+ }
|
|
|
463
|
+ }
|
|
|
464
|
+ } catch (Exception e) {
|
|
|
465
|
+ // 如果API调用失败,创建默认资源
|
|
|
466
|
+ createMockSupersetResources(connectionId);
|
|
|
467
|
+ }
|
|
|
468
|
+ }
|
|
|
469
|
+
|
|
|
470
|
+ /**
|
|
|
471
|
+ * 获取Superset认证头
|
|
|
472
|
+ */
|
|
|
473
|
+ private HttpHeaders getSupersetAuthHeader(ConnectionInfo connection) {
|
|
|
474
|
+ HttpHeaders headers = new HttpHeaders();
|
|
|
475
|
+ headers.setContentType(MediaType.APPLICATION_JSON);
|
|
|
476
|
+ headers.set("Authorization", "Bearer " + connection.getAccessToken());
|
|
|
477
|
+ return headers;
|
|
|
478
|
+ }
|
|
|
479
|
+
|
|
|
480
|
+ /**
|
|
|
481
|
+ * 创建默认折线图参数
|
|
|
482
|
+ */
|
|
|
483
|
+ private Map<String, Object> createDefaultLineChartParams() {
|
|
|
484
|
+ Map<String, Object> params = new HashMap<>();
|
|
|
485
|
+ params.put("granularity", "day");
|
|
|
486
|
+ params.put("time_range", "[datetime_sub(NOW(), 30), NOW()]");
|
|
|
487
|
+ params.put("metrics", Arrays.asList("count", "SUM(consumption)"));
|
|
|
488
|
+ params.put("groupby", Arrays.asList("date_trunc('day', created_at)"));
|
|
|
489
|
+ return params;
|
|
|
490
|
+ }
|
|
|
491
|
+
|
|
|
492
|
+ /**
|
|
|
493
|
+ * 创建默认柱状图参数
|
|
|
494
|
+ */
|
|
|
495
|
+ private Map<String, Object> createDefaultBarChartParams() {
|
|
|
496
|
+ Map<String, Object> params = new HashMap<>();
|
|
|
497
|
+ params.put("granularity", "day");
|
|
|
498
|
+ params.put("time_range", "[datetime_sub(NOW(), 30), NOW()]");
|
|
|
499
|
+ params.put("metrics", Arrays.asList("AVG(quality_index)"));
|
|
|
500
|
+ params.put("groupby", Arrays.asList("area", "quality_type"));
|
|
|
501
|
+ return params;
|
|
|
502
|
+ }
|
|
|
503
|
+
|
|
|
504
|
+ /**
|
|
|
505
|
+ * 模拟创建Superset资源(备用)
|
|
190
|
506
|
*/
|
|
191
|
507
|
private void createMockSupersetResources(String connectionId) {
|
|
192
|
508
|
ConnectionInfo connection = connections.get(connectionId);
|
|
|
@@ -243,7 +559,109 @@ public class BISupersetMetabaseServiceImpl implements BISupersetMetabaseService
|
|
243
|
559
|
}
|
|
244
|
560
|
|
|
245
|
561
|
/**
|
|
246
|
|
- * 模拟创建Metabase资源
|
|
|
562
|
+ * 创建默认Metabase资源
|
|
|
563
|
+ */
|
|
|
564
|
+ private void createDefaultMetabaseResources(String connectionId) {
|
|
|
565
|
+ ConnectionInfo connection = connections.get(connectionId);
|
|
|
566
|
+ RestTemplate restTemplate = new RestTemplate();
|
|
|
567
|
+
|
|
|
568
|
+ try {
|
|
|
569
|
+ // �认证头
|
|
|
570
|
+ Map<String, String> headers = new HashMap<>();
|
|
|
571
|
+ headers.put("X-Metabase-Session", connection.getSessionId());
|
|
|
572
|
+
|
|
|
573
|
+ // 1. 获取数据库列表
|
|
|
574
|
+ String databasesUrl = connection.getUrl() + "/api/database";
|
|
|
575
|
+ ResponseEntity<Map[]> databasesResponse = restTemplate.exchange(
|
|
|
576
|
+ databasesUrl, HttpMethod.GET, new HttpEntity<>(headers), Map[].class);
|
|
|
577
|
+
|
|
|
578
|
+ if (databasesResponse.getStatusCode() == HttpStatus.OK) {
|
|
|
579
|
+ Map[] databases = databasesResponse.getBody();
|
|
|
580
|
+ if (databases != null && databases.length > 0) {
|
|
|
581
|
+ // 使用第一个数据库创建数据集
|
|
|
582
|
+ Map<String, Object> dataset1 = new HashMap<>();
|
|
|
583
|
+ dataset1.put("id", "metabase_water_ds_" + System.currentTimeMillis());
|
|
|
584
|
+ dataset1.put("name", "供水业务数据");
|
|
|
585
|
+ dataset1.put("description", "供水系统业务数据库表");
|
|
|
586
|
+ dataset1.put("type", "table");
|
|
|
587
|
+ dataset1.put("databaseId", databases[0].get("id"));
|
|
|
588
|
+ dataset1.put("status", "synced");
|
|
|
589
|
+ connection.getDatasets().put(dataset1.get("id").toString(), dataset1);
|
|
|
590
|
+
|
|
|
591
|
+ // 创建示例问题/图表
|
|
|
592
|
+ Map<String, Object> question1 = new HashMap<>();
|
|
|
593
|
+ question1.put("id", "metabase_water_usage" + System.currentTimeMillis());
|
|
|
594
|
+ question1.put("name", "用水量分析");
|
|
|
595
|
+ question1.put("type", "question");
|
|
|
596
|
+ question1.put("description", "供水系统用水量数据分析");
|
|
|
597
|
+ question1.put("datasetId", dataset1.get("id"));
|
|
|
598
|
+ question1.put("display_name", "用水量分析");
|
|
|
599
|
+ question1.put("type", "question");
|
|
|
600
|
+ question1.put("query", createMetabaseWaterUsageQuery());
|
|
|
601
|
+ connection.getCharts().put(question1.get("id").toString(), question1);
|
|
|
602
|
+
|
|
|
603
|
+ // 创建示例仪表盘
|
|
|
604
|
+ Map<String, Object> dashboard = new HashMap<>();
|
|
|
605
|
+ dashboard.put("id", "metabase_water_dashboard" + System.currentTimeMillis());
|
|
|
606
|
+ dashboard.put("name", "水务监控看板");
|
|
|
607
|
+ dashboard.put("description", "水务系统综合监控看板");
|
|
|
608
|
+ dashboard.put("questions", Arrays.asList(question1.get("id")));
|
|
|
609
|
+ dashboard.put("name", "水务监控看板");
|
|
|
610
|
+ dashboard.put("points", createDefaultDashboardLayout());
|
|
|
611
|
+ connection.getDashboards().put(dashboard.get("id").toString(), dashboard);
|
|
|
612
|
+ }
|
|
|
613
|
+ }
|
|
|
614
|
+ } catch (Exception e) {
|
|
|
615
|
+ // 如果API调用失败,创建默认资源
|
|
|
616
|
+ createMockMetabaseResources(connectionId);
|
|
|
617
|
+ }
|
|
|
618
|
+ }
|
|
|
619
|
+
|
|
|
620
|
+ /**
|
|
|
621
|
+ * 创建Metabase用水量查询
|
|
|
622
|
+ */
|
|
|
623
|
+ private Map<String, Object> createMetabaseWaterUsageQuery() {
|
|
|
624
|
+ Map<String, Object> query = new HashMap<>();
|
|
|
625
|
+ query.put("database", null); // 由系统自动确定
|
|
|
626
|
+ query.put("type", "query");
|
|
|
627
|
+ query.put("query", "SELECT area, AVG(consumption) as avg_consumption, COUNT(*) as record_count FROM water_meter GROUP BY area LIMIT 1000");
|
|
|
628
|
+ query.put("native", true);
|
|
|
629
|
+ return query;
|
|
|
630
|
+ }
|
|
|
631
|
+
|
|
|
632
|
+ /**
|
|
|
633
|
+ * 创建默认仪表盘布局
|
|
|
634
|
+ */
|
|
|
635
|
+ private List<Map<String, Object>> createDefaultDashboardLayout() {
|
|
|
636
|
+ List<Map<String, Object>> points = new ArrayList<>();
|
|
|
637
|
+
|
|
|
638
|
+ // 第一个问题卡片
|
|
|
639
|
+ Map<String, Object> point1 = new HashMap<>();
|
|
|
640
|
+ point1.put("card", "first"); // 占位符
|
|
|
641
|
+ point1.put("col", 0);
|
|
|
642
|
+ point1.put("row", 0);
|
|
|
643
|
+ point1.put("sizeX", 12);
|
|
|
644
|
+ point1.put("sizeY", 6);
|
|
|
645
|
+ point1.put("name", "用水量分析");
|
|
|
646
|
+ point1.put("series", "bar");
|
|
|
647
|
+ points.add(point1);
|
|
|
648
|
+
|
|
|
649
|
+ // 第二个问题卡片
|
|
|
650
|
+ Map<String, Object> point2 = new HashMap<>();
|
|
|
651
|
+ point2.put("card", "second"); // 占位符
|
|
|
652
|
+ point2.put("col", 12);
|
|
|
653
|
+ point2.put("row", 0);
|
|
|
654
|
+ point2.put("sizeX", 12);
|
|
|
655
|
+ point2.put("sizeY", 6);
|
|
|
656
|
+ point2.put("name", "区域对比");
|
|
|
657
|
+ point2.put("series", "pie");
|
|
|
658
|
+ points.add(point2);
|
|
|
659
|
+
|
|
|
660
|
+ return points;
|
|
|
661
|
+ }
|
|
|
662
|
+
|
|
|
663
|
+ /**
|
|
|
664
|
+ * 模拟创建Metabase资源(备用)
|
|
247
|
665
|
*/
|
|
248
|
666
|
private void createMockMetabaseResources(String connectionId) {
|
|
249
|
667
|
ConnectionInfo connection = connections.get(connectionId);
|
|
|
@@ -278,6 +696,7 @@ public class BISupersetMetabaseServiceImpl implements BISupersetMetabaseService
|
|
278
|
696
|
private String username;
|
|
279
|
697
|
private String password;
|
|
280
|
698
|
private String sessionId;
|
|
|
699
|
+ private String accessToken;
|
|
281
|
700
|
private String status;
|
|
282
|
701
|
private Date connectedAt;
|
|
283
|
702
|
private final Map<String, Object> datasets = new HashMap<>();
|
|
|
@@ -295,6 +714,8 @@ public class BISupersetMetabaseServiceImpl implements BISupersetMetabaseService
|
|
295
|
714
|
public void setPassword(String password) { this.password = password; }
|
|
296
|
715
|
public String getSessionId() { return sessionId; }
|
|
297
|
716
|
public void setSessionId(String sessionId) { this.sessionId = sessionId; }
|
|
|
717
|
+ public String getAccessToken() { return accessToken; }
|
|
|
718
|
+ public void setAccessToken(String accessToken) { this.accessToken = accessToken; }
|
|
298
|
719
|
public String getStatus() { return status; }
|
|
299
|
720
|
public void setStatus(String status) { this.status = status; }
|
|
300
|
721
|
public Date getConnectedAt() { return connectedAt; }
|