org.frameworkset.tran.plugin.mongodb.output.MongoDBOutPutDataTran Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bboss-datatran-mongodb Show documentation

bboss etl,datastream,elasticsearch client with restful and java api without elasticsearch jar dependended.

There is a newer version: 7.3.0

Show newest version

package org.frameworkset.tran.plugin.mongodb.output;

import org.frameworkset.tran.AbstraCommonRecordOutPutDataTran;
import org.frameworkset.tran.JobCountDownLatch;
import org.frameworkset.tran.TranResultSet;
import org.frameworkset.tran.context.ImportContext;
import org.frameworkset.tran.schedule.Status;
import org.frameworkset.tran.schedule.TaskContext;
import org.frameworkset.tran.task.TaskCommand;
import org.frameworkset.tran.task.TaskCommandContext;


public class MongoDBOutPutDataTran extends AbstraCommonRecordOutPutDataTran {
	protected MongoDBOutputConfig mongoDBOutputConfig ;

	@Override
	public void init(){
		super.init();
		mongoDBOutputConfig = (MongoDBOutputConfig) importContext.getOutputConfig();
		StringBuilder builder = new StringBuilder();

		if(mongoDBOutputConfig != null){
			builder.append("Import data to mongodb[").append(mongoDBOutputConfig.getDB())
					.append("] collection[").append(mongoDBOutputConfig.getDBCollection())
					.append("]");
		}

		taskInfo = builder.toString();
	}
	public MongoDBOutPutDataTran(TaskContext taskContext, TranResultSet jdbcResultSet, ImportContext importContext, Status currentStatus,JobCountDownLatch countDownLatch) {
		super(   taskContext,jdbcResultSet,importContext,   currentStatus,countDownLatch);
	}

	public MongoDBOutPutDataTran(TaskContext taskContext, TranResultSet jdbcResultSet, ImportContext importContext, Status currentStatus) {
		super(   taskContext,jdbcResultSet,importContext,   currentStatus);
	}
    @Override
	protected TaskCommand buildTaskCommand(TaskCommandContext taskCommandContext){
		if(!mongoDBOutputConfig.isMultiCollections()) {
			return new MongoDBTaskCommandImpl(taskCommandContext);
		}
		else{
			return new MongoDBMultiTargetTaskCommandImpl(  taskCommandContext);
		}
	}




}