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 | 325 +++++++++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 299 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..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,36 +1,33 @@
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.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 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;
@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 +62,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 +82,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 +93,287 @@
*
* @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;
+ }
}
--
Gitblit v1.8.0