java 使用最新api操作mongodb

Posted 2020-08-09

//

package com.auto.test.dbmodel;

import java.util.ArrayList;

import org.bson.Document;
import org.bson.conversions.Bson;

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

/**
* 类说明
*
* @author luozt
* @date 2016年9月19日 新建
*/
public class DBConnUtil {

public String ip;
public int port;
public String dbname;
MongoClient mongoClient;
MongoDatabase db;
FindIterable<Document> cursor=null;
long update_count=0;
long delete_count=0;


public DBConnUtil(String ip,int port,String dbname){

this.ip=ip;
this.port=port;
this.dbname=dbname;

}


public MongoDatabase conn()
{
mongoClient = new MongoClient(ip , port );
db = mongoClient.getDatabase(dbname);

return db;

}

/**query data from database
*
* @param collection_name
* @param document 为查询条件
* @param document_projection为只返回需要的字段
* @return
* crud method please see http://mongodb.github.io/mongo-java-driver/3.0/driver/getting-started/quick-tour/
*/

@SuppressWarnings("unchecked")
public FindIterable<Document> queryData(String collection_name,Bson document,int flag){
//collection_name为要查询的collection名字
MongoCollection<Document> coll = db.getCollection(collection_name);

if(flag==0){
//0 代表查询所有
cursor=coll.find(document);
}
else if(flag==1){
//1 代表只查询一条数据
cursor=(FindIterable<Document>) coll.find(document).first();

}

else if(flag==2){
//2 代表没有查询条件
cursor=coll.find();

}
return cursor;
}

@SuppressWarnings("unchecked")
public FindIterable<Document> queryData(String collection_name,Bson document,Bson document_projection,int flag){

MongoCollection<Document> coll = db.getCollection(collection_name);
if(flag==0){
//0 代表查询所有
cursor=coll.find(document).projection(document_projection);
}

else if(flag==1){
//1 代表只查询一条数据
cursor=(FindIterable<Document>) coll.find(document).projection(document_projection).first();
}

else if(flag==2){
//2 代表没有查询条件
cursor=coll.find().projection(document_projection);
}
return cursor;
}



/**insert data to database;
*
* crud method please see http://mongodb.github.io/mongo-java-driver/3.0/driver/getting-started/quick-tour/
*/
public void insertData(String collection_name,ArrayList<Document> documents){


MongoCollection<Document> coll = db.getCollection(collection_name);

coll.insertMany(documents);


}


/**
*
* update data to database
*
* filter 通过此filter过滤需要更新的数据
* update 需要更新的内容
*
*
*
* crud method please see http://mongodb.github.io/mongo-java-driver/3.0/driver/getting-started/quick-tour/
*/


public long updateData(String collection_name,Bson filter,Bson update,int flag){


MongoCollection<Document> coll = db.getCollection(collection_name);
if(flag==0){
update_count=coll.updateMany(filter, update).getModifiedCount();
// return update_count;
}

else if(flag==1){

update_count=coll.updateOne(filter, update).getModifiedCount();
// return update_count;
}
return update_count;
}




/**
*
* remove data from database
*
* filter 通过此filter去过滤需要删除的数据
* crud method please see http://mongodb.github.io/mongo-java-driver/3.0/driver/getting-started/quick-tour/
*/

public long deleteData(String collection_name,Bson filter,int flag){

MongoCollection<Document> coll = db.getCollection(collection_name);
if(flag==0){

delete_count=coll.deleteMany(filter).getDeletedCount();
// return delete_count;
}
else if(flag==1){

delete_count=coll.deleteOne(filter).getDeletedCount();
// return delete_count;

}

return delete_count;
}

}

 

 

//dbunit

package com.auto.test.testcase;

import java.util.ArrayList;

import org.bson.Document;
import com.auto.test.basedata.DBConfig;
import com.auto.test.dbmodel.DBConnUtil;
import com.mongodb.Block;
import com.mongodb.client.FindIterable;

/**
* 类说明
*
* @author luozt
* @date 2016年9月19日 新建
*/
public class dbUnit_test implements DBConfig{

String collection_name="test_api";

public static void main(String[] args) {
// TODO Auto-generated method stub
dbUnit_test test=new dbUnit_test();
}

public dbUnit_test(){

DBConnUtil dbhelp=new DBConnUtil(mongo_ip,mongo_port,mongo_dbname);
dbhelp.conn();


/**
* insert
* if database or collection not exist ,it will be created by Automatic
*/

ArrayList<Document> documents=new ArrayList<Document>();

Document document=new Document("name","mongodb2")
.append("type", "database2")
.append("count", "112")
.append("info", new Document("x",2032).append("y", 1022));

documents.add(document);

dbhelp.insertData(collection_name, documents);



/**
* query
*
*/
Document query_document=new Document("info.x",new Document("$gt",300));
Document fiter_document=new Document("name",1).append("count", 1);

FindIterable<Document> iterable=dbhelp.queryData(collection_name, query_document, fiter_document, 0);
iterable.forEach(new Block<Document>() {
@Override
public void apply(final Document document) {
System.out.println(document);
}
});

/**
* update
*
*/

Document update_document=new Document("name","mongodb2");
Document update_option=new Document("$set",new Document("info.y",2000));
long update_count=dbhelp.updateData(collection_name, update_document, update_option, 0);
System.out.println(update_count);



/**
*
* delete
*
*
*/
Document delete_document=new Document("type","database3");

long delete_count=dbhelp.deleteData(collection_name, delete_document, 0);
System.out.println("delete count is "+delete_count+"");

}




}