admin
2020-05-19 744594ef1a2f530fc3e86ea9dc48b62247f79420
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,8 +14,11 @@
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;
@@ -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;
   }
}