|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+package com.water.data_engine.service;
|
|
|
2
|
+
|
|
|
3
|
+import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
4
|
+import lombok.RequiredArgsConstructor;
|
|
|
5
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
6
|
+import org.springframework.jdbc.core.JdbcTemplate;
|
|
|
7
|
+import org.springframework.kafka.annotation.KafkaListener;
|
|
|
8
|
+import org.springframework.kafka.core.KafkaTemplate;
|
|
|
9
|
+import org.springframework.stereotype.Service;
|
|
|
10
|
+
|
|
|
11
|
+import java.time.Instant;
|
|
|
12
|
+import java.util.*;
|
|
|
13
|
+
|
|
|
14
|
+@Slf4j
|
|
|
15
|
+@Service
|
|
|
16
|
+@RequiredArgsConstructor
|
|
|
17
|
+public class DataCollectService {
|
|
|
18
|
+
|
|
|
19
|
+ private final KafkaTemplate<String, String> kafkaTemplate;
|
|
|
20
|
+ private final JdbcTemplate jdbcTemplate;
|
|
|
21
|
+ private final ObjectMapper mapper = new ObjectMapper();
|
|
|
22
|
+
|
|
|
23
|
+ /** 数据汇聚入口:接收各来源数据,统一写入 Kafka */
|
|
|
24
|
+ public void ingest(String sourceType, String sourceId, Map<String, Object> rawData) {
|
|
|
25
|
+ try {
|
|
|
26
|
+ Map<String, Object> envelope = new LinkedHashMap<>();
|
|
|
27
|
+ envelope.put("sourceType", sourceType); // iot/manual/api
|
|
|
28
|
+ envelope.put("sourceId", sourceId);
|
|
|
29
|
+ envelope.put("timestamp", Instant.now().toEpochMilli());
|
|
|
30
|
+ envelope.put("data", rawData);
|
|
|
31
|
+ String json = mapper.writeValueAsString(envelope);
|
|
|
32
|
+
|
|
|
33
|
+ // 根据来源路由到不同 topic
|
|
|
34
|
+ String topic = switch (sourceType) {
|
|
|
35
|
+ case "iot" -> "iot.raw.generic";
|
|
|
36
|
+ case "manual" -> "data.manual";
|
|
|
37
|
+ case "api" -> "data.api";
|
|
|
38
|
+ default -> "data.raw";
|
|
|
39
|
+ };
|
|
|
40
|
+ kafkaTemplate.send(topic, sourceId, json);
|
|
|
41
|
+ log.debug("Ingested: {} -> {}", sourceType, sourceId);
|
|
|
42
|
+ } catch (Exception e) {
|
|
|
43
|
+ log.error("Ingest error: {}", e.getMessage());
|
|
|
44
|
+ }
|
|
|
45
|
+ }
|
|
|
46
|
+
|
|
|
47
|
+ /** Kafka 实时流消费:写入 TDengine 时序库 */
|
|
|
48
|
+ @KafkaListener(topics = "iot.raw.generic", groupId = "wm-data-engine")
|
|
|
49
|
+ public void consumeIotRaw(String message) {
|
|
|
50
|
+ try {
|
|
|
51
|
+ @SuppressWarnings("unchecked")
|
|
|
52
|
+ Map<String, Object> envelope = mapper.readValue(message, Map.class);
|
|
|
53
|
+ @SuppressWarnings("unchecked")
|
|
|
54
|
+ Map<String, Object> data = (Map<String, Object>) envelope.get("data");
|
|
|
55
|
+
|
|
|
56
|
+ String deviceSn = (String) data.getOrDefault("deviceSn", "unknown");
|
|
|
57
|
+ @SuppressWarnings("unchecked")
|
|
|
58
|
+ List<Map<String, Object>> metrics = (List<Map<String, Object>>) data.getOrDefault("metrics", List.of());
|
|
|
59
|
+
|
|
|
60
|
+ for (Map<String, Object> metric : metrics) {
|
|
|
61
|
+ String key = (String) metric.get("key");
|
|
|
62
|
+ Object value = metric.get("value");
|
|
|
63
|
+ // 写入 TDengine(简化:用标准 SQL)
|
|
|
64
|
+ String sql = "INSERT INTO water_iot.iot_telemetry (ts, device_sn, metric_key, metric_value, quality) VALUES (NOW, ?, ?, ?, 1)";
|
|
|
65
|
+ jdbcTemplate.update(sql, deviceSn, key, value);
|
|
|
66
|
+ }
|
|
|
67
|
+ } catch (Exception e) {
|
|
|
68
|
+ log.error("Consume error: {}", e.getMessage());
|
|
|
69
|
+ }
|
|
|
70
|
+ }
|
|
|
71
|
+
|
|
|
72
|
+ /** 批量数据采集 API */
|
|
|
73
|
+ public void batchIngest(List<Map<String, Object>> batchData) {
|
|
|
74
|
+ for (Map<String, Object> data : batchData) {
|
|
|
75
|
+ String sourceType = (String) data.getOrDefault("sourceType", "batch");
|
|
|
76
|
+ String sourceId = (String) data.getOrDefault("sourceId", UUID.randomUUID().toString());
|
|
|
77
|
+ @SuppressWarnings("unchecked")
|
|
|
78
|
+ Map<String, Object> rawData = (Map<String, Object>) data.getOrDefault("data", new HashMap<>());
|
|
|
79
|
+ ingest(sourceType, sourceId, rawData);
|
|
|
80
|
+ }
|
|
|
81
|
+ }
|
|
|
82
|
+}
|