gunshi-project-ss/src/main/java/com/gunshi/project/hsz/controller/WaterDispatchController.java

package com.gunshi.project.hsz.controller;

import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.gunshi.core.result.R;
import com.gunshi.project.hsz.entity.so.WaterDispatchPageSo;
import com.gunshi.project.hsz.entity.so.XlPlanPageSo;
import com.gunshi.project.hsz.model.IaCDanad;
import com.gunshi.project.hsz.model.WaterDispatch;
import com.gunshi.project.hsz.model.XlPlan;
import com.gunshi.project.hsz.service.FileAssociationsService;
import com.gunshi.project.hsz.service.WaterDispatchService;
import com.gunshi.project.hsz.validate.markers.Insert;
import com.gunshi.project.hsz.validate.markers.Update;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.io.Serializable;
import java.util.Objects;

@Tag(name = "水量调度")
@RestController
@RequestMapping(value="/waterDis")
public class WaterDispatchController extends AbstractCommonFileController {

    @Autowired
    private WaterDispatchService waterDispatchService;

    @Autowired
    private FileAssociationsService fileService;

    @Operation(summary = "新增")
    @PostMapping("/insert")
    public R<WaterDispatch> insert(@Validated(Insert.class) @RequestBody WaterDispatch dto) {
        dto.setId(IdWorker.getId());
        boolean flag = waterDispatchService.saveData(dto);
        if(flag){
            fileService.saveFile(dto.getFiles(),getGroupId(),dto.getId().toString());
        }
        return R.ok(dto);
    }

    @Operation(summary = "下发")
    @GetMapping("/distribute/{id}")
    public R<Boolean> distribute(@PathVariable("id") String id) {
        WaterDispatch byId = waterDispatchService.getById(id);
        if(byId == null){
            throw new IllegalArgumentException("抱歉，该调度不存在，无法下发");
        }
        byId.setExeStatus(2);
        boolean flag = waterDispatchService.updateById(byId);
        return R.ok(flag);
    }

    @Operation(summary = "反馈")
    @PostMapping("feedback")
    public R<WaterDispatch>  feedback(@Validated(Update.class) @RequestBody WaterDispatch dto) {
        boolean flag = waterDispatchService.feedBack(dto);
        if(flag){
            fileService.saveFile(dto.getFiles(),getGroupId(),dto.getId().toString());
        }
        return R.ok(dto);
    }

    @Operation(summary = "更新")
    @PostMapping("/update")
    public R<WaterDispatch> update(@Validated(Update.class) @RequestBody WaterDispatch dto) {
        boolean flag = waterDispatchService.update(dto);
        if(flag){
            fileService.saveFile(dto.getFiles(),getGroupId(),dto.getId().toString());
        }
        return R.ok(dto);
    }

    @Operation(summary = "删除")
    @GetMapping("/del/{id}")
    public R<Boolean> del(@Schema(name = "id") @PathVariable("id") Serializable id) {
        WaterDispatch byId = waterDispatchService.getById(id);
        if(Objects.isNull(byId)){
            throw new IllegalArgumentException("该计划不存在");
        }
        boolean flag = waterDispatchService.removeById(id);
        if(flag){
            fileService.deleteFile(getGroupId(),byId.getId().toString());
        }
        return R.ok(true);
    }

    @Operation(summary = "分页")
    @PostMapping("/page")
    public R<Page<WaterDispatch>> page(@RequestBody WaterDispatchPageSo pageSo) {
        Page<WaterDispatch> res = waterDispatchService.pageQuery(pageSo);
        if(!CollectionUtils.isEmpty(res.getRecords())){
            res.getRecords().forEach(o -> o.setFiles(
                    fileService.getFiles(getGroupId(),o.getId().toString())
            ));
        }
        return R.ok(res);
    }

    @Override
    public String getGroupId() {
        return "waterDispatch";
    }
}