足迹、收藏订单兼容新需求

yujian

2020-05-23 28cf328a098334b51a3e9d2d56f983fb8c862211

 fanli/src/main/java/com/yeshi/fanli/dao/user/invite/TeamDailyRecordDao.java

@@ -9,10 +9,16 @@
import java.util.Locale;

import java.util.Map;



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.mongodb.BasicDBList;

import com.mongodb.BasicDBObject;

@@ -21,6 +27,7 @@
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 extends MongodbBaseDao<TeamDailyRecord> {

@@ -180,9 +187,9 @@
   private String getTeamNumReduce() {

      StringBuilder builder = new StringBuilder();

      builder.append("function(doc, aggr){ ");

      builder.append(" aggr.directOrderNum += doc.firstNum;");

      builder.append(" aggr.inDirectOrderNum += doc.secondNum;");

      builder.append(" aggr.beyondOrderNum += doc.beyondNum;");

      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();

   }

@@ -213,9 +220,9 @@
      }



      Map<String, Object> map = new HashMap<>();

      map.put("directOrderNum", 0);

      map.put("inDirectOrderNum", 0);

      map.put("beyondOrderNum", 0);

      map.put("firstTotal", 0);

      map.put("secondTotal", 0);

      map.put("beyondTotal", 0);

      BasicDBObject initial = new BasicDBObject(map);



      // 进行按天、周、月分组

@@ -243,9 +250,10 @@
               Date dtime = sdf.parse(date);

               TeamDailyRecord dailyCount = new TeamDailyRecord();

               dailyCount.setCountDay(dtime);

               dailyCount.setFirstNum(new BigDecimal(dbObject.get("directOrderNum").toString()).intValue());

               dailyCount.setSecondNum(new BigDecimal(dbObject.get("inDirectOrderNum").toString()).intValue());

               dailyCount.setBeyondNum(new BigDecimal(dbObject.get("beyondOrderNum").toString()).intValue());

               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();

@@ -281,9 +289,9 @@
      }



      Map<String, Object> map = new HashMap<>();

      map.put("directOrderNum", 0);

      map.put("inDirectOrderNum", 0);

      map.put("beyondOrderNum", 0);

      map.put("firstTotal", 0);

      map.put("secondTotal", 0);

      map.put("beyondTotal", 0);

      BasicDBObject initial = new BasicDBObject(map);



      // 进行按天、周、月分组

@@ -307,9 +315,9 @@
            

            TeamDailyRecord dailyCount = new TeamDailyRecord();

            dailyCount.setYearMonth(object.toString());

            dailyCount.setFirstNum(new BigDecimal(dbObject.get("directOrderNum").toString()).intValue());

            dailyCount.setSecondNum(new BigDecimal(dbObject.get("inDirectOrderNum").toString()).intValue());

            dailyCount.setBeyondNum(new BigDecimal(dbObject.get("beyondOrderNum").toString()).intValue());

            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);

         }

@@ -342,9 +350,9 @@
      }



      Map<String, Object> map = new HashMap<>();

      map.put("directOrderNum", 0);

      map.put("inDirectOrderNum", 0);

      map.put("beyondOrderNum", 0);

      map.put("firstTotal", 0);

      map.put("secondTotal", 0);

      map.put("beyondTotal", 0);

      BasicDBObject initial = new BasicDBObject(map);



      // 进行按天、周、月分组

@@ -367,13 +375,90 @@
            }

            

            TeamDailyRecord dailyCount = new TeamDailyRecord();

            dailyCount.setFirstNum(new BigDecimal(dbObject.get("directOrderNum").toString()).intValue());

            dailyCount.setSecondNum(new BigDecimal(dbObject.get("inDirectOrderNum").toString()).intValue());

            dailyCount.setBeyondNum(new BigDecimal(dbObject.get("beyondOrderNum").toString()).intValue());

						
            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;

   }

	
	
	


   /**

    * 根据订单排行榜进行uid分组订单数量排序

    * @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;

   }

}