com.cybermkd.mongo.kit.MongoKit Maven / Gradle / Ivy
The newest version!
package com.cybermkd.mongo.kit;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.mongodb.Block;
import com.mongodb.DBRef;
import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.IndexModel;
import com.mongodb.client.model.InsertManyOptions;
import com.mongodb.client.result.DeleteResult;
import com.mongodb.client.result.UpdateResult;
import org.bson.BsonDocument;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.validation.ConstraintViolation;
import javax.validation.Validation;
import javax.validation.Validator;
import java.beans.BeanInfo;
import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.lang.reflect.Method;
import java.util.*;
/**
* 创建人:T-baby
* 创建日期: 16/4/15
* 文件描述:MongoDB操作工具类
*/
public enum MongoKit {
/*
*枚举实现单例模式
*/
INSTANCE;
private static MongoClient client;
private static MongoDatabase defaultDb;
private static Logger logger = LoggerFactory.getLogger(MongoKit.class.getName());
public MongoClient getClient() {
return client;
}
public void init(MongoClient client, String database) {
MongoKit.client = client;
MongoKit.defaultDb = client.getDatabase(database);
}
public MongoCollection getCollection(String collectionName) {
return defaultDb.getCollection(collectionName);
}
public void insert(String collectionName, List docs, InsertManyOptions ops) {
getCollection(collectionName).insertMany(uniding(docs), ops);
}
public void insert(String collectionName, Document doc) {
getCollection(collectionName).insertOne(uniding(doc));
}
public List aggregate(String collectionName, List query, boolean allowDiskUse) {
final List list = new ArrayList();
Block block = new Block() {
public void apply(Document document) {
document = iding(document);
list.add(parseObject(document.toJson()));
}
};
getCollection(collectionName).aggregate(query).allowDiskUse(allowDiskUse).forEach(block);
return list;
}
public List aggregate(String collectionName, List query, boolean allowDiskUse, final Class clazz) {
final List list = new ArrayList();
Block block = new Block() {
public void apply(Document document) {
document = iding(document);
list.add(parseObject(document, clazz));
}
};
getCollection(collectionName).aggregate(query).allowDiskUse(allowDiskUse).forEach(block);
return list;
}
public List find(String collectionName, Bson projection) {
return find(collectionName, new BsonDocument(), new BsonDocument(), projection, 0, 0, "");
}
public List find(String collectionName, int limit, Bson sort, Bson projection) {
return find(collectionName, new BsonDocument(), sort, projection, limit, 0, "");
}
public List find(String collectionName, int limit, int skip, Bson sort, Bson projection, String join) {
return find(collectionName, new BsonDocument(), sort, projection, limit, skip, join);
}
public List find(String collectionName, int limit, Bson sort, Bson projection, Class clazz) {
return find(collectionName, new BsonDocument(), sort, projection, limit, 0, "", clazz);
}
public List find(String collectionName, int limit, int skip, Bson sort, Bson projection, String join, Class clazz) {
return find(collectionName, new BsonDocument(), sort, projection, limit, skip, join, clazz);
}
public List find(String collectionName, Bson query, Bson projection) {
return find(collectionName, query, new BsonDocument(), projection, 0, 0, "");
}
public long count(String collectionName, Bson query) {
return getCollection(collectionName).count(query);
}
public long count(String collectionName) {
return getCollection(collectionName).count();
}
public JSONObject findOne(String collectionName, Bson query, Bson sort, String join) {
return toJSON(
iding(jointing(getCollection(collectionName).find(query).sort(sort).first(), join))
);
}
public T findOne(String collectionName, Bson query, Bson sort, String join, Class clazz) {
return parseObject(
iding(jointing(getCollection(collectionName).find(query).sort(sort).first(), join))
, clazz);
}
public List find(String collectionName, Bson query, Bson sort, Bson projection, int limit,
int skip, final String join) {
final List list = new ArrayList();
Block block = new Block() {
public void apply(Document document) {
document = iding(document);
document = jointing(document, join);
list.add(toJSON(document));
}
};
getCollection(collectionName).find(query).projection(projection).sort(sort).limit(limit).skip(skip).forEach(block);
return list;
}
public List find(String collectionName, Bson query, Bson sort, Bson projection, int limit, int skip,
final String join, final Class clazz) {
final List list = new ArrayList();
Block block = new Block() {
public void apply(Document document) {
document = iding(document);
document = jointing(document, join);
list.add(parseObject(document, clazz));
}
};
getCollection(collectionName).find(query).projection(projection).sort(sort).limit(limit).skip(skip).forEach(block);
return list;
}
public long update(String collectionName, Bson queue, Bson data) {
UpdateResult updateResult = getCollection(collectionName).updateMany(queue, data);
return updateResult.getModifiedCount();
}
public long updateOne(String collectionName, Bson queue, Bson data) {
UpdateResult updateResult = getCollection(collectionName).updateOne(queue, data);
return updateResult.getModifiedCount();
}
public long replaceOne(String collectionName, Bson queue, Document document) {
UpdateResult updateResult = getCollection(collectionName).replaceOne(queue, document);
return updateResult.getModifiedCount();
}
public long delete(String collectionName, Bson queue) {
DeleteResult deleteResult = getCollection(collectionName).deleteMany(queue);
return deleteResult.getDeletedCount();
}
public long deleteOne(String collectionName, Bson queue) {
DeleteResult deleteResult = getCollection(collectionName).deleteOne(queue);
return deleteResult.getDeletedCount();
}
public String validation(Object obj) {
StringBuffer buffer = new StringBuffer(64);//用于存储验证后的错误信息
Validator validator = Validation.buildDefaultValidatorFactory()
.getValidator();
Set> constraintViolations = validator
.validate(obj);//验证某个对象,其实也可以只验证其中的某一个属性的
Iterator iter = constraintViolations.iterator();
while (iter.hasNext()) {
ConstraintViolation c = (ConstraintViolation) iter.next();
buffer.append(c.getMessage());
}
return buffer.toString();
}
//校验单个属性
public String validation(Object obj, String[] keys) {
StringBuffer buffer = new StringBuffer(64);//用于存储验证后的错误信息
Validator validator = Validation.buildDefaultValidatorFactory()
.getValidator();
Set> constraintViolations = new HashSet>();
for (String key : keys) {
Iterator> it = validator.validateProperty(obj, key).iterator();
if (it.hasNext()) {
constraintViolations.add(it.next());
}
}
Iterator iter = constraintViolations.iterator();
while (iter.hasNext()) {
ConstraintViolation c = (ConstraintViolation) iter.next();
buffer.append(c.getMessage());
}
return buffer.toString();
}
public String setIndex(String collectionName, Bson bson) {
return getCollection(collectionName).createIndex(bson);
}
public List setIndex(String collectionName, List list) {
return getCollection(collectionName).createIndexes(list);
}
public List getIndex(String collectionName) {
final List list = new ArrayList();
Block block = new Block() {
public void apply(final Document document) {
list.add(parseObject(document.toJson()));
}
};
getCollection(collectionName).listIndexes().forEach(block);
return list;
}
public void deleteIndex(String collectionName, Bson bson) {
getCollection(collectionName).dropIndex(bson);
}
public void deleteIndex(String collectionName) {
getCollection(collectionName).dropIndexes();
}
private Document iding(Document document) {
try {
if (document == null || document.get("_id") == null) {
return document;
} else {
document.put("_id", document.get("_id").toString());
}
} catch (ClassCastException e) {
/*如果转换出错直接返回原本的值,不做任何处理*/
}
return document;
}
private List uniding(List list) {
List newList = new ArrayList();
for (Document doc : list) {
newList.add(uniding(doc));
}
return newList;
}
private Document uniding(Document document) {
try {
if (document == null || document.get("_id") == null) {
return document;
} else {
document.remove("_id");
}
} catch (ClassCastException e) {
/*如果转换出错直接返回原本的值,不做任何处理*/
}
return document;
}
private Document jointing(Document document, String join) {
if (join != null && !join.isEmpty()) {
try {
DBRef dbRef = document.get(join, DBRef.class);
Document joinDoc = getCollection(dbRef.getCollectionName())
.find(new Document("_id", dbRef.getId())).first();
joinDoc = iding(joinDoc);
joinDoc.put("id",joinDoc.getString("_id"));
joinDoc.remove("_id");
document.put(join, joinDoc);
} catch (ClassCastException e) {
/*用于避免如果key对应的值并不是DBRef,如果转换出错直接返回原本的值,不做任何处理*/
}
}
return document;
}
/*由于fastjson转换空对象时就会直接抛出异常,而在实际查询中查不到东西是很正常的
* ,所以为了避免会有空异常,特别做了异常处理*/
private JSONObject parseObject(String json) {
try {
if (json != null && !json.isEmpty()) {
return JSON.parseObject(json);
}
return new JSONObject();
} catch (NullPointerException e) {
error("parseObject", json);
return new JSONObject();
}
}
private T parseObject(Document doc, Class clazz) {
try {
if (doc == null) {
return JSON.parseObject(new JSONObject().toJSONString(), clazz);
}
return JSON.parseObject(JSON.toJSONString(doc), clazz);
} catch (NullPointerException e) {
error("parseObject", clazz.getName());
return JSON.parseObject(new JSONObject().toJSONString(), clazz);
}
}
private JSONObject toJSON(Object obj) {
try {
return (JSONObject) JSON.toJSON(obj);
} catch (NullPointerException e) {
error("toJSON", obj.getClass().getName());
return new JSONObject();
}
}
protected void error(String funName, String text) {
logger.error("MongKit tips: (づ￣ 3￣)づ " + funName + " is error ! " + text);
}
public Map toMap(Object obj) {
if(obj == null){
return null;
}
Map map = new HashMap();
try {
BeanInfo beanInfo = Introspector.getBeanInfo(obj.getClass());
PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
for (PropertyDescriptor property : propertyDescriptors) {
String key = property.getName();
/*过滤class属性 */
if (!key.equals("class")) {
//*得到property对应的getter方法*/
Method getter = property.getReadMethod();
Object value = getter.invoke(obj);
map.put(key, value);
}
}
} catch (Exception e) {
MongoKit.INSTANCE.error("MongKit.class","toMap is error " + e.getMessage());
}
return map;
}
public void setDebug(boolean debug) {
ch.qos.logback.classic.Level level;
if (debug) {
level = ch.qos.logback.classic.Level.DEBUG;
} else {
level = ch.qos.logback.classic.Level.WARN;
}
ch.qos.logback.classic.Logger root = (ch.qos.logback.classic.Logger) LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME);
root.setLevel(level);
}
}