TProjectInfoController.java

package com.zehong.web.controller.supervise;
import java.io.File;
import java.text.ParseException;
import java.util.List;
import com.zehong.common.config.GassafetyProgressConfig;
import com.zehong.common.core.domain.entity.SysUser;
import com.zehong.common.utils.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.zehong.common.annotation.Log;
import com.zehong.common.core.controller.BaseController;
import com.zehong.common.core.domain.AjaxResult;
import com.zehong.common.enums.BusinessType;
import com.zehong.system.domain.TProjectInfo;
import com.zehong.system.service.ITProjectInfoService;
import com.zehong.common.utils.poi.ExcelUtil;
import com.zehong.common.core.page.TableDataInfo;

/**
 * 工程项目信息Controller
 *
 * @author zehong
 * @date 2022-03-16
 */
@RestController
@RequestMapping("/project/info")
public class TProjectInfoController extends BaseController
{
    @Autowired
    private ITProjectInfoService tProjectInfoService;

    /**
     * 查询工程项目信息列表
     */
    @GetMapping("/list")
    public TableDataInfo list(TProjectInfo tProjectInfo)
    {
        //获取用户信息
        SysUser user = SecurityUtils.getLoginUser().getUser();
        tProjectInfo.setBeyondEnterpriseId(user.getDeptId());
        startPage();
        List<TProjectInfo> list = tProjectInfoService.selectTProjectInfoList(tProjectInfo);
        return getDataTable(list);
    }

    /**
     * 导出工程项目信息列表
     */
    @Log(title = "工程项目信息", businessType = BusinessType.EXPORT)
    @GetMapping("/export")
    public AjaxResult export(TProjectInfo tProjectInfo)
    {
        //获取用户信息
        SysUser user = SecurityUtils.getLoginUser().getUser();
        tProjectInfo.setBeyondEnterpriseId(user.getDeptId());
        List<TProjectInfo> list = tProjectInfoService.selectTProjectInfoList(tProjectInfo);
        ExcelUtil<TProjectInfo> util = new ExcelUtil<TProjectInfo>(TProjectInfo.class);
        return util.exportExcel(list, "工程项目信息数据");
    }

    /**
     * 获取工程项目信息详细信息
     */
    @GetMapping(value = "/{projectId}")
    public AjaxResult getInfo(@PathVariable("projectId") Long projectId)
    {
        return AjaxResult.success(tProjectInfoService.selectTProjectInfoById(projectId));
    }

    /**
     * 新增工程项目信息
     */
    @Log(title = "工程项目信息", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody TProjectInfo tProjectInfo) throws ParseException {
        //获取用户信息
        SysUser user = SecurityUtils.getLoginUser().getUser();
        tProjectInfo.setBeyondEnterpriseId(user.getDeptId());
        //查询是否已经有重复的年份数据
        List<TProjectInfo> tProjectInfos = tProjectInfoService.selectSameYear(tProjectInfo.getProjectYear(),tProjectInfo.getBeyondEnterpriseId());
        //说明有重复的年份 将重复的年份删除
        if (tProjectInfos.size()!=0){
            //重复年份删除方法
            tProjectInfoService.deleteisSameYear(tProjectInfo.getProjectYear(),tProjectInfo.getBeyondEnterpriseId());
        }
        if (!"-2".equals(user.getDeptId())){
            //查询企业名称
            String enterpriseName = tProjectInfoService.selectEnterpriseName(user.getDeptId());
            tProjectInfo.setEnterpriseCode(String.valueOf(enterpriseName));
        }else {
            tProjectInfo.setEnterpriseCode("政府部门");
        }
        //添加企业
        tProjectInfo.setBeyondEnterpriseId(user.getDeptId());
        return toAjax(tProjectInfoService.insertTProjectInfo(tProjectInfo));
    }

    /**
     * 修改工程项目信息
     */
    @Log(title = "工程项目信息", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody TProjectInfo tProjectInfo)
    {
        //查询修改文件之前的路径
        TProjectInfo tProjectInfos = tProjectInfoService.selectNearbyAddress(tProjectInfo.getProjectId());
        if (tProjectInfos!=null){
            // 上传文件路径
            String filePath = GassafetyProgressConfig.getUploadPath();
            String[] strs = tProjectInfos.getNearbyAddress().split("upload");
            //删除
            File file = new File(filePath+strs[1].toString());
            // 上传文件路径
            if(file.isFile()){
                file.delete();
            }
        }
        return toAjax(tProjectInfoService.updateTProjectInfo(tProjectInfo));
    }

    /**
     * 删除工程项目信息
     */
    @Log(title = "工程项目信息", businessType = BusinessType.DELETE)
	@DeleteMapping("/{projectIds}")
    public AjaxResult remove(@PathVariable Long[] projectIds)
    {
        return toAjax(tProjectInfoService.deleteTProjectInfoByIds(projectIds));
    }

    @GetMapping(value = "/selectNum")
    public AjaxResult selectNum()
    {
        return AjaxResult.success(tProjectInfoService.selectNum());
    }


}