MongoDB操作

重要的事情说三遍：如果你的业务需要复杂的查询语句，MongoDB是非结构型数据库，不要用！不要用！不要用！不然后面坑死你！能用MySQL、Oracle千万不要用它，复杂查询的资料网上很难找到。

一、MongoDB的DML语句

在发布生产云服务器的时候，经常会出现导入脚本报错。以阿里云的MongoDB为例，目前只支持.csv，且导入经常失败，此时需要DML语句。

insert()方法：

下面是在inventory集合中插入一个三个字段的文档：

db.inventory.insert( { _id: 10, type: "misc", item: "card", qty: 15 } )

save()方法:

使用save()方法插入一个文档，通过该方法保存一个不包含_id字段的文档或者包含_id字段但该字段值不存在集合中的文档。下面的示例创建一个新的文档在inventory集合：

db.inventory.save( { type: "book", item: "notebook", qty: 40 } )

当在使用save方法时，插入不指定_id时，会自动生成一个。

Insert和Save的区别是：如果插入的集合的“_id”值，在集合中已经存在,用Insert执行插入操作会报异常，已经存在"_id"的键。用Save如果系统中没有相同的"_id"就执行插入操作，有的话就执行覆盖掉原来的值。相当于修改操作。

update()方法：

调用update()方法使用upsert标志创建一个新文档当没有匹配查询条件的文档时。下面的例子当inventory集合中没有包含{type:"books",item:"journal"}的文档时创建一个新文档：

db.s_city.update(
　　{

　　cityName : "深圳"
　　},
　　{
　　$set: {cityName:"深圳市"}
});

二、Criteria操作MongoDB

基于：org.springframework.data.mongodb.core.query

简单查询

public List<Msg> getUserVideo(int userId) {
    Query query = createQuery("userId",userId);
    addToQuery(query, "body.type", 4);
    return queryListsByQuery(query);
}

查询返回一个字段

 public List<ObjectId> getPraiseIds(Integer userId) {
        Query query = createQuery("userId", userId);
        List<ObjectId> msgIds = getDatastore().findDistinct(query, "msgId", Praise.class, ObjectId.class);
        return msgIds;
    }

 public List<Integer> queryBlacklistUserId(int userId) {
        Query query = createQuery("userId", userId);
        addToQuery(query, "blacklist", 1);
        return distinct("toUserId", query, Integer.class);
    }

in查询

 public long getPraiseCountByMsg(List<String> msgId) {
        Query query = new Query();
        query.addCriteria(Criteria.where("msgId").in(msgId));
        return count(query);
    }

去除

query.addCriteria(Criteria.where("customer").ne(1)); //去掉客服群

or查询

query.addCriteria(new Criteria().orOperator(Criteria.where("manualPay_status").is(2),
        new Criteria().andOperator(Criteria.where("status").is(1)
        )));

  query.addCriteria(
                            new Criteria().orOperator(Criteria.where("nickname").regex(keyWord),
                                    Criteria.where("telephone").regex(keyWord),
                                    Criteria.where("account").regex(keyWord)));

修改

public void updateByCommentId(ObjectId commentId, Map<String, Object> map) {
        Query query = createQuery("commentId", commentId);
        Update ops = createUpdate();
        map.forEach((key, value) -> {
            ops.set(key, value);
        });
        update(query, ops);
    }

模糊查询

if(null != nickName){
            query.addCriteria(Criteria.where("nickName").regex(".*" + nickName + ".*"));
        }

 if (!StringUtil.isEmpty(roomName)) {
            query.addCriteria(containsIgnoreCase("name", roomName));
        }

删除

public void deleteMsg(int userId) {
        Query query = createQuery("userId", userId);
        deleteByQuery(query);
    }

聚合分页

 /**
     * 按时间区间统计每个用户看广告的收益
     * @param beginTime 开始时间
     * @param endTime 结束时间
     * @param pageIndex  页码（必须分页处理，海量数据）
     * @param pageSize
     * @return
     */
    @Override
    public List<WatchAdvertising> getTotalMoney(long beginTime, long endTime, int pageIndex, int pageSize) {
        //封装查询条件
        List<AggregationOperation> operations = new ArrayList<>();
        if(beginTime > 0) {
            operations.add(Aggregation.match(Criteria.where("time").gte(beginTime)));
        }
        if(endTime > 0) {
            operations.add(Aggregation.match(Criteria.where("time").lte(endTime)));
        }
        //totalMoney为WatchAdvertising类中新建的属性，用于接收统计后的总收益；当然也可以使用money（或其他属性）接收,后面的 first("userId").as("userId") 代表需要查询的字段映射到实体里的名称
        operations.add(Aggregation.group("userId").first("userId").as("userId").sum("money").as("totalMoney"));
        operations.add(Aggregation.project("userId","totalMoney").andExclude("_id")); //去掉id
        int startRows = (pageIndex - 1) * pageSize;//计算起始下标
        startRows = startRows < 0 ? 0 : startRows;
        operations.add(Aggregation.skip((long) startRows));
        operations.add(Aggregation.limit((long) pageSize));
        Aggregation aggregation = Aggregation.newAggregation(operations);
        //查询、并获取结果
        AggregationResults<WatchAdvertising> results = mongoTemplate.aggregate(aggregation, "watchAdvertising", WatchAdvertising.class);
        return results.getMappedResults();
    }

分组去重

    // 查询条件
    Criteria criteria = Criteria
            .where("logsType").is("LOGIN");
    // 格式化日期 YY-MM-DD，并且指定时区，Mongo不区分时区，按月则%Y-%m
    AggregationExpression dateExpression = DateOperators.DateToString
            .dateOf("createTime")
            .toString("%Y-%m-%d")
            .withTimezone(
                    DateOperators.Timezone.valueOf("+08")
            );

    Aggregation aggregation = Aggregation.newAggregation(
            Aggregation.match(criteria),
            Aggregation.project("account")
                    .and(dateExpression).as("date"),
            // 按日期和用户分组，去重用户，一个用户，一天只算一次
            Aggregation.group("date", "account")
                    .first("date").as("date")
                    .first("account").as("account"),
            // 统计按日期分组，统计每天登录的成功的总人数
            Aggregation.group("date")
                    .count().as("loginUserNumber")
                    .first("date").as("date")
    );

    List<JSONObject> list = mongoTemplate.aggregate(aggregation, "USER_LOGIN_RECORD", JSONObject.class).getMappedResults();