org.teasoft.beex.mongodb.ds.MongodbSimpleDataSource Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016-2023 the original author.All rights reserved.
 * Kingstar([email protected])
 * The license,see the LICENSE file.
 */

package org.teasoft.beex.mongodb.ds;

import java.io.IOException;

import org.teasoft.honey.database.ClientDataSource;

import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoDatabase;

/**
 * @author Kingstar
 * @since  2.0
 */
public class MongodbSimpleDataSource extends ClientDataSource {

	private MongodbManager mongodbManager = null;

	private String url;
	private String username;
	private String password;

	private boolean inited = false;
	
	public MongodbSimpleDataSource() {}

	public MongodbSimpleDataSource(String url, String username, String password) {
		this.url = url;
		this.username = username;
		this.password = password;

		init();
	}

	public String getUrl() {
		return url;
	}

	public void init() {
		if (inited) return;
		mongodbManager = new MongodbManager(getUrl(), getUsername(), getPassword());
		inited = true;
	}
	
	@Override
	public Object getDbConnection() {
		return getMongoDb();
	}
	
	private MongoDatabase getMongoDb() {
		MongoClient client=mongodbManager.getMongoClient();
		//从资源池中拿, 或放到上下文or当前线程中管理.   todo
		MongoContext.setCurrentMongoClient(client);
		return client.getDatabase(mongodbManager.getDatabaseName());
	}

	@Override
	public void close() throws IOException {
		
		MongoContext.removeMongoClient();
	}
	
	
	// get,set
	public void setUrl(String url) {
		this.url = url;
	}

	public String getUsername() {
		return username;
	}

	public void setUsername(String username) {
		this.username = username;
	}

	public String getPassword() {
		return password;
	}

	public void setPassword(String password) {
		this.password = password;
	}

}