tld-consumer-9560/src/main/java/com/tld/service/impl/GoodsServiceImpl.java

package com.tld.service.impl;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.tld.excel.ExcelUtils;
import com.tld.mapper.GoodsMapper;
import com.tld.model.Goods;
import com.tld.service.GoodsService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.servlet.http.HttpServletResponse;
import java.text.SimpleDateFormat;
import java.util.*;

@Service
public class GoodsServiceImpl implements GoodsService {

    @Autowired
    private GoodsMapper goodsMapper;

    @Override
    public Map<String, Object> getGoods(Goods goods) {
        Map<String, Object> map = new HashMap<>();
        try{
            PageHelper.startPage(goods.getPage(), goods.getLimit());
            PageInfo<Goods> list = new PageInfo<>(goodsMapper.getStorage(goods));
            map.put("data", list);
            map.put("msg", "200");
        }catch (Exception e){
            e.printStackTrace();
            map.put("msg", "500");
            map.put("errMsg", "服务器请求异常，请稍后再试");
        }
        return map;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Map<String, Object> addGoods(Goods goods) {
        Map<String, Object> map = new HashMap<>();
        try{
            goodsMapper.addGoods(goods);
            map.put("msg", "200");
        }catch (Exception e){
            e.printStackTrace();
            map.put("msg", "500");
            map.put("errMsg", "服务器请求异常，请稍后再试");
        }
        return map;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Map<String, Object> delGoods(String id) {
        Map<String, Object> map = new HashMap<>();
        try{
            goodsMapper.delGoods(id);
            map.put("msg", "200");
        }catch (Exception e){
            e.printStackTrace();
            map.put("msg", "500");
            map.put("errMsg", "服务器请求异常，请稍后再试");
        }
        return map;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Map<String, Object> updateGoods(Goods goods) {
        Map<String, Object> map = new HashMap<>();
        try{
            goodsMapper.updateGoods(goods);
            map.put("msg", "200");
        }catch (Exception e){
            e.printStackTrace();
            map.put("msg", "500");
            map.put("errMsg", "服务器请求异常，请稍后再试");
        }
        return map;
    }

    @Override
    public void export(Goods goods, HttpServletResponse response) {
        try{
            //导出数据汇总
            List<List<Object>> sheetDataList = new ArrayList<>();
            //表头数据
            List<Object> head =
                    Arrays.asList("品号", "品名", "类型","库位容纳小盒","收容数","库位最大容纳","供应商/客户","库位","投入方式","使用部门","低储量","高储量","等级","包装类别","单位");
            //查询数据
            PageHelper.startPage(goods.getPage(), goods.getLimit());
            PageInfo<Map<String, Object>> list = new PageInfo<>(goodsMapper.export(goods));
            sheetDataList.add(head);
            for(Map<String, Object> userMap : list.getList()){
                List<Object> listSheet = new ArrayList<>();
                for(String key: userMap.keySet()){
                    listSheet.add(userMap.get(key));
                }
                sheetDataList.add(listSheet);
            }
            //当前时间
            Date time = new Date();
            SimpleDateFormat sdf = new SimpleDateFormat("yyyyMddHHmmss");
            ExcelUtils.export(response, "货品数据导出" + sdf.format(time), sheetDataList);
        }catch (Exception e){
            e.printStackTrace();
        }
    }
}