package com.qxw.mongodb;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.serializer.SerializeFilter;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.alibaba.fastjson.serializer.ValueFilter;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.result.DeleteResult;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import static com.mongodb.client.model.Aggregates.*;
import static com.mongodb.client.model.Filters.eq;
/**
* mongodb工具类
* @author qinxuewu
*
*/
@Component
public class MongoSdkBase {
@Autowired
private MongoClient mongoClient;
/**
* 表连接
*/
public MongoCollection<Document> getColl(String tableName) {
MongoDatabase db = mongoClient.getDatabase("actuator");
return db.getCollection(tableName);
}
public MongoCollection<Document> getColl(String dbName, String tableName) {
MongoDatabase db = mongoClient.getDatabase(dbName);
return db.getCollection(tableName);
}
/***
* 插入单条记录
* @param table 表连接
* @param doc Document
* @return
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
public String insertOneDocument(MongoCollection table, Document doc) {
if (doc == null){ return null;}
doc.remove("_id");
doc.put("_id", new ObjectId().toString());
table.insertOne(doc);
return doc.get("_id").toString();
}
/***
* 插入单条记录
* @param table 表连接
* @param obj 单条数据
* obj double 处理不规范
* @return
*/
public String insertOne(MongoCollection table, Object obj) {
if (obj == null) {return null;}
Document docine =Document.parse(diyObjectIdToJson(obj));
docine.remove("_id");
docine.put("_id", new ObjectId().toString());
table.insertOne(docine);
return docine.get("_id").toString();
}
/**
* 删除找到的第一条记录
*
* @param table 表连接
* @return
*/
public int deleteOne(MongoCollection table, String id) {
Bson filter = eq("_id", id);
DeleteResult re = table.deleteOne(filter);
return (int) re.getDeletedCount();
}
/**
* 根据条件更新单条记录
*
* @param table 表连接
* @param obj
* @return
*/
@SuppressWarnings("rawtypes")
public boolean updateOne(MongoCollection table, String id, Object obj) {
Bson filter = eq("_id", id);
table.updateOne(filter, set(diyObjectIdToJson(obj)));
return true;
}
/**
* 查询 单条记录 返回json字符串
* @param table 表
* @return
*/
public String seleteOne(MongoCollection table, String id) {
Bson filter = eq("_id", id);
return diyObjectIdToJson(seleteOneDocument(table, filter));
}
/**
* 查询 单条记录 返回json字符串
*
* @param table 表连接
* @return
*/
public String seleteOne(MongoCollection table, Bson filter) {
return diyObjectIdToJson(seleteOneDocument(table, filter));
}
/**
* 查询 单条记录 返回 org.bson.Document 对象
*
* @param table 表连接
* @return
*/
public Document seleteOneDocument(MongoCollection table, Bson filter) {
FindIterable<Document> result = table.find(filter);
return result.first();
}
/**
* 查询所有记录 代码控制返回结果数
*
* @param table 表连接
* @param filter 条件 com.mongodb.client.model.Filter
* @param sort 排序 com.mongodb.client.model.Sorts 可空
* @return
*/
public List<JSONObject> getAll(MongoCollection table, Bson filter, Bson sort) {
List<JSONObject> list = new ArrayList<JSONObject>();
FindIterable<Document> result = null;
if (filter == null) {
result = table.find().sort(sort);
} else {
result = table.find(filter).sort(sort);
}
MongoCursor<Document> iterator = result.iterator();
while (iterator.hasNext()) {
Object ddd = iterator.next();
list.add(JSON.parseObject(diyObjectIdToJson(ddd)));
}
return list;
}
public List<JSONObject> getAllbyCid(MongoCollection table, int cid) {
return getAll(table, eq("cid", cid), null);
}
/**
* 分页查询
* @param table 表连接
* @param filter 条件 com.mongodb.client.model.Filter
* @param sort 排序 com.mongodb.client.model.Sorts
* @param pageNum
* @param pageSize
* @return
*/
public JSONObject getPage(MongoCollection table, Bson filter, Bson sort, int pageNum, int pageSize) {
int totalCount = (int) (filter == null ? table.count(): table.count(filter));
int totalPage = (int) (totalCount / pageSize + ((totalCount % pageSize == 0) ? 0 : 1));
if (pageNum > totalPage){ pageNum = totalPage;}
JSONObject msg = new JSONObject();
msg.put("pageNum", pageNum);
msg.put("pageSize", pageSize);
msg.put("totalCount", totalCount);
msg.put("totalPage", totalPage);
List<JSONObject> list = new ArrayList<JSONObject>();
if (totalCount > 0) {
int startRow = pageNum > 0 ? (pageNum - 1) * pageSize : 0;
FindIterable<Document> result = null;
if (filter == null) {
result = table.find().sort(sort).skip(startRow).limit(pageSize);
} else {
result = table.find(filter).sort(sort).skip(startRow).limit(pageSize);
}
MongoCursor<Document> iterator = result.iterator();
while (iterator.hasNext()) {
Document ddd = (Document) iterator.next();
list.add(JSON.parseObject(diyObjectIdToJson(ddd)));
}
}
msg.put("data", list);
return msg;
}
/**
* 分组分页查询
*
* @param table 表连接
* @param filter 条件 com.mongodb.client.model.Filter
* @param pageNum
* @param pageSize
* @return
*/
public JSONObject getGroupPage(MongoCollection table, Bson filter, Bson group, Bson sorts, int pageNum, int pageSize) {
List<JSONObject> list = new ArrayList<>();
int startRow = pageNum > 0 ? (pageNum - 1) * pageSize : 0;
MongoCursor<Document> iterator = table.aggregate(Arrays.asList(match(filter), group, sort(sorts), skip(startRow), limit(pageSize))).iterator();
while (iterator.hasNext()) {
Document doc = iterator.next();
list.add(JSON.parseObject(diyObjectIdToJson(doc)));
}
int totalCount = list != null && list.size() > 0 ? list.size() : 0;
int totalPage = (int) (totalCount / pageSize + ((totalCount % pageSize == 0) ? 0 : 1));
JSONObject msg = new JSONObject();
msg.put("pageNum", pageNum);
msg.put("pageSize", pageSize);
msg.put("totalCount", totalCount);
msg.put("totalPage", totalPage);
msg.put("data", list);
return msg;
}
/**
* 聚哈函数 统计 ,查询
* @param table
* @param filter
* @param group
* @param sort
* @return
*/
public List<JSONObject> getAggregateList(MongoCollection table, Bson filter, DBObject group, Bson sort){
List<JSONObject> list=new ArrayList<JSONObject>();
MongoCursor<Document> iterator=table.aggregate(Arrays.asList(match(filter),group,sort(sort))).iterator();
while (iterator.hasNext()) {
Document doc = iterator.next();
list.add(JSON.parseObject(diyObjectIdToJson(doc)));
}
return list;
}
/**
* 获取所有数据库名
* @return
*/
public List<String> getDbList() {
return mongoClient.getDatabaseNames();
}
/***
* 获取表想详细内存占用信息
* @return
*/
public BasicDBObject getStats(String name, String collection){
DB db=new DB(mongoClient,name);
return db.getCollection(collection).getStats();
}
public MongoDatabase getMongoDb(String databaseName) {
return mongoClient.getDatabase(databaseName);
}
/**
* 更新数据 注意 多余字段 会在库表记录追加
*
* @param json
* @return
*/
public static Document set(String json) {
Document b =Document.parse(json);
b.remove("_id");
return new Document("$set", b);
}
private static SerializeFilter objectIdSerializer = new ValueFilter() {
@Override
public Object process(Object object, String name, Object value) {
if ("_id".equals(name)) {
if (value instanceof ObjectId) {
return value.toString();
}
}
return value;
}
};
/**
* 出库后查询
* @param object
* @return
*/
public static final String diyObjectIdToJson(Object object) {
return JSON.toJSONString(object, objectIdSerializer,
SerializerFeature.WriteDateUseDateFormat,
SerializerFeature.WriteNullNumberAsZero,
SerializerFeature.WriteMapNullValue,
SerializerFeature.WriteNullStringAsEmpty);
}
/**
* 出库后查询 null跳过不转换为json
*
* @param object
* @return
*/
public static final String diyObjToJson(Object object) {
return JSON.toJSONString(object, objectIdSerializer,
SerializerFeature.WriteDateUseDateFormat);
}
}
