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

xiaoyu28

　　在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 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;
  }