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;
    }
}