gas-web/src/main/java/cc/niushuai/gasweb/controller/ChartController.java
2024-11-18 16:22:24 +08:00

141 lines
4.7 KiB
Java

/*
* Copyright (C) 2023 niushuai233 niushuai.cc
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package cc.niushuai.gasweb.controller;
import cc.niushuai.gasweb.entity.CH2O;
import cc.niushuai.gasweb.entity.ChartResp;
import cc.niushuai.gasweb.entity.Gas;
import cc.niushuai.gasweb.repository.CH2ORepository;
import cc.niushuai.gasweb.repository.GasRepository;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateUtil;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* 图表请求入口
*
* @author niushuai233
* @date 2024/6/7 10:33
* @since 0.0.1
*/
@Tag(name = "图表请求入口")
@Slf4j
@Controller
@RequestMapping("/chart")
public class ChartController {
@Resource
private GasRepository gasRepository;
@Resource
private CH2ORepository ch2ORepository;
@Operation(summary = "test")
@GetMapping("/test")
@ResponseBody
public String test() {
return "hello test";
}
@Operation(summary = "根据类型获取不同表的数据")
@PostMapping("/fetch")
@ResponseBody
public Map<String, ChartResp> fetch(String startDateTime, String endDateTime) {
Map<String, ChartResp> map = new HashMap<>();
fetchGas(startDateTime, endDateTime, map);
fetchCH2O(startDateTime, endDateTime, map);
return map;
}
private void fetchCH2O(String startDateTime, String endDateTime, Map<String, ChartResp> map) {
try {
List<CH2O> allList = ch2ORepository.findAllByTimeBetweenOrderByIdAsc(DateUtil.parseDateTime(startDateTime), DateUtil.parseDateTime(endDateTime));
if (CollUtil.isNotEmpty(allList)) {
ChartResp ch2o = new ChartResp("CH2O浓度");
for (CH2O ch2O : allList) {
String dateTime = DateUtil.formatDateTime(ch2O.getTime());
ch2o.getData().add(new ChartResp.Data(dateTime, ch2O.getCh2o()));
}
map.put("ch2o", ch2o);
}
} catch (Exception e) {
log.error("fetch ch2o failed", e);
map.put("ch2oError", new ChartResp("fetch error: " + e.getMessage()));
}
}
private void fetchGas(String startDateTime, String endDateTime, Map<String, ChartResp> map) {
try {
List<Gas> allList = gasRepository.findAllByTimeBetweenOrderByIdAsc(DateUtil.parseDateTime(startDateTime), DateUtil.parseDateTime(endDateTime));
if (CollUtil.isNotEmpty(allList)) {
ChartResp co = new ChartResp("CO浓度");
ChartResp co2 = new ChartResp("CO2浓度");
ChartResp ch4 = new ChartResp("CH4浓度");
ChartResp temperature = new ChartResp("环境温度");
ChartResp rp2024_temperature = new ChartResp("rp2024温度");
for (Gas gas : allList) {
String dateTime = DateUtil.formatDateTime(gas.getTime());
co.getData().add(new ChartResp.Data(dateTime, gas.getCo()));
co2.getData().add(new ChartResp.Data(dateTime, gas.getCo2()));
ch4.getData().add(new ChartResp.Data(dateTime, gas.getCh4()));
temperature.getData().add(new ChartResp.Data(dateTime, gas.getTemperature()));
rp2024_temperature.getData().add(new ChartResp.Data(dateTime, gas.getRp2040Temperature()));
}
map.put("co", co);
map.put("co2", co2);
map.put("ch4", ch4);
map.put("temperature", temperature);
map.put("rp2024_temperature", rp2024_temperature);
}
} catch (Exception e) {
log.error("fetch gas failed", e);
map.put("gasError", new ChartResp("fetch error: " + e.getMessage()));
}
}
}