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

DocBuilder类的简要代码如下，通过传入context参数到EntityProcessorWrapper类的构造函数，然后执行EntityProcessorWrapper的init()方法初始化数据源；
然后调用EntityProcessorWrapper的相关方法获取数据（全部导入或增量导入数据）

public class DocBuilder {

private void doFullDump() {

buildDocument(getVariableResolver(), null, null, root, true, null);

}

private void doDelta() {

Set<Map<String, Object>> deletedKeys = new HashSet<Map<String, Object>>();

Set<Map<String, Object>> allPks = collectDelta(root, resolver, deletedKeys);

buildDocument(vri, null, map, root, true, null);

}

private void buildDocument(VariableResolverImpl vr, DocWrapper doc, Map<String, Object> pk, DataConfig.Entity entity, boolean isRoot, ContextImpl parentCtx) {

buildDocument(vr, doc, pk, entity, isRoot, parentCtx, entitiesToDestroy);

}

private void buildDocument(VariableResolverImpl vr, DocWrapper doc, Map<String, Object> pk, DataConfig.Entity entity, boolean isRoot, ContextImpl parentCtx, List<EntityProcessorWrapper> entitiesToDestroy) {

EntityProcessorWrapper entityProcessor = getEntityProcessor(entity);

ContextImpl ctx = new ContextImpl(entity, vr, null,

pk == null ? Context.FULL_DUMP : Context.DELTA_DUMP,

session, parentCtx, this);

entityProcessor.init(ctx);

//Map<String, Object> arow = entityProcessor.nextRow();

//其他代码略

}

private EntityProcessorWrapper getEntityProcessor(DataConfig.Entity entity) {

EntityProcessor entityProcessor = new SqlEntityProcessor();

return entity.processor = new EntityProcessorWrapper(entityProcessor, this);

}

DataImporter dataImporter;

public static final String LAST_INDEX_TIME = "last_index_time";

public static final String INDEX_START_TIME = "index_start_time";

public DocBuilder(DataImporter dataImporter, SolrWriter solrWriter, DIHPropertiesWriter propWriter, DataImporter.RequestParams reqParams) {

this.dataImporter = dataImporter;

}

public void execute() {

doDelta();

doFullDump();

}

public Set<Map<String, Object>> collectDelta(DataConfig.Entity entity, VariableResolverImpl resolver, Set<Map<String, Object>> deletedRows) {

//someone called abort

EntityProcessor entityProcessor = getEntityProcessor(entity);

ContextImpl context1 = new ContextImpl(entity, resolver, null, Context.FIND_DELTA, session, null, this);

entityProcessor.init(context1);

Set<Map<String, Object>> myModifiedPks = new HashSet<Map<String, Object>>();

//Map<String, Object> row = entityProcessor.nextModifiedRowKey();

return myModifiedPks;

}

}
相关类图如下