Ejemplo n.º 1
0
 @Override
 public List<SystemUserDTO> getAccount() {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   return ObjectUtils.convert(
       mongoTemplate.find(Query.query(Criteria.where("state").is(0)), getEntityClass()),
       getDTOClass());
 }
Ejemplo n.º 2
0
 @Override
 public void uploadImg(byte[] bytes) {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   mongoTemplate.updateFirst(
       Query.query(Criteria.where("userName").is(AppContext.getUser())),
       Update.update("img", bytes),
       getEntityClass());
 }
Ejemplo n.º 3
0
 @Override
 public SystemUserDTO getCurrUserInfo() {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   SystemUserEntity entity =
       mongoTemplate.findOne(
           Query.query(Criteria.where("userName").is(AppContext.getUser())), getEntityClass());
   return ObjectUtils.convert(entity, getDTOClass());
 }
Ejemplo n.º 4
0
 @Override
 public List<SystemUserDTO> getAllSysUserAccount() {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   List<SystemUserEntity> userEntityList =
       mongoTemplate.find(
           Query.query(Criteria.where("access").is(2).and("acstate").is(1).and("state").is(1)),
           getEntityClass());
   return ObjectUtils.convertToList(userEntityList, getDTOClass());
 }
Ejemplo n.º 5
0
 /**
  * @param dates
  * @return
  */
 @Override
 public List<AccountReportDTO> getAccountReports(List<Date> dates) {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getUserReportMongo();
   Long baiduAccountId = AppContext.getAccountId();
   return ObjectUtils.convert(
       mongoTemplate.find(
           Query.query(Criteria.where(ACCOUNT_ID).is(baiduAccountId).and("date").in(dates)),
           AccountReportEntity.class),
       AccountReportDTO.class);
 }
Ejemplo n.º 6
0
 @Override
 public boolean updateSysAccount(String userName, Long state) {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   Update update = new Update();
   update.set("acstate", state);
   WriteResult writeResult =
       mongoTemplate.updateFirst(
           Query.query(Criteria.where("userName").is(userName)), update, "sys_user");
   return writeResult.isUpdateOfExisting();
 }
Ejemplo n.º 7
0
 @Override
 public boolean updateBaiDuName(String name, Long baiduId) {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   Update update = new Update();
   update.set("bdAccounts.$.baiduRemarkName", name);
   WriteResult writeResult =
       mongoTemplate.updateFirst(
           Query.query(Criteria.where("bdAccounts._id").is(baiduId)), update, "sys_user");
   return writeResult.isUpdateOfExisting();
 }
Ejemplo n.º 8
0
 @Override
 public boolean updatePwd(String account, String pwd) {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   WriteResult writeResult =
       mongoTemplate.updateFirst(
           Query.query(Criteria.where("userName").is(account)),
           Update.update("password", pwd),
           "sys_user");
   return writeResult.isUpdateOfExisting();
 }
Ejemplo n.º 9
0
 @Override
 public int updateAccountStruts(String userName) {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   WriteResult writeResult =
       mongoTemplate.updateFirst(
           Query.query(Criteria.where("userName").is(userName)),
           Update.update("state", 1),
           "sys_user");
   int i = 0;
   if (writeResult.isUpdateOfExisting()) i = 1;
   return i;
 }
Ejemplo n.º 10
0
  @Override
  public Double getYesterdayCost(Long accountId) {
    MongoTemplate mongoTemplate = BaseMongoTemplate.getUserReportMongo();
    Date date = DateUtils.getYesterday();
    Aggregation aggregation =
        newAggregation(
            match(Criteria.where(ACCOUNT_ID).is(accountId).and("date").is(date)),
            project("pccost"));
    AggregationResults<AccountReportEntity> results =
        mongoTemplate.aggregate(aggregation, TBL_ACCOUNT_REPORT, AccountReportEntity.class);
    if (results == null) return 0d;

    AccountReportEntity reportEntity = results.getUniqueMappedResult();
    if (reportEntity != null) return reportEntity.getPcCost().doubleValue();
    else return 0d;
  }
Ejemplo n.º 11
0
  @Override
  public void updateBaiduAccountInfo(BaiduAccountInfoDTO dto) {
    MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
    String currUser = AppContext.getUser();
    Update update = new Update();
    if (dto.getBudget() != null) update.set("bdAccounts.$.bgt", dto.getBudget());

    if (dto.getIsDynamicCreative() != null)
      update.set("bdAccounts.$.dc", dto.getIsDynamicCreative());

    if (dto.getExcludeIp() != null) update.set("bdAccounts.$.exIp", dto.getExcludeIp());

    mongoTemplate.updateFirst(
        Query.query(Criteria.where("userName").is(currUser).and("bdAccounts._id").is(dto.getId())),
        update,
        getEntityClass());
  }
