From 744594ef1a2f530fc3e86ea9dc48b62247f79420 Mon Sep 17 00:00:00 2001 From: admin <weikou2014> Date: 星期二, 19 五月 2020 17:13:23 +0800 Subject: [PATCH] 饿了么绘图,添加口碑 --- fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java | 224 +++++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 215 insertions(+), 9 deletions(-) diff --git a/fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java b/fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java index 730ea5d..8ff1d39 100644 --- a/fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java +++ b/fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java @@ -1,10 +1,12 @@ package com.yeshi.fanli.dao.user.invite; import java.math.BigDecimal; +import java.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Date; import java.util.HashMap; import java.util.List; +import java.util.Locale; import java.util.Map; import org.springframework.data.mongodb.core.query.Criteria; @@ -12,15 +14,18 @@ import org.springframework.data.mongodb.core.query.Update; import org.springframework.stereotype.Repository; +import com.mongodb.BasicDBList; import com.mongodb.BasicDBObject; +import com.mongodb.DBCollection; import com.mongodb.DBObject; +import com.mongodb.GroupCommand; import com.yeshi.fanli.dao.MongodbBaseDao; import com.yeshi.fanli.entity.bus.user.invite.TeamDailyRecord; @Repository public class TeamDailyRecordDao extends MongodbBaseDao<TeamDailyRecord> { - private static String collectionName = "team_daily_Record"; + private static String collectionName = "teamDailyRecord"; /** @@ -146,10 +151,10 @@ Map<String, Object> map = new HashMap<>(); map.put("firstDaRen", 0); - map.put("secondDaRen", 0); map.put("firstHighVIP", 0); - map.put("secondHighVIP", 0); map.put("firstSuperVIP", 0); + map.put("secondDaRen", 0); + map.put("secondHighVIP", 0); map.put("secondSuperVIP", 0); BasicDBObject dbObject = new BasicDBObject(map); DBObject result = mongoTemplate.getCollection(collectionName).group(agg, query.getQueryObject(), dbObject, reduce); @@ -159,15 +164,216 @@ BasicDBObject object = mapResult.get(i + ""); TeamDailyRecord record = new TeamDailyRecord(); - record.setFirstDaRen(new BigDecimal(object.get("totalNum").toString()).intValue()); - record.setFirstHighVIP(new BigDecimal(object.get("totalNum").toString()).intValue()); - record.setFirstSuperVIP(new BigDecimal(object.get("totalNum").toString()).intValue()); - record.setSecondDaRen(new BigDecimal(object.get("totalNum").toString()).intValue()); - record.setSecondHighVIP(new BigDecimal(object.get("totalNum").toString()).intValue()); - record.setSecondSuperVIP(new BigDecimal(object.get("totalNum").toString()).intValue()); + record.setFirstDaRen(new BigDecimal(object.get("firstDaRen").toString()).intValue()); + record.setFirstHighVIP(new BigDecimal(object.get("firstHighVIP").toString()).intValue()); + record.setFirstSuperVIP(new BigDecimal(object.get("firstSuperVIP").toString()).intValue()); + record.setSecondDaRen(new BigDecimal(object.get("secondDaRen").toString()).intValue()); + record.setSecondHighVIP(new BigDecimal(object.get("secondHighVIP").toString()).intValue()); + record.setSecondSuperVIP(new BigDecimal(object.get("secondSuperVIP").toString()).intValue()); listDailyCount.add(record); } } return listDailyCount; } + + + private String getTeamNumReduce() { + StringBuilder builder = new StringBuilder(); + builder.append("function(doc, aggr){ "); + builder.append(" if(doc.firstNum > 0){ aggr.firstTotal += doc.firstNum;}"); + builder.append(" aggr.secondTotal += doc.secondNum;"); + builder.append(" aggr.beyondTotal += doc.beyondNum;"); + builder.append(" }"); + return builder.toString(); + } + + + /** + * 缁熻姣忔湀璁㈠崟鏀跺叆銆佹暟閲� + * @param uid + * @param minDate + * @param maxDate + * @return + */ + public List<TeamDailyRecord> sumTeamNumGroupByCountDay(Long uid, Date minDate, Date maxDate) { + // 鏌ヨ鏉′欢 + List<Criteria> list = new ArrayList<Criteria>(); + list.add(Criteria.where("uid").is(uid)); + if (minDate != null) + list.add(Criteria.where("countDay").gte(minDate)); + if (maxDate != null) + list.add(Criteria.where("countDay").lte(maxDate)); + + Query query = new Query(); + if (list.size() > 0) { + Criteria[] cas = new Criteria[list.size()]; + for (int i = 0; i < list.size(); i++) + cas[i] = list.get(i); + query.addCriteria(new Criteria().andOperator(cas)); + } + + Map<String, Object> map = new HashMap<>(); + map.put("firstTotal", 0); + map.put("secondTotal", 0); + map.put("beyondTotal", 0); + BasicDBObject initial = new BasicDBObject(map); + + // 杩涜鎸夊ぉ銆佸懆銆佹湀鍒嗙粍 + BasicDBObject agg = new BasicDBObject("countDay", ""); + DBCollection collection = mongoTemplate.getCollection(collectionName); + GroupCommand xx = new GroupCommand(collection, agg, query.getQueryObject(), initial, getTeamNumReduce(), null); + BasicDBList objects = (BasicDBList) collection.group(xx); + + List<TeamDailyRecord> results = new ArrayList<>(); + if (objects != null) { + SimpleDateFormat sdf = new SimpleDateFormat("EEE MMM dd HH:mm:ss z yyyy", Locale.US); + for (int i = 0; i < objects.size(); i++) { + BasicDBObject dbObject = (BasicDBObject) objects.get(i); + if (dbObject == null) { + continue; + } + + Object object = dbObject.get("countDay"); + if (object == null) { + continue; + } + + try { + String date = object.toString(); + Date dtime = sdf.parse(date); + TeamDailyRecord dailyCount = new TeamDailyRecord(); + dailyCount.setCountDay(dtime); + dailyCount.setFirstNum(new BigDecimal(dbObject.get("firstTotal").toString()).intValue()); + dailyCount.setSecondNum(new BigDecimal(dbObject.get("secondTotal").toString()).intValue()); + dailyCount.setBeyondNum(new BigDecimal(dbObject.get("beyondTotal").toString()).intValue()); + + results.add(dailyCount); + } catch (Exception e) { + e.printStackTrace(); + } + } + } + return results; + } + + + /** + * 缁熻姣忔湀璁㈠崟鏀跺叆銆佹暟閲� + * @param uid + * @param minDate + * @param maxDate + * @return + */ + public List<TeamDailyRecord> sumTeamNumGroupByYearMonth(Long uid, Date minDate, Date maxDate) { + // 鏌ヨ鏉′欢 + List<Criteria> list = new ArrayList<Criteria>(); + list.add(Criteria.where("uid").is(uid)); + if (minDate != null) + list.add(Criteria.where("countDay").gte(minDate)); + if (maxDate != null) + list.add(Criteria.where("countDay").lte(maxDate)); + + Query query = new Query(); + if (list.size() > 0) { + Criteria[] cas = new Criteria[list.size()]; + for (int i = 0; i < list.size(); i++) + cas[i] = list.get(i); + query.addCriteria(new Criteria().andOperator(cas)); + } + + Map<String, Object> map = new HashMap<>(); + map.put("firstTotal", 0); + map.put("secondTotal", 0); + map.put("beyondTotal", 0); + BasicDBObject initial = new BasicDBObject(map); + + // 杩涜鎸夊ぉ銆佸懆銆佹湀鍒嗙粍 + BasicDBObject agg = new BasicDBObject("yearMonth", ""); + DBCollection collection = mongoTemplate.getCollection(collectionName); + GroupCommand xx = new GroupCommand(collection, agg, query.getQueryObject(), initial, getTeamNumReduce(), null); + BasicDBList objects = (BasicDBList) collection.group(xx); + + List<TeamDailyRecord> results = new ArrayList<>(); + if (objects != null) { + for (int i = 0; i < objects.size(); i++) { + BasicDBObject dbObject = (BasicDBObject) objects.get(i); + if (dbObject == null) { + continue; + } + + Object object = dbObject.get("yearMonth"); + if (object == null) { + continue; + } + + TeamDailyRecord dailyCount = new TeamDailyRecord(); + dailyCount.setYearMonth(object.toString()); + dailyCount.setFirstNum(new BigDecimal(dbObject.get("firstTotal").toString()).intValue()); + dailyCount.setSecondNum(new BigDecimal(dbObject.get("secondTotal").toString()).intValue()); + dailyCount.setBeyondNum(new BigDecimal(dbObject.get("beyondTotal").toString()).intValue()); + + results.add(dailyCount); + } + } + return results; + } + + /** + * 缁熻姣忔湀璁㈠崟鏀跺叆銆佹暟閲� + * @param uid + * @param minDate + * @param maxDate + * @return + */ + public List<TeamDailyRecord> sumTeamNumGroupByUid(Long uid, Date minDate, Date maxDate) { + // 鏌ヨ鏉′欢 + List<Criteria> list = new ArrayList<Criteria>(); + list.add(Criteria.where("uid").is(uid)); + if (minDate != null) + list.add(Criteria.where("countDay").gte(minDate)); + if (maxDate != null) + list.add(Criteria.where("countDay").lte(maxDate)); + + Query query = new Query(); + if (list.size() > 0) { + Criteria[] cas = new Criteria[list.size()]; + for (int i = 0; i < list.size(); i++) + cas[i] = list.get(i); + query.addCriteria(new Criteria().andOperator(cas)); + } + + Map<String, Object> map = new HashMap<>(); + map.put("firstTotal", 0); + map.put("secondTotal", 0); + map.put("beyondTotal", 0); + BasicDBObject initial = new BasicDBObject(map); + + // 杩涜鎸夊ぉ銆佸懆銆佹湀鍒嗙粍 + BasicDBObject agg = new BasicDBObject("uid", ""); + DBCollection collection = mongoTemplate.getCollection(collectionName); + GroupCommand xx = new GroupCommand(collection, agg, query.getQueryObject(), initial, getTeamNumReduce(), null); + BasicDBList objects = (BasicDBList) collection.group(xx); + + List<TeamDailyRecord> results = new ArrayList<>(); + if (objects != null) { + for (int i = 0; i < objects.size(); i++) { + BasicDBObject dbObject = (BasicDBObject) objects.get(i); + if (dbObject == null) { + continue; + } + + Object object = dbObject.get("uid"); + if (object == null) { + continue; + } + + TeamDailyRecord dailyCount = new TeamDailyRecord(); + dailyCount.setFirstNum(new BigDecimal(dbObject.get("firstTotal").toString()).intValue()); + dailyCount.setSecondNum(new BigDecimal(dbObject.get("secondTotal").toString()).intValue()); + dailyCount.setBeyondNum(new BigDecimal(dbObject.get("beyondTotal").toString()).intValue()); + results.add(dailyCount); + } + } + return results; + } } -- Gitblit v1.8.0