package com.mes.energy.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.mes.energy.entity.EnergyConsumption;
|
import com.mes.energy.service.EnergyConsumptionService;
|
import com.mes.utils.Result;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.time.LocalDateTime;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@Api(tags = "能源管理")
|
@RestController
|
@RequestMapping("/energy/consumption")
|
public class EnergyConsumptionController {
|
|
@Autowired
|
private EnergyConsumptionService energyService;
|
|
@ApiOperation("获取能耗数据列表")
|
@PostMapping("/listEnergy")
|
@ResponseBody
|
public Result list(@RequestBody Map<String, Object> params) {
|
Integer page = params.get("page") == null ? 1 : (Integer) params.get("page");
|
Integer pageSize = params.get("pageSize") == null ? 10 : (Integer) params.get("pageSize");
|
|
Page<EnergyConsumption> pageData = energyService.page(
|
new Page<>(page, pageSize),
|
new QueryWrapper<EnergyConsumption>().orderByDesc("record_date")
|
);
|
|
return Result.build(200, "查询成功", pageData);
|
}
|
|
@ApiOperation("添加能耗数据")
|
@PostMapping("/addEnergy")
|
@ResponseBody
|
public Result add(@RequestBody EnergyConsumption consumption) {
|
consumption.setCreatedAt(new Date());
|
consumption.setUpdatedAt(new Date());
|
int count = energyService.save(consumption) ? 1 : 0;
|
String message = count > 0 ? "添加成功：" + count : "添加失败!";
|
return Result.build(200, message, count);
|
}
|
|
@ApiOperation("修改能耗数据")
|
@PostMapping("/updateEnergy")
|
@ResponseBody
|
public Result update(@RequestBody EnergyConsumption consumption) {
|
consumption.setUpdatedAt(new Date());
|
int count = energyService.updateById(consumption) ? 1 : 0;
|
String message = count > 0 ? "修改成功：" + count : "修改失败!";
|
energyService.notifyEnergyUpdate(consumption);
|
return Result.build(200, message, count);
|
}
|
|
@ApiOperation("删除能耗数据")
|
@PostMapping("/deleteEnergy")
|
@ResponseBody
|
public Result delete(@RequestBody EnergyConsumption consumption) {
|
int count = energyService.removeById(consumption.getId()) ? 1 : 0;
|
String message = count > 0 ? "删除成功：" + count : "删除失败!";
|
return Result.build(200, message, count);
|
}
|
|
@ApiOperation("获取图表数据")
|
@PostMapping("/chartEnergy")
|
@ResponseBody
|
public Result getChartData() {
|
List<EnergyConsumption> data = energyService.list(
|
new QueryWrapper<EnergyConsumption>()
|
.orderByAsc("record_date")
|
.last("limit 30")
|
);
|
|
Map<String, Object> result = new HashMap<>();
|
result.put("actual", data);
|
result.put("planned", null);
|
|
return Result.build(200, "查询成功", result);
|
}
|
}
|
