solr dataimport 数据导入源码分析（七）

在solr dataimport 数据导入源码分析（五） 提到了contextimpl类调用DataImporter对象获取数据源 

 contextimpl.java

  private DataSource ds;

 

  

private DataImporter dataImporter;

   

@Override

   public DataSource getDataSource() {

     if (ds !=  null)  return ds;

     if(entity ==  null)  return   null;

     if (entity.dataSrc ==  null) {

      entity.dataSrc = dataImporter.getDataSourceInstance(entity, entity.dataSource,  this);

    }

     if (entity.dataSrc !=  null && docBuilder !=  null && docBuilder.verboseDebug &&

             Context.FULL_DUMP.equals(currentProcess())) {

       // debug is not yet implemented properly for deltas

      entity.dataSrc = docBuilder.getDebugLogger().wrapDs(entity.dataSrc);

    }

     return entity.dataSrc;

  }

我们跟踪源码，在DataImporter.java里面调用了jdbcDataSource对象方法

参考solr dataimport 数据导入源码分析（二） DataImporter.java 源码

DataImporter.java  

 DataSource getDataSourceInstance(DataConfig.Entity key, String name, Context ctx) {

    Properties p = dataSourceProps.get(name);
     if (p ==  null)
      p = config.dataSources.get(name);
     if (p ==  null)
      p = dataSourceProps.get( null); //  for default data source
     if (p ==  null)
      p = config.dataSources.get( null);
     if (p ==  null)  
       throw  new DataImportHandlerException(SEVERE,
              "No dataSource :" + name + " available for entity :"
                      + key.name);
    String type = p.getProperty(TYPE);
    DataSource dataSrc =  null;
     if (type ==  null) {
       dataSrc =  new  JdbcDataSource();
    }  else {
       try {
        dataSrc = (DataSource) DocBuilder.loadClass(type, getCore()).newInstance();
      }  catch (Exception e) {
        wrapAndThrow(SEVERE, e, "Invalid type for data source: " + type);
      }
    }
     try {
      Properties copyProps =  new Properties();
      copyProps.putAll(p);
      Map<String, Object> map = ctx.getRequestParameters();
       if (map.containsKey("rows")) {
         int rows = Integer.parseInt((String) map.get("rows"));
         if (map.containsKey("start")) {
          rows += Integer.parseInt((String) map.get("start"));
        }
        copyProps.setProperty("maxRows", String.valueOf(rows));
      }
      dataSrc.init(ctx, copyProps);
    }  catch (Exception e) {
      wrapAndThrow(SEVERE, e, "Failed to initialize DataSource: " + key.dataSource);
    }
     return dataSrc;
  }