MongoDB入门（5）- 我们自己封装的MongoDB-Java版本
用法
实体定义
package com.wisdombud.mongotool;
import java.io.Serializable;
import java.util.Date;
import java.util.List;
import org.springframework.data.mongodb.core.mapping.Document;
@Document(collection = "Project")
public class Project extends MongoDbEntity implements Serializable {
/**
*
*/
private static final long serialVersionUID = 1L;
private String ProjectId;
private String weituodanwei;
private String weituodanweidizhi;
private String post;
private String yangpinzhizaodanwei;
private String zhizaodanweidizhi;
private String phone;
private String contact;
private String yangpinjiazhi;
public String getProjectId() {
return ProjectId;
}
public void setProjectId(String projectId) {
ProjectId = projectId;
}
public String getWeituodanwei() {
return weituodanwei;
}
public void setWeituodanwei(String weituodanwei) {
this.weituodanwei = weituodanwei;
}
public String getWeituodanweidizhi() {
return weituodanweidizhi;
}
public void setWeituodanweidizhi(String weituodanweidizhi) {
this.weituodanweidizhi = weituodanweidizhi;
}
public String getPost() {
return post;
}
public void setPost(String post) {
this.post = post;
}
public String getYangpinzhizaodanwei() {
return yangpinzhizaodanwei;
}
public void setYangpinzhizaodanwei(String yangpinzhizaodanwei) {
this.yangpinzhizaodanwei = yangpinzhizaodanwei;
}
public String getZhizaodanweidizhi() {
return zhizaodanweidizhi;
}
public void setZhizaodanweidizhi(String zhizaodanweidizhi) {
this.zhizaodanweidizhi = zhizaodanweidizhi;
}
public String getPhone() {
return phone;
}
public void setPhone(String phone) {
this.phone = phone;
}
public String getContact() {
return contact;
}
public void setContact(String contact) {
this.contact = contact;
}
public String getYangpinjiazhi() {
return yangpinjiazhi;
}
public void setYangpinjiazhi(String yangpinjiazhi) {
this.yangpinjiazhi = yangpinjiazhi;
}
public String getShenqingleixing() {
return shenqingleixing;
}
public void setShenqingleixing(String shenqingleixing) {
this.shenqingleixing = shenqingleixing;
}
public String getYangpinmingcheng() {
return yangpinmingcheng;
}
public void setYangpinmingcheng(String yangpinmingcheng) {
this.yangpinmingcheng = yangpinmingcheng;
}
public String getXinghao() {
return xinghao;
}
public void setXinghao(String xinghao) {
this.xinghao = xinghao;
}
public String getNumber() {
return number;
}
public void setNumber(String number) {
this.number = number;
}
public String getJianceyiju() {
return jianceyiju;
}
public void setJianceyiju(String jianceyiju) {
this.jianceyiju = jianceyiju;
}
public List<String> getJiancexiangmu() {
return jiancexiangmu;
}
public void setJiancexiangmu(List<String> jiancexiangmu) {
this.jiancexiangmu = jiancexiangmu;
}
public String getYangpinzhuangkuang() {
return yangpinzhuangkuang;
}
public void setYangpinzhuangkuang(String yangpinzhuangkuang) {
this.yangpinzhuangkuang = yangpinzhuangkuang;
}
public String getYangpinxuhao() {
return yangpinxuhao;
}
public void setYangpinxuhao(String yangpinxuhao) {
this.yangpinxuhao = yangpinxuhao;
}
public String getYangpinfujian() {
return yangpinfujian;
}
public void setYangpinfujian(String yangpinfujian) {
this.yangpinfujian = yangpinfujian;
}
public String getYangpintijiaofangshi() {
return yangpintijiaofangshi;
}
public void setYangpintijiaofangshi(String yangpintijiaofangshi) {
this.yangpintijiaofangshi = yangpintijiaofangshi;
}
public String getYangpinbaogaotijiaofangshi() {
return yangpinbaogaotijiaofangshi;
}
public void setYangpinbaogaotijiaofangshi(String yangpinbaogaotijiaofangshi) {
this.yangpinbaogaotijiaofangshi = yangpinbaogaotijiaofangshi;
}
public String getYangpinchuzhifangshi() {
return yangpinchuzhifangshi;
}
public void setYangpinchuzhifangshi(String yangpinchuzhifangshi) {
this.yangpinchuzhifangshi = yangpinchuzhifangshi;
}
public Date getCreateDay() {
return CreateDay;
}
public void setCreateDay(Date createDay) {
CreateDay = createDay;
}
private String shenqingleixing;
private String yangpinmingcheng;
private String xinghao;
private String number;
private String jianceyiju;
private List<String> jiancexiangmu;
private String yangpinzhuangkuang;
private String yangpinxuhao;
private String yangpinfujian;
private String yangpintijiaofangshi;
private String yangpinbaogaotijiaofangshi;
private String yangpinchuzhifangshi;
private Date CreateDay;
}
DAO实现
package com.wisdombud.mongotool;
import org.springframework.data.mongodb.core.MongoOperations;
public class ProjectDaoImpl extends SimpleMongoDaoImpl<Project, String> implements ProjectDao {
public ProjectDaoImpl(MongoOperations mongoOper) {
super(mongoOper);
}
}
核心源码
package com.wisdombud.mongotool;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.lang.ArrayUtils;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.domain.Sort.Order;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import common.toolkit.java.entity.PageEntity;
import common.toolkit.java.orm.hibernate.GeneralQuery;
import common.toolkit.java.util.ReflectUtils;
public class SimpleMongoDaoImpl<T, PK extends Serializable> implements SimpleMongoDao<T, PK> {
@Autowired
protected MongoOperations mongoOper;
protected Class<T> entityClazz;
/**
* 字符串分隔的标志符号，现在用:表示，在关联查询的时候用。
***/
private final String splitMarker = "::";
public SimpleMongoDaoImpl(MongoOperations mongoOper) {
this.mongoOper = mongoOper;
this.entityClazz = ReflectUtils.getSuperClassGenricType(getClass());
}
@Override
public PK save(T entity) {
return null;
}
@Override
public void update(T entity) {
mongoOper.save(entity);
}
@Override
public void saveOrUpdate(T entity) {
mongoOper.save(entity);
}
@Override
public void saveOrUpdate(Collection<T> entities) {
for (T entity : entities) {
mongoOper.save(entity);
}
}
@Override
public void delete(T entity) {
mongoOper.remove(entity);
}
@Override
public void deleteAll(Collection<T> entities) {
for (T entity : entities) {
mongoOper.remove(entity);
}
}
@Override
public void deleteById(PK id) {
mongoOper.remove(new Query(Criteria.where("_id").is(id)), entityClazz);
}
@Override
public T get(PK id) {
return mongoOper.findById(id, entityClazz);
}
@Override
public List<T> getAll() {
return this.mongoOper.findAll(entityClazz);
}
@Override
public T load(PK id) {
// TODO Auto-generated method stub
return null;
}
@Override
public void refresh(T entity) {
// TODO Auto-generated method stub
}
@Override
public void merge(T entity) {
// TODO Auto-generated method stub
}
@Override
public void merge(String entityName, T entity) {
// TODO Auto-generated method stub
}
@Override
public void evict(T entity) {
// TODO Auto-generated method stub
}
@Override
public boolean contains(Object entity) {
// TODO Auto-generated method stub
return false;
}
@Override
public List<T> findBy(String propertyName, Object value) {
HashMap<String, Object> paramMap = new HashMap<String, Object>();
paramMap.put(propertyName, value);
return this.findBy(paramMap, null);
}
@Override
public List<T> findBy(Map<String, Object> paramMap, List<Order> orderList) {
Query query = this.buildQuery(paramMap, orderList);
return mongoOper.find(query, entityClazz);
}
@Override
public T findUniqueBy(String propertyName, Object value) {
HashMap<String, Object> paramMap = new HashMap<String, Object>();
paramMap.put(propertyName, value);
Query query = this.buildQuery(paramMap, null);
return mongoOper.findOne(query, entityClazz);
}
@Override
public T findUniqueBy(Map<String, Object> paramMap, List<Order> orderList) {
Query query = this.buildQuery(paramMap, orderList);
return mongoOper.findOne(query, entityClazz);
}
@Override
public void pageList(Map<String, Object> paramMap, List<Order> orderList, PageEntity<T> pageEntity) {
Query query = this.buildQuery(paramMap, orderList);
pageEntity.setTotalResults(mongoOper.count(query, entityClazz));// 总记录数
pageEntity.setTotalPages();// 总页数
query.skip(pageEntity.getStart());
query.limit(pageEntity.getPageSize());
pageEntity.setResults(mongoOper.find(query, entityClazz));// 分页数据
pageEntity.setCurrentPageLength(pageEntity.getResults().size());// 当前页面记录数
}
public Class<T> getEntityClazz() {
return entityClazz;
}
public void setEntityClazz(Class<T> entityClazz) {
this.entityClazz = entityClazz;
}
protected void buildQuery(String key, Object value, Query query) {
// 如果为数字
if (key.contains("_id")) {
query.addCriteria(Criteria.where(key).is(new ObjectId(value.toString())));
} else {
if (value instanceof Integer || value instanceof Float || value instanceof Double
|| value instanceof Boolean) {
query.addCriteria(Criteria.where(key).is(value));
} else {
// 字符串 模糊查询
// i为查询大小写不敏感
query.addCriteria(Criteria.where(key).regex(value.toString(), "i"));
}
}
}
protected Query buildQuery(Map<String, Object> paramMap, List<Order> orderList) {
Query query = new Query();
if (paramMap != null && paramMap.size() > 0) {
for (String key : paramMap.keySet()) {
Object value = paramMap.get(key);
if (key.contains(this.splitMarker)) {// 处理关联查询的情况
String[] arr = key.split(this.splitMarker);
if (arr.length != 2) {
continue;
}
String p1 = arr[0];
String p2 = arr[1];
Class<T> clazz = ReflectUtils.getSuperClassGenricType(getClass());
Field[] fs = this.getAllFields(clazz);
// ReflectUtils.getAllFields(clazz);
Class<?> clazz1 = null;
for (Field f : fs) {
if (f.getName().equals(p1)) {
clazz1 = f.getType();
break;
}
}
Query query0 = new Query();
this.buildQuery(p2, value, query0);
query.addCriteria(Criteria.where(p1).in(mongoOper.find(query0, clazz1)));
} else {
this.buildQuery(key, value, query);
}
}
}
if (orderList != null && orderList.size() > 0) {
Sort sort = new Sort(orderList);
query.with(sort);
}
return query;
}
private Field[] getAllFields(Class<? super T> clazz) {
Class<?> originClazz = clazz;
Field[] resultField = clazz.getDeclaredFields();
for (; clazz != Object.class; clazz = clazz.getSuperclass()) {
if (clazz == originClazz) {
continue;
}
try {
Field[] superFields = clazz.getDeclaredFields();
resultField = (Field[]) ArrayUtils.addAll(resultField, superFields);
} catch (Exception e) {
System.out.println("error");
}
}
return resultField;
}
@Override
public List<T> findBy(Map<String, Object> paramsMap) {
// TODO Auto-generated method stub
return null;
}
@Override
public GeneralQuery<T> createGeneralQuery() {
// TODO Auto-generated method stub
return null;
}
}