Ejemplo n.º 12
0
 @Override
 public Double getCostRate() {
   Long accountId = AppContext.getAccountId();
   Double cost1 = getYesterdayCost(accountId);
   Double cost2 = 0d;
   Double costRate;
   MongoTemplate mongoTemplate = BaseMongoTemplate.getUserReportMongo();
   Long baiduAccountId = AppContext.get().getAccountId();
   Date date =
       ((List<Date>) DateUtils.getsLatestAnyDays("MM-dd", 2).get(DateUtils.KEY_DATE)).get(1);
   AccountReportEntity reportEntity =
       mongoTemplate.findOne(
           Query.query(Criteria.where("date").is(date).and(ACCOUNT_ID).is(baiduAccountId)),
           AccountReportEntity.class);
   if (reportEntity != null) cost2 = reportEntity.getPcCost().doubleValue();
   if (cost2 == 0d) return 0d;
   costRate = (cost1 - cost2) / cost2;
   costRate = new BigDecimal(costRate * 100).setScale(2, BigDecimal.ROUND_HALF_UP).doubleValue();
   return costRate;
 }
  @Override
  public void getAdgroupReportData(
      List<AdgroupReportDTO> adgroupReportDTOs, SystemUserDTO systemUser, String dateStr, int i) {
    MongoTemplate mongoTemplate;
    mongoTemplate =
        BaseMongoTemplate.getMongoTemplate(DBNameUtils.getReportDBName(systemUser.getUserName()));

    List<AdgroupReportEntity> adgroupReportEntities =
        new ArrayList<>(ObjectUtils.convert(adgroupReportDTOs, AdgroupReportEntity.class));
    List<AdgroupReportEntity> adgroupReportEntities1;
    if (mongoTemplate.collectionExists(dateStr + "-adgroup")) {
      if (i > 1) {
        adgroupReportEntities1 =
            new ArrayList<>(
                mongoTemplate.find(new Query(), AdgroupReportEntity.class, dateStr + "-adgroup"));
        adgroupReportEntities.addAll(adgroupReportEntities1);
      }
      mongoTemplate.dropCollection(dateStr + "-adgroup");
    }

    mongoTemplate.insert(adgroupReportEntities, dateStr + "-adgroup");
  }
  @Override
  public void getAccountReportData(
      List<AccountReportDTO> accountReportDTOs,
      SystemUserDTO systemUser,
      String dateStr,
      String baiduUserName) {
    MongoTemplate mongoTemplate;
    mongoTemplate =
        BaseMongoTemplate.getMongoTemplate(DBNameUtils.getReportDBName(systemUser.getUserName()));

    List<AccountReportEntity> accountReportEntities =
        ObjectUtils.convert(accountReportDTOs, AccountReportEntity.class);
    Date date = new Date();
    SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
    String dateString = df.format(date);

    List<AccountReportEntity> entities = null;
    if (!dateStr.equals(dateString)) {
      try {
        entities =
            mongoTemplate.find(
                Query.query(
                    Criteria.where("date")
                        .is(new SimpleDateFormat("yyyy-MM-dd").parse(dateStr))
                        .and("acna")
                        .is(baiduUserName)),
                AccountReportEntity.class);
      } catch (ParseException e) {
        e.printStackTrace();
      }

      if (entities == null || entities.size() == 0) {
        mongoTemplate.insert(accountReportEntities, TBL_ACCOUNT_REPORT);
      }
    }
  }
Ejemplo n.º 15
0
  /**
   * 百度账户树
   *
   * @return
   */
  @Override
  public ArrayNode getAccountTree() {
    ObjectMapper mapper = new ObjectMapper();
    ArrayNode arrayNode = mapper.createArrayNode();
    ObjectNode objectNode;

    Long accountId = AppContext.getAccountId();
    MongoTemplate mongoTemplate = BaseMongoTemplate.getUserMongo();
    Aggregation aggregation1 =
        Aggregation.newAggregation(
            match(Criteria.where(ACCOUNT_ID).is(accountId)),
            project(CAMPAIGN_ID, NAME, SYSTEM_ID),
            sort(Sort.Direction.ASC, CAMPAIGN_ID));
    // 推广计划树
    AggregationResults<CampaignVO> results1 =
        mongoTemplate.aggregate(aggregation1, TBL_CAMPAIGN, CampaignVO.class);
    for (CampaignVO vo : results1) {
      objectNode = mapper.createObjectNode();

      if (vo.getCampaignId() == null) {
        objectNode.put("id", vo.getId());
      } else {
        objectNode.put("id", vo.getCampaignId());
      }

      objectNode.put("pId", 0);
      objectNode.put("name", vo.getCampaignName());
      arrayNode.add(objectNode);
    }

    Aggregation aggregation2 =
        Aggregation.newAggregation(
            match(Criteria.where(ACCOUNT_ID).is(accountId)),
            project(CAMPAIGN_ID, OBJ_CAMPAIGN_ID, ADGROUP_ID, SYSTEM_ID, NAME),
            sort(Sort.Direction.ASC, ADGROUP_ID));
    AggregationResults<AdgroupVO> results2 =
        mongoTemplate.aggregate(aggregation2, TBL_ADGROUP, AdgroupVO.class);
    for (AdgroupVO vo : results2) {
      objectNode = mapper.createObjectNode();
      objectNode.put("name", vo.getAdgroupName());
      if (vo.getCampaignId() == null) {
        objectNode.put("id", vo.getId());
        objectNode.put("pId", vo.getCampaignObjId());
        arrayNode.add(objectNode);
        continue;
      }

      if (vo.getAdgroupId() == null) {
        objectNode.put("id", vo.getId());
        objectNode.put("pId", vo.getCampaignId());
        arrayNode.add(objectNode);
        continue;
      }

      objectNode.put("id", vo.getAdgroupId());
      objectNode.put("pId", vo.getCampaignId());
      arrayNode.add(objectNode);
    }

    return arrayNode;
  }
Ejemplo n.º 16
0
 @Override
 public List<SystemUserDTO> getAccountAll() {
   MongoTemplate mongoTemplate = BaseMongoTemplate.getSysMongo();
   return ObjectUtils.convert(mongoTemplate.find(new Query(), getEntityClass()), getDTOClass());
 }