MongoDB自增长主键，批量操作，distinct记录，条件组合查询

一，创建自增长主键

在 mongodb 的命令行终端，创建一个 collection 比如 sequence，然后插入一条记录：

MongoDB Enterprise > db.sequence.insert({_id:"userseq",seq:0})
WriteResult({ "nInserted" : 1 })

创建一个Service，调用 mongodb 的一个 API，修改这个 sequence 里面的对象，并返回更改后的值作为程序即将保存的对象的 ID。

@Service
public class SequenceService {
	private Logger logger = LoggerFactory.getLogger(SequenceService.class);
	
	private static final String SEQ_COLLECTION = "sequence";
	private static final String PRIMARY_KEY = "_id";
	private static final String SEQUENCE_FIELD = "seq";
	
	@Autowired
	private MongoTemplate temp;
	
	public Integer getNextUserSeq(){
		Integer seq = getNextSequence("userseq");
		logger.info("getNextUserSeq() 的下一个值 ：：： " + seq);
		return seq;
	}
	
	private Integer getNextSequence(String name) {
		DBObject ret = temp.getCollection(SEQ_COLLECTION).findAndModify(
				Query.query(Criteria.where(PRIMARY_KEY).is(name)).getQueryObject(), null, null, false,
				new Update().inc(SEQUENCE_FIELD, 1).getUpdateObject(), true, false);
		return ((Double) ret.get(SEQUENCE_FIELD)).intValue();
	}
}

注意， findAndModify 这个方法是重载的，如果 public DBObject findAndModify(final DBObject query, final DBObject update) 这个接口，将会返回修改前的值，比如值已经被这个接口从 4 改到 5 了，但是返回的却是 4，不是最新的 5。所以我程序里使用的是更多参数的接口，其倒是第二个参数设为 true, 意思是返回修改后的值。

还有一个就是这个返回值的类型是 Double 型的，最终返回前做一下类型转换。

最后，程序在保存对象的时候，调用 SequenceService.getNextUserSeq() 返回的值作为 ID 的值。

public void createUser(User user){
	encryptPassword(user);
	user.setId(seqService.getNextUserSeq());
	userDao.insert(user);
}

二，使用 BulkOperations 类可以批量增删改对象。

	public int enableUsers(List<String> userIds){
		BulkOperations bulkOps = temp.bulkOps(BulkMode.ORDERED, User.class);
		bulkOps.updateMulti(Query.query(Criteria.where("id").in(userIds)), Update.update("enabled", "1"));
		bulkOps.execute();
		return 1;
	}

三， distinct 获取记录的唯一字段数据

public List<String> getLoginNames(String mobile) {
	return (List<String>)temp.getCollection("user")
			.distinct("loginName", Query.query(Criteria.where("mobile").is(mobile)).getQueryObject());
}

但是如果希望通过 id 作为 Query 里的查询字段，那么注意了，不是我们定义的实体类比如这里 User 里的 id 字段，而应该是 MongoDB 保存 Collection 里面的的字段名，也就是 “_id”, 也就是说下面的语句才是可工作的：

public List<String> getLoginNames(Integer id) {
	return (List<String>)temp.getCollection("user").distinct("loginName", 
	Query.query(Criteria.where("_id").is(id)).getQueryObject());
}

四，条件组合查询，比如我们的查询是： A and (B or C), 可以使用 Criteria 的 andOperator / orOperator 接口实现：

public List<User> getUsersByNameLike(String keyword) {
	Query query = new Query();
	query.addCriteria(Criteria.where("loginName").ne("admin"))
		 .addCriteria(new Criteria()
				 .orOperator(Criteria.where("loginName").regex(keyword), 
							 Criteria.where("fullName").regex(keyword)));
	
	List<User> users = temp.find(query, User.class);
	return users;
}

MongoDB自增长主键，批量操作，distinct记录，条件组合查询

猜你喜欢