CsvUtil

来源:互联网 发布:newsql数据库 编辑:程序博客网 时间:2024/04/29 18:53
package cn.net.zzfz.center.common.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.lang.reflect.InvocationTargetException;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
import java.util.Map;

import org.apache.commons.beanutils.BeanUtils;
import org.apache.commons.beanutils.PropertyUtils;
import org.apache.commons.lang3.StringUtils;

import au.com.bytecode.opencsv.CSVReader;
import au.com.bytecode.opencsv.bean.ColumnPositionMappingStrategy;

public class CsvUtil {

    private static String character_encode = "UTF-8";
    
    static SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");    
    
    public static <T> List<T> readCSV(String filename,Map<String, String> headerProp, T bean) throws IOException, IllegalAccessException, InvocationTargetException, NoSuchMethodException, ParseException, InstantiationException{
        List<T> list = new ArrayList<T>();
        InputStreamReader reader = new InputStreamReader(new FileInputStream(new File(filename)), character_encode);
        CSVReader csv = new CSVReader(reader);
        
        String[] header = csv.readNext();
        List<String[]> content = csv.readAll();
        for (String[] str : content) {
            Object obj = bean.getClass().newInstance();
            for (int i = 0; i < str.length; i++) {
                if(StringUtils.isBlank(str[i])){
                    continue;
                }
                String key = StringUtils.trimToEmpty(header[i]);
                if(PropertyUtils.getPropertyType(obj, headerProp.get(key)).isAssignableFrom(Calendar.class)){
                    Calendar calendar = DateJodaTimeUtils.stringToCalendar(str[i]);
                    BeanUtils.setProperty(obj, headerProp.get(key), calendar);
                }else if(PropertyUtils.getPropertyType(obj, headerProp.get(key)).isAssignableFrom(Date.class)){
                    Date date = sdf.parse(str[i]);
                    BeanUtils.setProperty(obj, headerProp.get(key), date);
                }else{
                    BeanUtils.copyProperty(obj, headerProp.get(key), str[i]);
                }
            }
            list.add((T) obj);
        }
        return list;
    }
    
    public static <T> void writeCSV(String filename, Map<String, String> headerMap, List<T> list){
        
        ColumnPositionMappingStrategy d;
    }
}

0 0
原创粉丝点击