org.frameworkset.tran.mongodb.MongoDBResultSet 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.mongodb;
/**
 * Copyright 2008 biaoping.yin
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * http://www.apache.org/licenses/LICENSE-2.0
 * 

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import com.mongodb.client.MongoCursor;
import org.bson.Document;
import org.frameworkset.tran.*;
import org.frameworkset.tran.context.ImportContext;
import org.frameworkset.tran.record.NextAssert;
import org.frameworkset.tran.record.TranMetaDataLazeLoad;

import java.util.Date;

/**
 * 
Description: 
 * 
 * Copyright (c) 2018
 * @Date 2019/8/3 12:27
 * @author biaoping.yin
 * @version 1.0
 */
public class MongoDBResultSet extends LastValue implements TranResultSet {
	private MongoCursor dbCursor;
	private Document dbObject;
//	private boolean stoped;
	public MongoDBResultSet(ImportContext importContext, MongoCursor dbCursor) {
		this.importContext = importContext;
		this.dbCursor = dbCursor;
	}
 

	@Override
	public Object getValue(int i, String colName, int sqlType) throws DataImportException {
		return getValue(  colName);
	}

	 

	@Override
	public Object getValue(String colName, int sqlType) throws DataImportException {
		return getValue(  colName);
	}

	@Override
	public Date getDateTimeValue(String colName) throws DataImportException {
		return record.getDateTimeValue(colName);

	}

	@Override
	public NextAssert next() throws DataImportException {
        NextAssert nextAssert = new NextAssert();
		if(isStop() || importContext.getInputPlugin().isStopCollectData())
			return nextAssert;
		boolean hasNext = dbCursor.hasNext();
		if( hasNext){
			dbObject = dbCursor.next();
			record = new MongoDBRecord(dbObject,getTaskContext(),importContext);
            record.setTranMeta(this.getMetaData());
		}
        nextAssert.setHasNext(hasNext);
		return nextAssert;
	}

	@Override
	public TranMeta getMetaData() {
		return new DefaultTranMetaData(new TranMetaDataLazeLoad() {
            @Override
            public String[] lazeLoad() {
                return DefaultTranMetaData.convert(dbObject.keySet());
            }
        });
	}

	public Object getKeys(){
		return record.getKeys();
	}
	@Override
	public Object getRecord() {
		return dbObject;
	}



//	@Override
//	public void stop() {
//		stoped = true;
//	}


	@Override
	public Object getMetaValue(String fieldName) {
		return getValue(fieldName);
	}

}