物料编码字典新建及数据导入

whycxzp

2022-08-15 48b5972d9a9a255ecbaecaa4116ec2cd05ae34d1

物料编码字典新建及数据导入

 src/main/java/com/whyc/controller/MaterialCodeDictionaryController.java

New file
@@ -0,0 +1,79 @@
package com.whyc.controller;

import com.whyc.pojo.ComponentApproving;
import com.whyc.pojo.MaterialCodeDictionary;
import com.whyc.service.MaterialCodeDictionaryService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.File;
import java.io.IOException;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;

@RequestMapping("materialCodeDictionary")
@RestController
@Api(tags = "物料编码字典")
public class MaterialCodeDictionaryController {

    @Autowired
    private MaterialCodeDictionaryService service;

    //解析导入数据
    @GetMapping("importData")
    @ApiOperation("导入数据")
    public void  importData() throws IOException, InvalidFormatException {
        List<MaterialCodeDictionary> list = new LinkedList<>();
        File fileTemp = new File("C:\\Users\\29550\\Desktop\\当前项目\\202207图纸管理\\物料类别(1).xlsx");
        Workbook workbook = null;
        workbook = WorkbookFactory.create(fileTemp);
        //取第一个sheet表
        Sheet sheet = workbook.getSheetAt(0);
        int lastRowNum = sheet.getLastRowNum();
        //取第三行,并以第三行开始
        Row rowTemp = sheet.getRow(1);
        short lastCellNum = rowTemp.getLastCellNum();
        //先单独获取产品型号和版本号/分类
        Row row = sheet.getRow(1);

        for (int l = 1; l < lastRowNum+1; l++) {
            Cell cellTemp = sheet.getRow(l).getCell(1);
            cellTemp.setCellType(Cell.CELL_TYPE_STRING);
            if(cellTemp.getStringCellValue().equals("")){
                break;
            }
            MaterialCodeDictionary dictionary = new MaterialCodeDictionary();

            for (int m = 0; m < lastCellNum; m++) {
                row = sheet.getRow(l);
                Cell cell = row.getCell(m);
                String cellValue = null;
                int cellValueInt = 0;
                Double cellValueDouble = null;
                /*if(m==0 || m==2){
                    cell.setCellType(Cell.CELL_TYPE_NUMERIC);
                    cellValueDouble = cell.getNumericCellValue();
                    cellValueInt = cellValueDouble.intValue();
                }else {
                    cellValue = cell.getStringCellValue();
                }*/

                switch (m){
                    case 0:{dictionary.setCode((cell.getStringCellValue()));}break;
                    case 1:{dictionary.setName(cell.getStringCellValue());}break;
                    case 2:{dictionary.setParentCode((cell.getStringCellValue()));}break;
                }
            }
            list.add(dictionary);
        }
        service.insertBatch(list);
    }

}

 src/main/java/com/whyc/mapper/MaterialCodeDictionaryMapper.java

New file
@@ -0,0 +1,6 @@
package com.whyc.mapper;

import com.whyc.pojo.MaterialCodeDictionary;

public interface MaterialCodeDictionaryMapper extends CustomMapper<MaterialCodeDictionary> {
}

 src/main/java/com/whyc/pojo/MaterialCodeDictionary.java

New file
@@ -0,0 +1,51 @@
package com.whyc.pojo;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import org.apache.ibatis.type.Alias;

/**
 * 物料编码字典
 */
@TableName(schema = "db_doc",value = "tb_material_code_dictionary")
@Alias("MaterialCodeDictionary")
public class MaterialCodeDictionary {

    private Integer id;
    private String parentCode;
    private String code;
    private String name;

    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public String getParentCode() {
        return parentCode;
    }

    public void setParentCode(String parentCode) {
        this.parentCode = parentCode;
    }

    public String getCode() {
        return code;
    }

    public void setCode(String code) {
        this.code = code;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}

 src/main/java/com/whyc/service/MaterialCodeDictionaryService.java

New file
@@ -0,0 +1,20 @@
package com.whyc.service;

import com.whyc.mapper.MaterialCodeDictionaryMapper;
import com.whyc.pojo.MaterialCodeDictionary;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class MaterialCodeDictionaryService {

    @Resource
    private MaterialCodeDictionaryMapper mapper;


    public void insertBatch(List<MaterialCodeDictionary> list) {
        mapper.insertBatchSomeColumn(list);
    }
}