java操作 mongodb 进行 按天、周、月分组统计

	public List<PlaybackEntity> queryPlaybackRecord(FunctionUseQueryEntity queryEntity){
		Criteria criteria = Criteria.where("createTime").gte(queryEntity.getStartTime()).lte(queryEntity.getEndTime())
							.and("status").is(1);
		Query query = Query.query(criteria);	
		
        BasicDBObject initial = new BasicDBObject();
        initial.put("playbackNum", 0);
        
        String reduce = getReduce();
        //进行按天、周、月分组
        String keyf = getKeyf(queryEntity);
           
        DBCollection collection = mongoTemplate.getCollection(COLLNAME);
        
        GroupCommand xx = new GroupCommand(collection, keyf, query.getQueryObject(), initial, reduce, null);
 
        BasicDBList  objects =  (BasicDBList)collection.group(xx);
        
        List<PlaybackEntity> results = new ArrayList<PlaybackEntity>();
        if (objects != null) {
            for (int i = 0; i < objects.size(); i++) {
            	PlaybackEntity play = new PlaybackEntity();
                DBObject obj = (DBObject) objects.get(i);
                play.setFormateTime((String)obj.get("strData"));
                play.setPlaybackNum((Double)obj.get("playbackNum"));
                results.add(play);
            }
        }
		return results;
	}

	private String getKeyf(FunctionUseQueryEntity queryEntity) {
		StringBuilder keyfBuilder = new StringBuilder();
        keyfBuilder.append("function(doc){ ")
        			.append("    var date = new Date(doc.createTime); ");
        if(DateType.DATE.getValue().equals(queryEntity.getType())){
        	keyfBuilder.append("    var dateKey = '' + date.getFullYear()+ '-' + (date.getMonth()+1) + '-' + date.getDate();");
        }else if(DateType.WEEK.getValue().equals(queryEntity.getType())){
        	keyfBuilder.append("	var firstDay = new Date(date.getFullYear(),0, 1);")
        			   .append("    var dayOfWeek = firstDay.getDay();") 
        			   .append("    var spendDay= 1;") 
        			   .append("    if (dayOfWeek !=0) {") 
        			   .append("         spendDay= 7 - dayOfWeek + 1;") 
        			   .append("    }") 
        			   .append("    firstDay = new Date(date.getFullYear(), 0, 1 + spendDay);") 
        			   .append("    var d =Math.ceil((date.valueOf()- firstDay.valueOf())/ 86400000);") 
        			   .append("    var week =Math.ceil(d/7);") 
        			   .append("    week = week + 1;"); 
        	keyfBuilder.append("    var dateKey = '' + date.getFullYear()+ '-' + week;");
        }else if(DateType.MONTH.getValue().equals(queryEntity.getType())){
        	keyfBuilder.append("    var dateKey = '' + date.getFullYear()+ '-' + (date.getMonth()+1);");
        }
        keyfBuilder.append("    return {'strData':dateKey};  ")
        		   .append(" }");
		return keyfBuilder.toString();
	}

	private String getReduce() {
		StringBuilder reduceBuilder = new StringBuilder();
        reduceBuilder.append("function Reduce(doc, out) { ")
        			 .append("    out.playbackNum += 1; ")
        			 .append(" }");
		return reduceBuilder.toString();
	}

  

转载于:https://www.cnblogs.com/huangziwu/p/9414580.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值