package com.whyc.controller;
|
|
import com.whyc.dto.BattDto;
|
import com.whyc.dto.InfoDto;
|
import com.whyc.dto.Response;
|
import com.whyc.factory.InfoFactory;
|
import com.whyc.pojo.db_station.BattInf;
|
import com.whyc.pojo.db_station.PowerInf;
|
import com.whyc.pojo.db_user.User;
|
import com.whyc.service.BaojigroupService;
|
import com.whyc.service.BattInfService;
|
import com.whyc.service.PowerInfService;
|
import com.whyc.util.ActionUtil;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
@RestController
|
@Api(tags = "机房，电源，电池组管理")
|
@RequestMapping("info")
|
public class InfoContoller {
|
|
@Autowired
|
private PowerInfService pinfService;
|
|
@Autowired
|
private BattInfService binfService;
|
|
@Autowired(required = false)
|
private BaojigroupService bjService;
|
|
@ApiOperation(value = "新建机房，电源，电池组信息")
|
@PostMapping("addInfo")
|
public Response addInfo(@RequestBody InfoDto info){
|
//先校验当前用户是否存在包机组不存在则不让添加电源
|
User user= ActionUtil.getUser();
|
int flag=bjService.checkUserBaojiGroup(user.getId());
|
String msg="添加成功";
|
boolean bl=true;
|
if(flag==0){
|
bl=false;
|
msg="当前用户不存在包机组";
|
}
|
try {
|
PowerInf pinf=InfoFactory.infoToPower(info);
|
pinfService.addPower(pinf);
|
if(info.getAddBinfFlag()==1){
|
BattInf binf=InfoFactory.infoToBatt(info);
|
binf.setStationId(pinf.getStationId());
|
binf.setPowerId(pinf.getPowerId());
|
binfService.addDev(binf);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
msg="添加失败";
|
bl=false;
|
}
|
return new Response().set(1,bl,msg);
|
}
|
|
@ApiOperation(value = "设备下添加电池组")
|
@PostMapping("addBatt")
|
public Response addBatt(@RequestBody BattInf binf){
|
binfService.addBatt(binf);
|
return new Response().set(1,true);
|
}
|
|
@ApiOperation(value = "删除电池组")
|
@GetMapping("delBatt")
|
public Response delBatt(@RequestParam Integer stationId,@RequestParam Integer powerId,@RequestParam(required = false) Integer battgroupId){
|
return binfService.delBatt(stationId, powerId, battgroupId);
|
}
|
|
@ApiOperation(value = "编辑机房，电源，电池组信息")
|
@PostMapping("updateInfo")
|
public Response updateInfo(@RequestBody InfoDto info){
|
return binfService.updateInfo(info);
|
}
|
|
@ApiOperation(value = "查询机房，电源，电池组信息")
|
@PostMapping("getInfo")
|
public Response getInfo(@RequestBody BattDto dto){
|
return binfService.getInfo(dto);
|
}
|
}
|
