cn.m1c.frame.mongodb.MongoDbUtil Maven / Gradle / Ivy

Go to download
package cn.m1c.frame.mongodb;

import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;

import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoOperations;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;

import cn.m1c.frame.page.PageBaseModel;
import cn.m1c.frame.utils.CollectionUtil;
import cn.m1c.frame.utils.StringUtil;


/**
 * 2016年7月27日  mongoDB工具类
 * @author  phil([email protected],m1c softCo.,ltd)
 * @version lannie
 */
public class MongoDbUtil {
	
	/**
	 * 更新字段，hashmap转update
	 */
	public static Update toUpdate( Map values ){
		Update update = new Update();
		for(Map.Entry value : values.entrySet()){
			update.set(value.getKey(), value.getValue());
		}
		return update;
	}
	
	/**
	 * 绑定分页
	 */
	public static void bindPaging(Class entityName, MongoOperations mongoOperations, Query query, PageBaseModel page){
		if(page == null){
			return;
		}
		if(page.getTotalCount() == 0){
			page.setTotalCount((int)mongoOperations.count(query, entityName));
		}
		query.skip(page.getStartNum()).limit(page.getPageSize());
	}
	
	/**
	 * 绑定排序
	 */
	public static void bindOrders(Query query, OrderBean orderBean){
		if(orderBean == null){
			return;
		}
		LinkedHashMap ov = orderBean.getOrderValue();
		for(Map.Entry o : ov.entrySet()){
			query.with(new Sort(o.getValue() ? Sort.Direction.ASC : Sort.Direction.DESC, o.getKey()));
		}
	}
	
	/**
	 * 定制字段
	 */
	public static void bindFields(Query query, String fields){
		if(StringUtil.hasLengthBytrim(fields)){
			String[] fs = fields.split("\\*");
			for(String f : fs){
				query.fields().include(f);
			}
		}
	}
	
	public static void bindFields(Query query, Set fields){
		if(CollectionUtil.isEmpty(fields) || "*".equals(fields.iterator().next())){
			return;
		}
		
		for(String f : fields){
			query.fields().include(f);
		}
	}

}