SRMDetailCheckController.java 5.17 KB
package com.huaheng.pc.srm.controller;


import com.huaheng.common.support.Convert;
import com.huaheng.common.utils.StringUtils;
import com.huaheng.common.utils.file.FileUtils;
import com.huaheng.common.utils.file.ZipUtils;
import com.huaheng.framework.aspectj.lang.annotation.Log;
import com.huaheng.framework.aspectj.lang.constant.BusinessType;
import com.huaheng.framework.config.HuaHengConfig;
import com.huaheng.framework.web.controller.BaseController;
import com.huaheng.framework.web.domain.AjaxResult;
import com.huaheng.framework.web.page.TableDataInfo;
import com.huaheng.pc.config.company.service.CompanyService;
import com.huaheng.pc.config.warehouse.domain.WarehouseU8;
import com.huaheng.pc.config.warehouse.service.WarehouseU8Service;

import com.huaheng.pc.srm.domain.SrmDetail;
import com.huaheng.pc.srm.service.SrmDetailService;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.util.List;


@Controller
@RequestMapping("/srm/srmDetailCheck")
public class SRMDetailCheckController extends BaseController {



    private static final Logger log = LoggerFactory.getLogger(SRMDetailCheckController.class);


    private String prefix = "srm/srmDetailCheck";

    @RequiresPermissions("srm:srmDetail:view")
    @GetMapping("/{headerId}")
    public String srmDetail(@PathVariable("headerId") String headerId, ModelMap mmap)
    {
        mmap.put("headerId", headerId);
        return prefix + "/srmDetailCheck";
    }


    @Resource
    private SrmDetailService srmDetailService;
    @Resource
    private CompanyService companyService;
    @Resource
    private WarehouseU8Service warehouseU8Service;

    /**
     * 查询已到送货单明细
     */
    @RequiresPermissions("srm:srmDetail:list")
    @Log(title = "送货单-送货单", operating="查看送货单明细", action = BusinessType.GRANT)
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(SrmDetail srmDetail)
    {
        srmDetail.setCheckCode(2);
        startPage();
        List<SrmDetail> list = srmDetailService.selectList(srmDetail) ;
        for (SrmDetail detail : list) {
            WarehouseU8 warehouseWu = new WarehouseU8();
            warehouseWu.setWarehouseCode("KS0001");
            warehouseWu.setUCompanyCode(detail.getCompanyCode());
            warehouseWu.setUWarehouseCode(detail.getUWarehouseCode());
            List<WarehouseU8> warelist= warehouseU8Service.getByDomain(warehouseWu);
            if(warelist != null&&warelist.size()>0){
                warehouseWu=warelist.get(0);
                detail.setUWarehouseName(warehouseWu.getUWarehouseName());
            }
        }
        return getDataTable(list);
    }


    @GetMapping("/uploadAnnex/{id}")
    public String uploadAnnex(@PathVariable String id, ModelMap modelMap) {
        modelMap.put("id", id);
        return prefix + "/uploadAnnex";
    }

    @PostMapping("/uploadAnnex")
    @ResponseBody
    public AjaxResult updateAnnex(SrmDetail srmDetail) {
        SrmDetail condition = new SrmDetail();
        condition.setId(srmDetail.getId());
        condition = srmDetailService.selectFirstModel(condition);
        if(condition == null){
            return AjaxResult.error("没有此送货单明细");
        }
        if(StringUtils.isEmpty(condition.getAnnex())){
            condition.setAnnex(srmDetail.getAnnex());
        }else {
            condition.setAnnex(condition.getAnnex()+";"+srmDetail.getAnnex());
        }
        return toAjax(srmDetailService.updateById(condition));
    }


    @GetMapping("/downloadAnnex/{id}")
    public void downloadAnnex(@PathVariable String id, HttpServletResponse response, HttpServletRequest request) {
        SrmDetail srmDetail = new SrmDetail();
        srmDetail.setId(Integer.valueOf(id));
        srmDetail = srmDetailService.selectFirstModel(srmDetail);

        String[] fileNameArray = Convert.toStrArray(";", srmDetail.getAnnex());
        File[] srcFiles = new File[fileNameArray.length];
        String fileName = "ZipFile.zip";
        File zipFile = new File(HuaHengConfig.getProfile() + fileName);
        for (int i = 0; i < fileNameArray.length; i++) {
            srcFiles[i] = new File(HuaHengConfig.getProfile() + fileNameArray[i]);
        }
        ZipUtils.zipFiles(srcFiles, zipFile);
        try {
            String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1);
            String filePath = HuaHengConfig.getProfile() + fileName;

            response.setCharacterEncoding("utf-8");
            response.setContentType("multipart/form-data");
            response.setHeader("Content-Disposition", "attachment;fileName=" + FileUtils.setFileDownloadHeader(request, realFileName));
            FileUtils.writeBytes(filePath, response.getOutputStream());
            FileUtils.deleteFile(filePath);

        } catch (Exception e) {
            log.error("下载文件失败", e);
        }
    }
}