您的位置：首页 > 数据库 > Mongodb

MongoDB操作类封装实例代码

2019-08-08 09:32 746 查看

前言

最近接到一个需求，要做MongoDB打点数据的统计，在学习过MongoDB的操作之后，封装了一个MongoDB的操作类，分为两部分，基本思想是参照了自己写过的mysql的操作类。一个是基本的操作类，包括所有基本操作的静态方法，还有一个是mongoobject，就是具体操作的实现类。

以后再写如何用spring boot写一个简单的统计服务。

MongoDB操作类封装

mongobase代码如下：

package com.fun.mongodb;

import com.fun.frame.SourceCode;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientURI;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;

import java.util.Arrays;
import java.util.List;

/**
* mongo操作类的基础类
*/
public class MongoBase extends SourceCode {

/**
* 获取服务地址list
*
* @param addresses
* @return
*/
public static List<ServerAddress> getServers(ServerAddress... addresses) {
return Arrays.asList(addresses);
}

/**
* 获取服务地址
*
* @param host
* @param port
* @return
*/
public static ServerAddress getServerAdress(String host, int port) {
return new ServerAddress(host, port);
}

/**
* 获取认证list
*
* @param credentials
* @return
*/
public static List<MongoCredential> getCredentials(MongoCredential... credentials) {
return Arrays.asList(credentials);
}

/**
* 获取验证
*
* @param userName
* @param database
* @param password
* @return
*/
public static MongoCredential getMongoCredential(String userName, String database, String password) {
return MongoCredential.createCredential(userName, database, password.toCharArray());
}

/**
* 获取mongo客户端
*
* @param addresses
* @param credentials
* @return
*/
public static MongoClient getMongoClient(List<ServerAddress> addresses, List<MongoCredential> credentials) {
return new MongoClient(addresses, credentials);
}

/**
* 连接mongo数据库
*
* @param mongoClient
* @param databaseName
* @return
*/
public static MongoDatabase getMongoDatabase(MongoClient mongoClient, String databaseName) {
return mongoClient.getDatabase(databaseName);
}

/**
* 连接mongo集
*
* @param mongoDatabase
* @param collectionName
* @return
*/
public static MongoCollection<Document> getMongoCollection(MongoDatabase mongoDatabase, String collectionName) {
return mongoDatabase.getCollection(collectionName);
}

/**
* 关闭数据库连接
*
* @param mongoClient
*/
public static void MongoOver(MongoClient mongoClient) {
mongoClient.close();
}

/**
* 获取mongo客户端对象，通过servers和credentials对象创建
*
* @param mongoObject
* @return
*/
public static MongoClient getMongoClient(MongoObject mongoObject) {
MongoClient mongoClient = new MongoClient(getServers(getServerAdress(mongoObject.host, mongoObject.port)), getCredentials(getMongoCredential(mongoObject.user, mongoObject.database, mongoObject.password)));
return mongoClient;
}

/**
* 获取mongo客户端对象,通过uri方式连接
*
* @param mongoObject
* @return
*/
public static MongoClient getMongoClientOnline(MongoObject mongoObject) {
String format = String.format("mongodb://%s:%s@%s:%d/%s", mongoObject.user, mongoObject.password, mongoObject.host, mongoObject.port, mongoObject.database);
return new MongoClient(new MongoClientURI(format));
}

/**
* 获取collection对象
*
* @param mongoObject
* @return
*/
public static MongoCollection<Document> getCollection(MongoObject mongoObject, String collectionName) {
return getMongoClient(mongoObject).getDatabase(mongoObject.database).getCollection(collectionName);
}

/**
* 获取collection对象
*
* @param mongoObject
* @return
*/
public static MongoCollection<Document> getCollectionOnline(MongoObject mongoObject, String collectionName) {
return getMongoClientOnline(mongoObject).getDatabase(mongoObject.database).getCollection(collectionName);
}

}

mongoobject的代码如下：

package com.fun.mongodb;

import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import org.bson.Document;

/**
* mongo数据库配置对象，针对单个数据服务，单个身份验证
*/
public class MongoObject extends MongoBase {

String host;

int port;

String user;

String password;

String database;

MongoClient mongoClient;

/**
* 创建数据连接
*
* @param host
* @param port
* @param user
* @param password
* @param database
*/
public MongoObject(String host, int port, String user, String password, String database) {
this.host = host;
this.port = port;
this.user = user;
this.password = password;
this.database = database;
this.mongoClient = getMongoClient(this);
}

/**
* 创建数据库连接
*
* @param port
* @param host
* @param user
* @param password
* @param database
*/
public MongoObject(int port, String host, String user, String password, String database) {
this.host = host;
this.port = port;
this.user = user;
this.password = password;
this.database = database;
this.mongoClient = getMongoClientOnline(this);
}

/**
* 获取colletion对象
*
* @param collectionName
* @return
*/
public MongoCollection<Document> getMongoCollection(String collectionName) {
MongoClient mongoClientOnline = getMongoClientOnline(this);
return mongoClientOnline.getDatabase(database).getCollection(collectionName);
}

/**
* 关闭连接
*/
public void over() {
MongoOver(this.mongoClient);
}
}
@Override
public MongoObject clone() {
return new MongoObject(this.host, this.port, this.user, this.password, this.database);
}

public MongoObject clone2() {
return new MongoObject(this.port, this.host, this.user, this.password, this.database);
}

具体效果非常不错，测试代码如下：

public static void main(String[] args) {
MongoObject ready = new MongoObject("*****", 5117, "fission_record", "fission_record", "fission_record");
MongoCollection<Document> app = ready.getMongoCollection("app_logs_20181109");
Document first = app.find().first();
output(first);
ready.over();
}

打印内容如下：

Document{{_id=5be4ce052ce01b21b6c26a64, _class=com.fission.next.record.bean.AppRecordBean, user_id=5482, action_type={"gameId":2,"userId":"5482"}, action_extern=DataSta_Game_Starts, client_version=15, client_ip=114.5.146.239, client_imei=UNKNOWN, client_dev=xiaomi-Redmi 5 Plus, client_type=200, server_time=1541721601655, os_name=200, os_version=15, client_time=1541721577025}}

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，谢谢大家对脚本之家的支持。

您可能感兴趣的文章:

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签： mongodb 操作类封装

相关文章推荐

新的分享

章节导航