From 28cf328a098334b51a3e9d2d56f983fb8c862211 Mon Sep 17 00:00:00 2001 From: yujian <yujian@163.com> Date: 星期六, 23 五月 2020 09:54:38 +0800 Subject: [PATCH] 足迹、收藏订单兼容新需求 --- fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java | 410 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 384 insertions(+), 26 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 8576988..a20d14b 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,36 +1,40 @@ 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 javax.annotation.Resource; - -import org.springframework.data.mongodb.core.MongoTemplate; +import org.springframework.data.domain.Sort; +import org.springframework.data.mongodb.core.aggregation.Aggregation; +import org.springframework.data.mongodb.core.aggregation.AggregationOperation; +import org.springframework.data.mongodb.core.aggregation.AggregationResults; +import org.springframework.data.mongodb.core.aggregation.LookupOperation; import org.springframework.data.mongodb.core.query.Criteria; import org.springframework.data.mongodb.core.query.Query; import org.springframework.data.mongodb.core.query.Update; import org.springframework.stereotype.Repository; +import org.yeshi.utils.DateUtil; -import com.yeshi.fanli.entity.bus.user.vip.TeamDailyRecord; +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; +import com.yeshi.fanli.vo.order.OrderRankingVO; @Repository -public class TeamDailyRecordDao { +public class TeamDailyRecordDao extends MongodbBaseDao<TeamDailyRecord> { - @Resource - private MongoTemplate mongoTemplate; + private static String collectionName = "teamDailyRecord"; - /** - * 鏂板 - * - * @param record - */ - public void save(TeamDailyRecord record) { - if (record == null) { - return; - } - mongoTemplate.save(record); - } - + /** * 閫夋嫨鎬ф洿鏂� * @param record @@ -65,7 +69,7 @@ if (record.getBeyondSuperVIP() != null) update.set("beyondSuperVIP", record.getBeyondSuperVIP()); update.set("updateTime", new Date()); - mongoTemplate.updateMulti(query, update, TeamDailyRecord.class); + update(query, update); } /** @@ -85,10 +89,10 @@ * * @return */ - public List<TeamDailyRecord> getbyUid(Long uid) { + public List<TeamDailyRecord> getByUid(Long uid) { Query query = new Query(); query.addCriteria(Criteria.where("uid").is(uid)); - return mongoTemplate.find(query, TeamDailyRecord.class); + return findList(query); } /** @@ -96,11 +100,365 @@ * * @return */ - public List<TeamDailyRecord> getbyUid(Long uid, Date minTime) { + public List<TeamDailyRecord> listByUid(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(); - query.addCriteria(Criteria.where("uid").is(uid)); - query.addCriteria(Criteria.where("countDay").lte(minTime)); - return mongoTemplate.find(query, TeamDailyRecord.class); + 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)); + } + return findList(query); } + + /** + * 缁熻骞存湀 + * @param uid + * @param minDate + * @param maxDate + * @return + */ + public List<TeamDailyRecord> sumGroupByYearMonth(Long uid, Date minDate, Date maxDate) { + List<TeamDailyRecord> listDailyCount = new ArrayList<>(); + // 缁熻鏂规硶 + String reduce = "function(doc, aggr){" + + " aggr.firstDaRen += doc.firstDaRen;" + + " aggr.secondDaRen += doc.secondDaRen;" + + " aggr.firstHighVIP += doc.firstHighVIP;" + + " aggr.secondHighVIP += doc.secondHighVIP;" + + " aggr.firstSuperVIP += doc.firstSuperVIP;" + + " aggr.secondSuperVIP += doc.secondSuperVIP;" + + "}"; + + // 鏌ヨ鏉′欢 + 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)); + } + + BasicDBObject agg = new BasicDBObject("yearMonth", ""); + + Map<String, Object> map = new HashMap<>(); + map.put("firstDaRen", 0); + map.put("firstHighVIP", 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); + Map<String, BasicDBObject> mapResult = result.toMap(); + if (mapResult.size() > 0) { + for (int i = 0; i < mapResult.size(); i++) { + BasicDBObject object = mapResult.get(i + ""); + + TeamDailyRecord record = new TeamDailyRecord(); + 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; + } + + + + + /** + * 鏍规嵁璁㈠崟鎺掕姒滆繘琛寀id鍒嗙粍璁㈠崟鏁伴噺鎺掑簭 + * @param uid + * @param minDate + * @param maxDate + * @return + */ + public List<OrderRankingVO> getRankingByFirstNum(Long uid, Date minDate, Date maxDate) { + LookupOperation lookupToLots = LookupOperation.newLookup(). + from("teamDailyRecord").//鍏宠仈琛ㄥ悕锛屽鏂� + localField("workerUid").// 涓昏〃鍏宠仈瀛楁 + foreignField("uid").// 娆¤〃瀛楁鍏宠仈瀛楁 + as("result"); // 娆¤〃鐨勫埆鍚� + + // 鏌ヨ鏉′欢 + List<Criteria> list = new ArrayList<Criteria>(); + list.add(Criteria.where("bossUid").is(uid)); + if (minDate != null) { + minDate = DateUtil.reduceDay(minDate, 1); + list.add(Criteria.where("orders.countDay").gt(minDate)); + } + if (maxDate != null) { + maxDate = DateUtil.reduceDay(maxDate, 1); + list.add(Criteria.where("orders.countDay").lte(maxDate)); + } + + // 杞崲鏁扮粍 + Criteria[] cas = new Criteria[list.size()]; + for (int i = 0; i < list.size(); i++) + cas[i] = list.get(i); + // 鏌ヨ鍙傛暟 + Criteria criteria = new Criteria().andOperator(cas); + + + List<AggregationOperation> aggs = new ArrayList<>(); + // 宸﹁繛鎺� + aggs.add(lookupToLots); + // 鎷嗗垎瀛愭暟缁� + aggs.add(Aggregation.unwind("result")); + // 鏉′欢鍖归厤 + aggs.add(Aggregation.match(criteria)); + // 闇�瑕佽繑鍥炵殑瀛楁 + aggs.add(Aggregation.project().and("workerUid").as("workerUid").and("level").as("level") + .and("result.firstNum").as("number")); + // 鍒嗙粍姹傚拰骞惰繑鍥� + aggs.add(Aggregation.group("workerUid").first("workerUid").as("workerUid").first("level").as("level") + .sum("number").as("number")); + // 鎺掑簭 + aggs.add(Aggregation.sort(new Sort(Sort.Direction.DESC, "number"))); + aggs.add(Aggregation.limit(10)); + + // 缁勭粐鏉′欢 + Aggregation agg = Aggregation.newAggregation(aggs); + // 鎵ц鏌ヨ + AggregationResults<BasicDBObject> results = mongoTemplate.aggregate(agg, "threeSaleDetail", BasicDBObject.class); + + // 杩斿洖缁撴灉澶勭悊 + List<BasicDBObject> livevideo = results.getMappedResults(); + + List<OrderRankingVO> resultList = new ArrayList<>(); + for (BasicDBObject obj : livevideo) { + Object object = obj.get("workerUid"); + if (object == null) { + continue; + } + + OrderRankingVO vo = new OrderRankingVO(); + vo.setUid(new BigDecimal(obj.get("workerUid").toString()).longValue()); + vo.setGrade(new BigDecimal(obj.get("level").toString()).intValue()); + vo.setNumber(new BigDecimal(obj.get("number").toString()).intValue()); + resultList.add(vo); + } + return resultList; + } } -- Gitblit v1.8.0