wrdep/wrdep-biz/src/main/java/com/jiayue/biz/controller/ManualEntryController.java

package com.jiayue.biz.controller;

import com.jiayue.biz.domain.*;
import com.jiayue.biz.service.*;
import com.jiayue.biz.vo.TotalityInfoForPInfoVo;
import com.jiayue.common.core.controller.BaseController;
import com.jiayue.common.core.domain.AjaxResult;
import lombok.AllArgsConstructor;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.parameters.P;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 人工录入Controller
 *
 * @date 2023-05-9
 */
@RequiredArgsConstructor(onConstructor_ = @Autowired)
@RestController
@RequestMapping("/manualEntry")
public class ManualEntryController extends BaseController {
    private final ManualEntryService manualEntryService;
    private final TotalityInfoService totalityInfoService;
    private final ProjectInfoService projectInfoService;
    private final StationInfoService stationInfoService;
    private final FanModelDataService fanModelDataService;
    private final ProjectProgressService projectProgressService;

    /*获取光伏站下拉框，风场站的下拉框*/
    @GetMapping("/stationInfo")
    public AjaxResult hauFuInfo() {
        return AjaxResult.success(manualEntryService.getStationInfo());
    }

    //获取华电以及全省资源概况
    @GetMapping("/getProvincialEnergyStations")
    public AjaxResult getProvincialEnergyStations() {
        return AjaxResult.success(manualEntryService.getProvincialEnergyStations());
    }

    //获取项目（场站）总体信息
    @GetMapping("/getTotalityInfo")
    public AjaxResult getTotalityInfo() {
        return AjaxResult.success(totalityInfoService.selectTotalityInfoList());
    }

    /**
     * 项目进展获取项目名称+id
     */
    @GetMapping("/getProjectInfo")
    public AjaxResult getProjectInfo() {
        List<Map<String, String>> list = new ArrayList<>();
        List<ProjectInfo> projectInfoList = projectInfoService.getProjectInfoList();
        for (ProjectInfo projectInfo : projectInfoList) {
            Map<String, String> map = new HashMap<>();
            map.put("id", projectInfo.getId());
            map.put("name", projectInfo.getProjectBasicInfo().getProjectName());
            list.add(map);
        }
        return AjaxResult.success(list);
    }

    /**
     * 获取场站信息
     */
    @GetMapping("/getStationInfo")
    public AjaxResult getStationInfo() {
        List<Map<String, String>> list = new ArrayList<>();
        List<StationInfo> stationInfos = stationInfoService.selectStationInfo();
        for (StationInfo stationInfo : stationInfos) {
            Map<String, String> map = new HashMap<>();
            map.put("id", stationInfo.getId());
            map.put("name", stationInfo.getStationBasicInfo().getStationName());
            map.put("easyName", stationInfo.getStationBasicInfo().getAbbreviation());
            list.add(map);
        }
        return AjaxResult.success(list);
    }

    /**
     * 资源概况（更新）
     *
     * @param provincialEnergyStations 资源概况
     */
    @PostMapping("/updateProvincialEnergyStations")
    public void update(@RequestBody ProvincialEnergyStations provincialEnergyStations) {
        manualEntryService.updateProvincialEnergyStations(provincialEnergyStations);
    }

    /*获取风机统计数据*/
    @GetMapping("/getFanModelDataList")
    public AjaxResult getFanModelDataList(String id) {
        return AjaxResult.success(fanModelDataService.queryFanModelDataList(id));
    }

    // 修改项目的总体信息
    @PostMapping("/updateProjectTotalityInfoList")
    public AjaxResult updateStationTotalityInfoList(@RequestBody TotalityInfoForPInfoVo totalityInfoForPInfoVo) {
        totalityInfoService.updateProjectTotalityInfoList(totalityInfoForPInfoVo);
        return AjaxResult.success("200");
    }

    // 修改场站的总体信息
    @PostMapping("/updateStationTotalityInfoList")
    public AjaxResult updateStationTotalityInfoList(@RequestBody String data) {
        totalityInfoService.updateStationTotalityInfoList(data);
        return AjaxResult.success("200");
    }

    @DeleteMapping("/deleteFanModelData/{id}")
    public AjaxResult deleteFanModelData(@PathVariable("id") String id) {
        fanModelDataService.deleteFanModelData(id);
        return AjaxResult.success("200");
    }

    @PostMapping("/saveFanModelData")
    public AjaxResult updateFanModelData(@RequestBody FanModelData fanModelData) {
        fanModelDataService.updateFanModelData(fanModelData);
        return AjaxResult.success("200");
    }

    //项目进展解析
    @PostMapping("/projectProgressParse")
    public AjaxResult projectProgressParse(@RequestParam("file") MultipartFile multipartFile, @RequestParam("id") String eqId, @RequestParam("name") String name) {

        projectProgressService.projectProgressParse(multipartFile, eqId);
        return AjaxResult.success("200");

    }

    /**
     * 保存附件
     *
     * @param multipartFile 文件
     * @param id            文件夹名
     */
    @PostMapping("/uploadProjectAnnex")
    public AjaxResult saveAttachment(@RequestParam("file") MultipartFile multipartFile, @RequestParam("id") String id) {

        projectProgressService.saveAttachment(multipartFile, id);
        return AjaxResult.success("200");
    }

    //下载文件
    @GetMapping("/exportProjectEvolve")
    public void downloadFile(HttpServletResponse response, String equipmentId) {
        projectProgressService.downloadProjectProgress(response, equipmentId);
    }


}