java 导入csv 文件反射调用Mapper接口

1.引入pom依赖


       org.apache.commons
       commons-csv
       1.6

2.创建工具类 CsvUtils

也可以自己在网上找模板,我也是网上借鉴的

public class CsvUtils {

    /**
     * io流导出
     * @author kpzc
     * @date 2018年12月29日 下午3:48:34
     * @param file csv文件(路径+文件名),csv文件不存在会自动创建
     * @param dataList 数据,字符串用逗号分隔
     * @return 返回导出是否成功 true成功 false失败
     */
    public static boolean exportCsv(File file, List dataList){
        boolean isSucess=false;

        FileOutputStream out=null;
        OutputStreamWriter osw=null;
        BufferedWriter bw=null;
        try {
            out = new FileOutputStream(file);
            //解决FileOutputStream中文乱码问题  解决MS office乱码问题
            osw = new OutputStreamWriter(out, "GBK");
            bw =new BufferedWriter(osw);
            if(dataList!=null && !dataList.isEmpty()){
                for(String data : dataList){
                    bw.append(data).append("\r");
                }
            }
            isSucess=true;
        } catch (Exception e) {
            isSucess=false;
        }finally{
            if(bw!=null){
                try {
                    bw.close();
                    bw=null;
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(osw!=null){
                try {
                    osw.close();
                    osw=null;
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if(out!=null){
                try {
                    out.close();
                    out=null;
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return isSucess;
    }

    /**
     * 导入
     * @author kpzc
     * @date 2018年12月29日 下午3:48:11
     * @param file  csv文件(路径+文件)
     * @return 返回List列表
     */
    public static List importCsv(MultipartFile file){
        List dataList=new ArrayList();
        BufferedReader br=null;
        try {
            br = new BufferedReader(new InputStreamReader(file.getInputStream(), "UTF-8"));
            String line = "";
            while ((line = br.readLine()) != null) {
                dataList.add(line.replace("\"",""));
            }
        }catch (Exception e) {
            e.printStackTrace();
        }finally{
            if(br!=null){
                try {
                    br.close();
                    br=null;
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return dataList;
    }

    /**
     * apache commons-csv导出
     * 注意jdk要在1.7及以上使用
     * map的数据个数要与header的个数相等 并且一一对应,可参照main方法
     * @author kpzc
     * @date 2019年1月4日 上午10:12:20
     * @param filePath 文件存储路径
     * @param list 数据列表
     * @param header 表头
     */
    public static void write(String filePath, List> list, String... header) {
        try {
            FileOutputStream fos = new FileOutputStream(filePath);
            OutputStreamWriter osw = new OutputStreamWriter(fos, "GBK");
            CSVFormat csvFormat = CSVFormat.DEFAULT.withHeader(header);
            CSVPrinter csvPrinter = new CSVPrinter(osw, csvFormat);
            //跟上面两行代码是一样的
            //CSVPrinter csvPrinter = CSVFormat.DEFAULT.withHeader(header).print(osw);
            for (Map map : list) {
                csvPrinter.printRecord(map.values());
            }
            csvPrinter.flush();
            csvPrinter.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}

3.创建SpringContextUtil 获取上下文对象

@Component
public class SpringContextUtil implements ApplicationContextAware {
    private static ApplicationContext appCtx = null;

    /**
     * @param applicationContext
     *            ApplicationContext.
     * @throws BeansException
     */
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        appCtx = applicationContext;
    }

    public static ApplicationContext getApplicationContext() {
        return appCtx;
    }

    public static Object getBean(String beanName) {
        return appCtx.getBean(beanName);
    }
}

4.导入文件类

@Override
    public ResultInfo importFile(ImportRo importRo, HttpServletRequest request) {
        String path = "com.senhua.dao." + importRo.getEntityName() + "Mapper";//Mapper文件都是放在此目录下,所以写死了
        List list = CsvUtils.importCsv(importRo.getFile());
        //System.out.println("list = " + list.get(1));
        try{
            if(!list.isEmpty()){
                Class c = Class.forName(path);
                // 得到对象中所有的方法
                Method[] methods = c.getMethods();
                Class param = null;
                Field[] fields = null;

                for (int i = 0; i < methods.length; i++) {
                    if(methods[i].getName().equals("insertSelective")){
                        Class[] paramTypes = methods[i].getParameterTypes();//获取方法参数类型
                        param = paramTypes[0];
                        // 得到对象中所有的属性
                        fields = param.getDeclaredFields();
                    }
                }
                //获取反射类方法
                Method m = c.getMethod("insertSelective",param);
                boolean flag = false;
                for(int i=1;i

 

 

 

 

 

 

 

 

你可能感兴趣的:(Java)