mongodb分组函数的使用(spring-data-mongodb)

这两天要做mongodb日志的模块,下面记录一下。

一、

首先要导入一批数据,使用springboot来完成。

配置mongodb的复制集:在application.yml文件中配置uri来完成

格式:mongodb://用户名:密码@ip:端口[,ip:端口] [,ip:端口]/数据库名

下面注入mongoTemplate:

 

二、

Mongodb在java中分组使用应该有两三种方式:可以百度一下

下面我贴上我们项目组使用的代码:

@Service("logMongoService")

public class LogMongoServiceImpl implements LogMongoService {

         /**

          * 按月统计登录次数使用的初始化的js对象

          */

         private static final String initMonthGroupByJs = "{total:0," +

                          "JanCount:0," +

                          "FebCount:0," +

                          "MarCount:0," +

                          "AprCount:0," +

                          "MayCount:0," +

                          "JuneCount:0," +

                          "JulyCount:0," +

                          "AugCount:0," +

                          "SepCount:0," +

                          "OctCount:0," +

                          "NovCount:0," +

                          "DecCount:0}";

         /**

          * 按月统计登录次数使用的js,mongodb在执行时会自动的执行此脚本

          */

         private static final String monthGroupByJs =

                          "function(doc, prev){" +

                                            "prev.total += 1;" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 0) {" +

                                                     "prev.JanCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 1) {" +

                                                     "prev.FebCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 2) {" +

                                                     "prev.MarCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 3) {" +

                                                     "prev.AprCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 4) {" +

                                                     "prev.MayCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 5) {" +

                                                     "prev.JuneCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 6) {" +

                                                     "prev.JulyCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 7) {" +

                                                     "prev.AugCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 8) {" +

                                                     "prev.SepCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 9) {" +

                                                     "prev.OctCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 10) {" +

                                                     "prev.NovCount += 1;" +

                                            "}" +

                                            "if (doc.startTime.getFullYear() == searchYear && doc.startTime.getMonth() == 11) {" +

                                                     "prev.DecCount += 1;" +

                                            "}" +

                          "}";

  @Override

    public void findMonthLogin(MongoPage page, SystemLogQuery systemLogQuery) throws Exception{

                  //添加年份的限制  也就是查询条件

                  Criteria[] logType ={Criteria.where("logType").is("2")};

                  Calendar startCalendar = Calendar.getInstance();

                  startCalendar.set(year, 0, 1, 0,0, 0);

                  Calendar endCalendar = Calendar.getInstance();

                  endCalendar.set(year + 1, 0, 1, 0,0, 0);

                  Criteria[] yearCriteria ={Criteria.where("startTime").gte(startCalendar.getTime()).lt(endCalendar.getTime())};

                  logType = (Criteria[]) ArrayUtils.addAll(logType, yearCriteria);

                  //组织分组时使用的js

                  String initGroupByJs = initMonthGroupByJs;

                  String groupByJs = monthGroupByJs.replace("searchYear", year + "");

                  //此处开始分组查询

             //以name字段进行分组

             GroupBy groupBy = new GroupBy("username","name","departmentName").

                                                     initialDocument(initGroupByJs).

                                                     reduceFunction(groupByJs);

             //执行分组查询

             GroupByResults groupByResults = logMongoDao.group(new Criteria().andOperator(logType), groupBy);

//取出分组后的数据 这里必须是retval

             BasicDBList list = (BasicDBList) groupByResults.getRawResults().get("retval");

                  page.setTotal(list.size());

                  //用来存放查询到的list集合

             List<SystemLogVO> logVOList = new ArrayList<SystemLogVO>();

        for (int i = (page.getPageNumber()-1) * page.getPageSize();

                               i < page.getPageNumber() * page.getPageSize() && i < list.size(); i++) {

                 SystemLogVO systemlog = new SystemLogVO();

                          //在这个obj中就包含了所有的分组后,我们自定义的哪些属性的值

                 BasicDBObject obj = (BasicDBObject)list.get(i);

                 systemlog.setName(obj.get("name") == null? "" : obj.get("name").toString());

                 systemlog.setDepartmentName(obj.get("departmentName") == null?"":obj.get("departmentName").toString());

                 systemlog.setUsername(obj.get("username") == null?"":obj.get("username").toString());

                          //获取其中的月份的值

                          obj.get("total");//总的值

                          obj.get("JanCount");//根据条件过滤后,统计的1月份的值

                          obj.get("DecCount");

        }

         }               

}


需要说明的一点是,groupByJs这个是一个回调函数, 

当mongodb在过滤数据时,会调用这个js方法,doc变量是将正在处理的一条数据传输进来,prev相当于上面定义的initMonthGroupByJs中的js对象,

最后在initMonthGroupByJs中定义的对象会返回给我们,我们可以从prev中取到我们所有需要的数据。

Additiong:

https://blog.csdn.net/liming_0820/article/details/78657146

http://www.cnblogs.com/anhaogoon/p/9354248.html

猜你喜欢

转载自www.cnblogs.com/yjk295722366/p/9486430.html