gunshi-project-ss/src/main/java/com/gunshi/project/xyt/controller/StZqrlBController.java

129 lines
4.6 KiB
Java

package com.gunshi.project.xyt.controller;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.gunshi.core.result.R;
import com.gunshi.project.xyt.model.AttResBase;
import com.gunshi.project.xyt.model.StZqrlB;
import com.gunshi.project.xyt.service.AttResBaseService;
import com.gunshi.project.xyt.service.StStbprpBService;
import com.gunshi.project.xyt.service.StZqrlBService;
import com.gunshi.project.xyt.validate.markers.Delete;
import com.gunshi.project.xyt.validate.markers.Insert;
import com.gunshi.project.xyt.validate.markers.Update;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Date;
import java.util.List;
import java.util.Objects;
/**
* 描述: 水位流量关系曲线表
* author: xusan
* date: 2024-07-08 17:40:37
*/
@Tag(name = "水位流量关系曲线表")
@RestController
@RequestMapping(value="/stZqrlB")
public class StZqrlBController {
@Autowired
private StZqrlBService service;
@Autowired
private AttResBaseService resBaseService;
@Autowired
private StStbprpBService stStbprpBService;
@Operation(summary = "新增")
@PostMapping("/insert")
public R<StZqrlB> insert(@Validated(Insert.class) @RequestBody StZqrlB dto) {
if (StringUtils.isNotBlank(dto.getResCode())){
AttResBase one = resBaseService.lambdaQuery().eq(AttResBase::getResCode, dto.getResCode()).one();
if (Objects.isNull(one)){
throw new IllegalArgumentException("水库编码不存在");
}
if (StringUtils.isBlank(dto.getStcd())){
dto.setStcd(one.getStcd());
}
}
if (StringUtils.isNotBlank(dto.getStcd()) && Objects.isNull(stStbprpBService.getById(dto.getStcd()))){
throw new IllegalArgumentException("测站编码不存在");
}
if (service.lambdaQuery()
.eq(StZqrlB::getStcd, dto.getStcd())
.eq(StZqrlB::getZ, dto.getZ())
.eq(StZqrlB::getQ, dto.getQ())
.count() > 0) {
throw new IllegalArgumentException("当前数据已存在");
}
dto.setId(IdWorker.getId());
dto.setModitime(new Date());
boolean result = service.save(dto);
return R.ok(result ? dto : null);
}
@Operation(summary = "修改")
@PostMapping("/update")
public R<StZqrlB> update(@Validated(Update.class) @RequestBody StZqrlB dto) {
if (StringUtils.isNotBlank(dto.getStcd()) && Objects.isNull(stStbprpBService.getById(dto.getStcd()))){
throw new IllegalArgumentException("测站编码不存在");
}
if (service.lambdaQuery()
.eq(StZqrlB::getStcd, dto.getStcd())
.eq(StZqrlB::getZ, dto.getZ())
.eq(StZqrlB::getQ, dto.getQ())
.ne(StZqrlB::getId,dto.getId())
.count() > 0) {
throw new IllegalArgumentException("当前数据已存在");
}
boolean result = service.lambdaUpdate()
.set(StZqrlB::getZ, dto.getZ())
.set(StZqrlB::getQ, dto.getQ())
.eq(StZqrlB::getId, dto.getId())
.update();
return R.ok(result ? dto : null);
}
@Operation(summary = "删除")
@PostMapping("/del")
public R<Boolean> del(@Validated(Delete.class) @RequestBody StZqrlB dto) {
if (StringUtils.isNotBlank(dto.getStcd()) && Objects.isNull(stStbprpBService.getById(dto.getStcd()))){
throw new IllegalArgumentException("测站编码不存在");
}
if (service.lambdaQuery()
.eq(StZqrlB::getId, dto.getId())
.count() == 0) {
throw new IllegalArgumentException("当前数据不存在");
}
return R.ok(service.lambdaUpdate()
.eq(StZqrlB::getId, dto.getId())
.remove() );
}
@Operation(summary = "列表")
@PostMapping("/list")
public R<List<StZqrlB>> list() {
return R.ok(service.lambdaQuery().orderByDesc(StZqrlB::getModitime).list());
}
// @Operation(summary = "分页")
// @PostMapping("/page")
public R<List<StZqrlB>> page() {
return R.ok(service.page(null,null));
}
}