/*
|
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the dreamlu.net developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: Chill 庄骞 (smallchill@163.com)
|
*/
|
package org.springblade.modules.energy.controller;
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
|
import com.google.common.collect.Lists;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import io.swagger.annotations.ApiParam;
|
import lombok.AllArgsConstructor;
|
import org.springblade.core.boot.ctrl.BladeController;
|
import org.springblade.core.mp.support.Condition;
|
import org.springblade.core.mp.support.Query;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.utils.Func;
|
import org.springblade.modules.dj.entity.Dj;
|
import org.springblade.modules.dj.service.IDjService;
|
import org.springblade.modules.dj.vo.DjVO;
|
import org.springblade.modules.dj.wrapper.DjWrapper;
|
import org.springblade.modules.energy.entity.Energy;
|
import org.springblade.modules.energy.entity.EnergyTree;
|
import org.springblade.modules.energy.service.IEnergyService;
|
import org.springblade.modules.energy.service.IEnergyTreeService;
|
import org.springblade.modules.energy.vo.EnergyVO;
|
import org.springblade.modules.energy.wrapper.EnergyWrapper;
|
import org.springframework.scheduling.annotation.EnableScheduling;
|
import org.springframework.scheduling.annotation.Scheduled;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletResponse;
|
import javax.validation.Valid;
|
import java.text.DateFormat;
|
import java.text.SimpleDateFormat;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 控制器
|
*
|
* @author BladeX
|
* @since 2020-07-19
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/energy")
|
@Api(value = "", tags = "接口")
|
public class EnergyController extends BladeController {
|
|
private final IEnergyService energyService;
|
private final IEnergyTreeService energyTreeService;
|
|
/**
|
* 详情
|
*/
|
@GetMapping("/detail")
|
@ApiOperationSupport(order = 1)
|
@ApiOperation(value = "详情", notes = "传入dj")
|
public R<EnergyVO> detail(Energy energy) {
|
Energy detail = energyService.getOne(Condition.getQueryWrapper(energy));
|
return R.data(EnergyWrapper.build().entityVO(detail));
|
}
|
|
/**
|
* 分页
|
*/
|
@GetMapping("/list")
|
@ApiOperationSupport(order = 2)
|
@ApiOperation(value = "分页", notes = "传入dj")
|
public R<IPage<EnergyVO>> list(Energy energy, Query query) {
|
IPage<Energy> pages = energyService.page(Condition.getPage(query), Condition.getQueryWrapper(energy));
|
return R.data(EnergyWrapper.build().pageVO(pages));
|
}
|
|
|
/**
|
* 新增
|
*/
|
@PostMapping("/save")
|
@ApiOperationSupport(order = 4)
|
@ApiOperation(value = "新增", notes = "传入dj")
|
public R save(@Valid @RequestBody Energy energy) {
|
List<String> list = energyService.selectEnergyTree(energy.getIdentification());
|
if (list == null) {
|
EnergyTree energyTree = new EnergyTree();
|
energyTree.setIdentification(energy.getIdentification());
|
energyTreeService.save(energyTree);
|
}
|
return R.status(energyService.save(energy));
|
}
|
|
/**
|
* 修改
|
*/
|
@PostMapping("/update")
|
@ApiOperationSupport(order = 5)
|
@ApiOperation(value = "修改", notes = "传入dj")
|
public R update(@Valid @RequestBody Energy energy) {
|
return R.status(energyService.updateById(energy));
|
}
|
|
/**
|
* 新增或修改
|
*/
|
@PostMapping("/submit")
|
@ApiOperationSupport(order = 6)
|
@ApiOperation(value = "新增或修改", notes = "传入dj")
|
public R submit(@Valid @RequestBody Energy energy) {
|
return R.status(energyService.saveOrUpdate(energy));
|
}
|
|
|
/**
|
* 删除
|
*/
|
@PostMapping("/remove")
|
@ApiOperationSupport(order = 8)
|
@ApiOperation(value = "删除", notes = "传入ids")
|
public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) {
|
return R.status(energyService.removeByIds(Func.toLongList(ids)));
|
}
|
|
|
/**
|
* 能量统计
|
*
|
* @return
|
*/
|
@PostMapping("/selectTreeNum")
|
public R selectTreeNum() {
|
Date date = new Date();
|
DateFormat format = new SimpleDateFormat("yyyy-MM-dd");
|
String time = format.format(date);
|
List<String> list = energyService.selectTreeNum(time);
|
List<List<String>> lists = splitList(list, 3);
|
return R.data(lists);
|
}
|
|
/**
|
* 拾取能量
|
*
|
* @param id
|
* @return
|
*/
|
@PostMapping("/updateEnergyId")
|
public R updateEnergyId(String id, String identification, int num) {
|
|
energyService.updateEnergyId(id, identification);
|
//能量树
|
String nums = energyTreeService.selectNum(identification);
|
Integer a = Integer.valueOf(nums);
|
Integer n = Integer.valueOf(num);
|
int i1 = n + a;
|
String i = String.valueOf(i1);
|
energyTreeService.updateEnergyTreeId(i, identification);
|
return R.success("拾取成功");
|
}
|
|
/**
|
* 拾取能量
|
*
|
* @return
|
*/
|
@PostMapping("/updateEnergyIds")
|
public R updateEnergyId(@Valid @RequestBody Energy energy) {
|
|
energyService.updateEnergyIds(energy.getId(), energy.getIdentification());
|
//能量树
|
String nums = energyTreeService.selectNum(energy.getIdentification());
|
Integer a = Integer.valueOf(nums);
|
Integer n = Integer.valueOf(energy.getNum());
|
int i1 = n + a;
|
String i = String.valueOf(i1);
|
energyTreeService.updateEnergyTreeId(i, energy.getIdentification());
|
return R.success("拾取成功");
|
}
|
|
|
private List<List<String>> splitList(List<String> list, int groupSize) {
|
int length = list.size();
|
// 计算可以分成多少组
|
int num = (length + groupSize - 1) / groupSize; // TODO
|
List<List<String>> newList = new ArrayList<>(num);
|
for (int i = 0; i < num; i++) {
|
// 开始位置
|
int fromIndex = i * groupSize;
|
// 结束位置
|
int toIndex = (i + 1) * groupSize < length ? (i + 1) * groupSize : length;
|
newList.add(list.subList(fromIndex, toIndex));
|
}
|
return newList;
|
}
|
|
/**
|
* 每天23点自动拾取能量
|
*/
|
@PostMapping("/pickup")
|
public void pickup(String ids, String time, String identification) {
|
String[] split = ids.split(",");
|
String strArrays = "";
|
for (int i = 0; i < split.length; i++) {
|
strArrays += "'" + split[i] + "',";
|
}
|
String code = strArrays.substring(0, strArrays.length() - 1);
|
energyService.updateEnergyType(code, time);
|
//能量树
|
String nums = energyTreeService.selectNum(identification);
|
Integer a = Integer.valueOf(nums);
|
int length = split.length;
|
int i = a + length;
|
String num = String.valueOf(i);
|
energyTreeService.updateEnergyTreeId(num, identification);
|
}
|
|
/**
|
* 浇水
|
*
|
* @param energy
|
* @return
|
*/
|
@PostMapping("/updateTre")
|
public R updateTre(@Valid @RequestBody Energy energy) {
|
//能量树
|
String nums = energyTreeService.selectNum(energy.getIdentification());
|
Integer a = Integer.valueOf(nums);
|
int i1 = a+1;
|
String i = String.valueOf(i1);
|
energyTreeService.updateEnergyTreeId(i, energy.getIdentification());
|
String nu = energyTreeService.selectNum(energy.getIdentification());
|
return R.data(nu);
|
}
|
}
|
