Java读取csv导入solr

最新推荐文章于 2024-09-14 18:46:05 发布

weixin_47152928

最新推荐文章于 2024-09-14 18:46:05 发布

阅读量700

点赞数 1

文章标签： solr java

本文链接：https://blog.csdn.net/weixin_47152928/article/details/122235600

版权

public static void main(String[] args)  {

    try { //CsvReader csvReader = null;

        CsvReader csvReader = new CsvReader(
                "F:\\ImportData\\01.csv", ',');
        csvReader.readHeaders();
   
        SolrCloudServiceImpl solr = new SolrCloudServiceImpl();
        Collection<SolrInputDocument> dwfdocs = new ArrayList<SolrInputDocument>();
        // 读取每行的内容
        
        while (csvReader.readRecord()) {
                SolrInputDocument overviewdoc = new SolrInputDocument();
                overviewdoc.addField("id", UUID.randomUUID().toString());
                overviewdoc.addField("linkedinid_s",csvReader.get(0));
                overviewdoc.addField("username_s",csvReader.get(1));
                overviewdoc.addField("fb_s",csvReader.get(2));
                overviewdoc.addField("tw_s",csvReader.get(3));
                overviewdoc.addField("github_s",csvReader.get(4));
                overviewdoc.addField("email_s",csvReader.get(5));
                overviewdoc.addField("phone_s",csvReader.get(6));
                dwfdocs.add(overviewdoc);
                if(dwfdocs.size()>=10000){
                    solr.connectSolrCloud();
                    solr.indexData("solr库名字", dwfdocs);
                    dwfdocs.clear();
                }
            }
        solr.connectSolrCloud();
        solr.indexData("solr库名字", dwfdocs);
        
    } catch (IOException e) {
        e.printStackTrace();
    }
}