package com.yeshi.fanli.dao.user.invite;
|
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
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.mongodb.BasicDBObject;
|
import com.mongodb.DBObject;
|
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";
|
|
|
/**
|
* 选择性更新
|
* @param record
|
*/
|
public void updateSelective(TeamDailyRecord record) {
|
Query query = new Query();
|
query.addCriteria(Criteria.where("id").is(record.getId()));
|
|
Update update = new Update();
|
if (record.getFirstNum() != null)
|
update.set("firstNum", record.getFirstNum());
|
if (record.getSecondNum() != null)
|
update.set("secondNum", record.getSecondNum());
|
if (record.getBeyondNum() != null)
|
update.set("beyondNum", record.getBeyondNum());
|
if (record.getFirstDaRen() != null)
|
update.set("firstDaRen", record.getFirstDaRen());
|
if (record.getSecondDaRen() != null)
|
update.set("secondDaRen", record.getSecondDaRen());
|
if (record.getBeyondDaRen() != null)
|
update.set("beyondDaRen", record.getBeyondDaRen());
|
if (record.getFirstHighVIP() != null)
|
update.set("firstHighVIP", record.getFirstHighVIP());
|
if (record.getSecondHighVIP() != null)
|
update.set("secondHighVIP", record.getSecondHighVIP());
|
if (record.getBeyondHighVIP() != null)
|
update.set("beyondHighVIP", record.getBeyondHighVIP());
|
if (record.getFirstSuperVIP() != null)
|
update.set("firstSuperVIP", record.getFirstSuperVIP());
|
if (record.getSecondSuperVIP() != null)
|
update.set("secondSuperVIP", record.getSecondSuperVIP());
|
if (record.getBeyondSuperVIP() != null)
|
update.set("beyondSuperVIP", record.getBeyondSuperVIP());
|
update.set("updateTime", new Date());
|
update(query, update);
|
}
|
|
/**
|
* 根据id查询数据
|
*
|
* @param id
|
* @return
|
*/
|
public TeamDailyRecord getById(String id) {
|
Query query = new Query();
|
query.addCriteria(Criteria.where("id").is(id));
|
return mongoTemplate.findOne(query, TeamDailyRecord.class);
|
}
|
|
/**
|
* 查询
|
*
|
* @return
|
*/
|
public List<TeamDailyRecord> getByUid(Long uid) {
|
Query query = new Query();
|
query.addCriteria(Criteria.where("uid").is(uid));
|
return findList(query);
|
}
|
|
/**
|
* 查询
|
*
|
* @return
|
*/
|
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();
|
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("secondDaRen", 0);
|
map.put("firstHighVIP", 0);
|
map.put("secondHighVIP", 0);
|
map.put("firstSuperVIP", 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("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());
|
listDailyCount.add(record);
|
}
|
}
|
return listDailyCount;
|
}
|
}
|
