package com.zhitan.service.impl;
|
|
import com.influxdb.client.domain.WritePrecision;
|
import com.influxdb.client.write.Point;
|
import com.zhitan.config.influxdb.InfluxdbConfig;
|
import com.zhitan.influxdb.InfluxdbRepository;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.stereotype.Service;
|
|
import java.time.Instant;
|
import java.time.LocalDateTime;
|
import java.time.ZoneId;
|
import java.util.Random;
|
|
@Slf4j
|
@Service
|
public class WaterMeterSimulator {
|
private final String TAG = "tag";
|
private final String FIELD_VALUE = "value";
|
private final String TAGNAME = "shuibiao01_StTotal";
|
private final InfluxdbRepository repository;
|
private final InfluxdbConfig influxdbConfig;
|
private double lastWaterTotal = 0;
|
private final Random random = new Random();
|
|
@Autowired
|
public WaterMeterSimulator(InfluxdbRepository repository, InfluxdbConfig influxdbConfig) {
|
this.repository = repository;
|
this.influxdbConfig = influxdbConfig;
|
// 初始化时从时序数据库加载上次的数据
|
loadLastWaterData();
|
}
|
|
private void loadLastWaterData() {
|
// 这里应该实现从InfluxDB查询最后一次记录的逻辑
|
double lastValue = repository.getLastPoint(influxdbConfig.getMeasurement(),TAG, TAGNAME);
|
if (lastValue>0) {
|
log.info("查询出最后一次写入influxdb的数据:{}", lastValue);
|
lastWaterTotal = lastValue;
|
}
|
}
|
|
@Scheduled(fixedRate = 60000) // 每分钟执行一次
|
public void simulateWaterUsage() {
|
LocalDateTime now = LocalDateTime.now();
|
double waterUsage = calculateWaterUsage(now);
|
lastWaterTotal += waterUsage;
|
|
Point point = Point
|
.measurement(influxdbConfig.getMeasurement())
|
.addTag(TAG, TAGNAME)
|
.addField(FIELD_VALUE, lastWaterTotal)
|
.time(Instant.now(), WritePrecision.S);
|
|
repository.writePoint(point);
|
log.info("写入水表数据: {}", lastWaterTotal);
|
}
|
|
private double calculateWaterUsage(LocalDateTime time) {
|
int hour = time.getHour();
|
boolean isWeekend = time.getDayOfWeek().getValue() > 5;
|
|
// 基础用水量
|
double baseUsage = 0.05;
|
|
// 工作时间(8-20点)用水量增加
|
if (hour >= 8 && hour < 20) {
|
baseUsage += 0.15;
|
|
// 工作日用水量更大
|
if (!isWeekend) {
|
baseUsage += 0.1;
|
}
|
}
|
|
// 添加随机波动
|
baseUsage += random.nextDouble() * 0.5;
|
|
return baseUsage;
|
}
|
}
|
