tocc-sq/ruoyi-admin/src/main/java/com/ruoyi/web/controller/zhdd/ResourceDetailController.java

package com.ruoyi.web.controller.zhdd;

import cn.afterturn.easypoi.entity.vo.NormalExcelConstants;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.enmus.ExcelType;
import cn.afterturn.easypoi.view.PoiBaseView;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.annotation.RepeatSubmit;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.entity.SysDictData;
import com.ruoyi.common.core.page.TableDataInfo;
import com.ruoyi.common.core.validate.AddGroup;
import com.ruoyi.common.core.validate.EditGroup;
import com.ruoyi.common.core.validate.QueryGroup;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.system.service.ISysDictTypeService;
import com.ruoyi.zhdd.domain.Resource;
import com.ruoyi.zhdd.domain.ResourceDetail;
import com.ruoyi.zhdd.domain.bo.ResourceDetailBo;
import com.ruoyi.zhdd.domain.vo.ResourceDetailExport;
import com.ruoyi.zhdd.domain.vo.ResourceDetailVo;
import com.ruoyi.zhdd.domain.vo.ResourceVo;
import com.ruoyi.zhdd.service.IResourceDetailService;
import com.ruoyi.zhdd.service.IResourceService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.ui.ModelMap;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.NotEmpty;
import javax.validation.constraints.NotNull;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 应急资源明细Controller
 *
 * @author xintong
 * @date 2021-10-27
 */
@Validated
@Api(value = "应急资源明细控制器", tags = {"应急资源明细管理"})
@RequiredArgsConstructor(onConstructor_ = @Autowired)
@RestController
@RequestMapping("/zhdd/resourceDetail")
public class ResourceDetailController extends BaseController {

    private final IResourceDetailService iResourceDetailService;
    private final IResourceService resourceService;
    private final ISysDictTypeService dictTypeService;

    /**
     * 查询应急资源明细列表
     */
    @ApiOperation("查询应急资源明细列表")
    @GetMapping("/list")
    public TableDataInfo<ResourceDetailVo> list(@Validated(QueryGroup.class) ResourceDetailBo bo) {
        return iResourceDetailService.queryPageList(bo);
    }

    /**
     * 导出应急资源明细列表
     */
    @ApiOperation("导出应急资源明细列表")
    @Log(title = "应急资源明细", businessType = BusinessType.EXPORT)
    @GetMapping("/export/{ids}")
    public void export(@PathVariable String[] ids, HttpServletResponse response, HttpServletRequest request, ModelMap modelMap) {
        List<ResourceDetailVo> list = iResourceDetailService.listVo(Wrappers.<ResourceDetail>lambdaQuery()
            .in(ResourceDetail::getResourceId, Arrays.asList(ids)).orderByAsc(ResourceDetail::getResourceId));
        // 查询仓库名称
        List<Resource> list1 = resourceService.list(Wrappers.<Resource>lambdaQuery().in(Resource::getId, Arrays.asList(ids)));
        Map<String, Resource> collect = list1.stream().collect(Collectors.toMap(Resource::getId, a -> a));

        Map<String, List<ResourceDetailVo>> collect1 = list.stream().collect(Collectors.groupingBy(ResourceDetailVo::getResourceId));

        // 行业主管单位
        List<SysDictData> zhdd_admin_org = dictTypeService.selectDictDataByType("zhdd_admin_org");
        Map<String, String> adminOrgCollect = zhdd_admin_org.stream().collect(Collectors.toMap(SysDictData::getDictValue, SysDictData::getDictLabel));

        List<ResourceDetailExport> exportList = new ArrayList<>();
        for (Map.Entry<String, List<ResourceDetailVo>> stringListEntry : collect1.entrySet()) {
            ResourceDetailExport bean = new ResourceDetailExport();
            Resource resource = collect.get(stringListEntry.getKey());
            bean.setAdminOrgName(StrUtil.isNotBlank(resource.getAdminOrgName()) ? adminOrgCollect.get(resource.getAdminOrgName()) : "");
            bean.setResourceName(resource.getName());
            bean.setManageUnit(resource.getManageUnit());
            bean.setContactName(resource.getContactName());
            bean.setContactPhone(resource.getContactPhone());
            bean.setAddress(resource.getAddress());
            bean.setDetail(stringListEntry.getValue());
            exportList.add(bean);
        }
        ExportParams params = new ExportParams("仓库物资明细", "仓库物资", ExcelType.XSSF);
        params.setFreezeCol(5);
        modelMap.put(NormalExcelConstants.DATA_LIST, exportList);
        modelMap.put(NormalExcelConstants.CLASS, ResourceDetailExport.class);
        modelMap.put(NormalExcelConstants.PARAMS, params);
        modelMap.put(NormalExcelConstants.FILE_NAME, "仓库物资明细");
        PoiBaseView.render(modelMap, request, response, NormalExcelConstants.EASYPOI_EXCEL_VIEW);
    }

    /**
     * 获取应急资源明细详细信息
     */
    @ApiOperation("获取应急资源明细详细信息")
    @GetMapping("/{id}")
    public AjaxResult<ResourceDetailVo> getInfo(@NotNull(message = "主键不能为空")
                                                @PathVariable("id") String id) {
        return AjaxResult.success(iResourceDetailService.queryById(id));
    }

    /**
     * 新增应急资源明细
     */
    @ApiOperation("新增应急资源明细")
    @Log(title = "应急资源明细", businessType = BusinessType.INSERT)
    @RepeatSubmit()
    @PostMapping()
    public AjaxResult<Void> add(@Validated(AddGroup.class) @RequestBody ResourceDetailBo bo) {
        return toAjax(iResourceDetailService.insertByBo(bo) ? 1 : 0);
    }

    /**
     * 修改应急资源明细
     */
    @ApiOperation("修改应急资源明细")
    @Log(title = "应急资源明细", businessType = BusinessType.UPDATE)
    @RepeatSubmit()
    @PutMapping()
    public AjaxResult<Void> edit(@Validated(EditGroup.class) @RequestBody ResourceDetailBo bo) {
        return toAjax(iResourceDetailService.updateByBo(bo) ? 1 : 0);
    }

    /**
     * 删除应急资源明细
     */
    @ApiOperation("删除应急资源明细")
    @Log(title = "应急资源明细", businessType = BusinessType.DELETE)
    @DeleteMapping("/{ids}")
    public AjaxResult<Void> remove(@NotEmpty(message = "主键不能为空")
                                   @PathVariable String[] ids) {
        return toAjax(iResourceDetailService.deleteWithValidByIds(Arrays.asList(ids), true) ? 1 : 0);
    }

    @ApiOperation("通过物资检索应急仓库")
    @GetMapping("/searchWarehouseByGoods")
    public AjaxResult<List<ResourceVo>> searchWarehouseByGoods(ResourceDetailBo bo) {
        List<ResourceVo> list = iResourceDetailService.searchWarehouseByGoods(bo);
        return AjaxResult.success(list);
    }
}