Exemplo n.º 1
0
  /**
   * 根据百度账户id获取账户信息
   *
   * @param baiduUserId
   * @return
   */
  @Override
  public BaiduAccountInfoDTO findByBaiduUserId(Long baiduUserId) {
    String currUser = AppContext.getUser();
    List<BaiduAccountInfoDTO> list = getBaiduAccountItems(currUser);

    BaiduAccountInfoDTO baiduAccount = null;
    for (BaiduAccountInfoDTO dto : list) {
      if (baiduUserId.equals(dto.getId())) {
        baiduAccount = dto;
        break;
      }
    }

    return baiduAccount;
  }
Exemplo n.º 2
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());
  }
  @Override
  public void initAccount(String userName, Long accountId) {
    logger.info("开始导入数据: 用户名=" + userName + ", 账号= " + accountId);

    SystemUserDTO systemUserDTO = getSystemUser(userName);
    if (systemUserDTO == null) {
      logger.warn("没有此账号: " + userName);
      return;
    }

    logger.info("清理已有数据...");
    clearAccountData(accountId);
    logger.info("清理数据完成!");

    for (BaiduAccountInfoDTO baiduAccountInfoDTO : systemUserDTO.getBaiduAccounts()) {

      Long aid = baiduAccountInfoDTO.getId();
      if (!Objects.equals(aid, accountId)) continue;
      CommonService commonService =
          BaiduServiceSupport.getCommonService(
              baiduAccountInfoDTO.getBaiduUserName(),
              baiduAccountInfoDTO.getBaiduPassword(),
              baiduAccountInfoDTO.getToken());
      BaiduApiService apiService = new BaiduApiService(commonService);

      logger.info("查询账户信息...");
      // 初始化账户数据
      AccountInfoType accountInfoType = apiService.getAccountInfo();
      if (accountInfoType == null) {
        logger.error("获取账户信息错误: " + ResHeaderUtil.getJsonResHeader(false).toString());
        continue;
      }
      boolean isDefault = baiduAccountInfoDTO.isDfault();
      baiduAccountInfoDTO = ObjectUtils.convert(accountInfoType, BaiduAccountInfoDTO.class);
      baiduAccountInfoDTO.setId(accountInfoType.getUserid());
      baiduAccountInfoDTO.setBaiduUserName(baiduAccountInfoDTO.getBaiduUserName());
      baiduAccountInfoDTO.setBaiduPassword(baiduAccountInfoDTO.getBaiduPassword());
      baiduAccountInfoDTO.setToken(baiduAccountInfoDTO.getToken());
      baiduAccountInfoDTO.setDfault(isDefault);

      // 新增百度账户
      systemUserDAO.insertAccountInfo(userName, baiduAccountInfoDTO);

      logger.info("查询账户推广计划...");
      List<CampaignType> campaignTypes = apiService.getAllCampaign();
      logger.info("查询结束: 计划数=" + campaignTypes.size());

      List<CampaignDTO> campaignEntities = EntityConvertUtils.convertToCamEntity(campaignTypes);

      // 查询推广单元
      List<Long> ids = new ArrayList<>(campaignEntities.size());

      for (CampaignDTO campaignEntity : campaignEntities) {
        campaignEntity.setAccountId(aid);
        ids.add(campaignEntity.getCampaignId());
      }

      logger.info("查询账户推广单元...");
      List<AdgroupType> adgroupTypeList = apiService.getAllAdGroup(ids);

      logger.info("查询结束: 单元数=" + adgroupTypeList.size());

      List<AdgroupDTO> adgroupEntities = EntityConvertUtils.convertToAdEntity(adgroupTypeList);
      ids.clear();
      for (AdgroupDTO adgroupEntity : adgroupEntities) {
        adgroupEntity.setAccountId(aid);
        ids.add(adgroupEntity.getAdgroupId());
      }

      logger.info("查询账户推广关键词...");

      //            List<KeywordType> keywordTypes = apiService.getAllKeyword(ids);
      //            logger.info("查询结束: 关键词数=" + keywordTypes.size());
      //
      //            List<KeywordDTO> keywordEntities =
      // EntityConvertUtils.convertToKwEntity(keywordTypes);
      //
      //            for (KeywordDTO keywordEntity : keywordEntities) {
      //                keywordEntity.setAccountId(aid);
      //            }

      // 分批次请求关键词数据
      List<Long> subList = new ArrayList<>(4);
      for (int i = 1; i <= ids.size(); i++) {
        Long adgroupId = ids.get(i - 1);
        subList.add(adgroupId);

        if (i % 4 == 0) {
          List<KeywordType> keywordTypes = apiService.getAllKeyword(subList);
          List<KeywordDTO> keywordEntities = EntityConvertUtils.convertToKwEntity(keywordTypes);

          for (KeywordDTO keywordEntity : keywordEntities) {
            keywordEntity.setAccountId(aid);
          }
          keywordDAO.save(keywordEntities);
          subList.clear();
        }
      }

      if (!subList.isEmpty()) {
        List<KeywordType> keywordTypes = apiService.getAllKeyword(subList);
        List<KeywordDTO> keywordEntities = EntityConvertUtils.convertToKwEntity(keywordTypes);

        for (KeywordDTO keywordEntity : keywordEntities) {
          keywordEntity.setAccountId(aid);
        }
        keywordDAO.save(keywordEntities);
        subList.clear();
      }

      logger.info("查询账户推广创意...");
      List<CreativeType> creativeTypes = apiService.getAllCreative(ids);
      logger.info("查询结束: 普通创意数=" + creativeTypes.size());

      List<CreativeDTO> creativeEntityList = EntityConvertUtils.convertToCrEntity(creativeTypes);

      for (CreativeDTO creativeEntity : creativeEntityList) {
        creativeEntity.setAccountId(aid);
      }

      // 开始保存数据
      campaignDAO.save(campaignEntities);
      adgroupDAO.save(adgroupEntities);
      creativeDAO.save(creativeEntityList);
    }
  }
  @Override
  public List<CampaignDTO> getCampaign(String userName, long accountId) {
    SystemUserDTO systemUserDTO = getSystemUser(userName);

    if (systemUserDTO == null) {
      return Collections.emptyList();
    }

    List<BaiduAccountInfoDTO> baiduAccountInfoDTOList = systemUserDTO.getBaiduAccounts();

    if (baiduAccountInfoDTOList == null || baiduAccountInfoDTOList.isEmpty()) {
      return Collections.emptyList();
    }

    BaiduAccountInfoDTO baiduAccountInfoDTO = null;
    for (BaiduAccountInfoDTO dto : baiduAccountInfoDTOList) {
      if (Long.valueOf(accountId).compareTo(dto.getId()) == 0) {
        baiduAccountInfoDTO = dto;
        break;
      }
    }

    Objects.requireNonNull(baiduAccountInfoDTO);
    Long acid = baiduAccountInfoDTO.getId();

    CommonService commonService =
        BaiduServiceSupport.getCommonService(
            baiduAccountInfoDTO.getBaiduUserName(),
            baiduAccountInfoDTO.getBaiduPassword(),
            baiduAccountInfoDTO.getToken());
    BaiduApiService apiService = new BaiduApiService(commonService);

    // 本地的推广单元
    List<CampaignDTO> campaignEntityList = Lists.newArrayList(campaignDAO.findAll());
    List<CampaignType> campaignTypes = apiService.getAllCampaign();
    List<CampaignDTO> campaignEntities = EntityConvertUtils.convertToCamEntity(campaignTypes);
    // 凤巢中的推广单元
    Map<Long, CampaignDTO> campaignEntityMap = new LinkedHashMap<>();
    for (CampaignDTO campaignEntity : campaignEntities) {
      campaignEntity.setAccountId(acid);
      campaignEntityMap.put(campaignEntity.getCampaignId(), campaignEntity);
    }

    List<CampaignDTO> sumList = new ArrayList<>(campaignEntityList);
    sumList.addAll(campaignEntities);
    for (CampaignDTO entity : sumList) {
      Long campaignId = entity.getCampaignId();
      if (campaignId == null) {
        continue;
      }
      if (campaignEntityMap.get(campaignId) != null) {
        campaignEntityMap.remove(campaignId);
      }
    }

    if (campaignEntityMap.size() == 0) {
      return Collections.emptyList();
    } else {
      List<CampaignDTO> campaignDTOList = new ArrayList<>();
      campaignEntityMap
          .values()
          .forEach(
              e -> {
                CampaignDTO campaignDTO = new CampaignDTO();
                BeanUtils.copyProperties(e, campaignDTO);
                campaignDTOList.add(campaignDTO);
              });
      return new ArrayList<>(campaignDTOList);
    }
  }
  @Override
  public void updateAccountData(String userName, long accountId, List<Long> camIds) {
    SystemUserDTO systemUserDTO = getSystemUser(userName);

    if (systemUserDTO == null) {
      return;
    }

    List<BaiduAccountInfoDTO> baiduAccountInfoDTOList = systemUserDTO.getBaiduAccounts();
    if (baiduAccountInfoDTOList == null || baiduAccountInfoDTOList.isEmpty()) {
      return;
    }

    BaiduAccountInfoDTO baiduAccountInfoDTO = null;
    for (BaiduAccountInfoDTO dto : baiduAccountInfoDTOList) {
      if (accountId == dto.getId()) {
        baiduAccountInfoDTO = dto;
        break;
      }
    }

    Objects.requireNonNull(baiduAccountInfoDTO);
    Long acid = baiduAccountInfoDTO.getId();

    CommonService commonService =
        BaiduServiceSupport.getCommonService(
            baiduAccountInfoDTO.getBaiduUserName(),
            baiduAccountInfoDTO.getBaiduPassword(),
            baiduAccountInfoDTO.getToken());
    BaiduApiService apiService = new BaiduApiService(commonService);

    // 获取账户总数据
    AccountInfoType accountInfoType = apiService.getAccountInfo();
    BeanUtils.copyProperties(accountInfoType, baiduAccountInfoDTO);

    // update account data
    updateBaiduAccountInfo(userName, accountId, baiduAccountInfoDTO);

    // 获取指定id的推广计划
    List<CampaignType> campaignTypes = apiService.getCampaignById(camIds);

    // 转换成本地系统的实体
    List<CampaignDTO> campaignEntities = EntityConvertUtils.convertToCamEntity(campaignTypes);

    List<Long> localAdgroupIds = getLocalAdgroupIds(accountId, camIds);
    List<Long> localKeywordIds = getLocalKeywordIds(accountId, localAdgroupIds);
    List<Long> localCreativeIds = getLocalCreativeIds(accountId, localAdgroupIds);

    // clear data
    clearCampaignData(accountId, camIds);
    clearAdgroupData(accountId, localAdgroupIds);
    clearKeywordData(accountId, localKeywordIds);
    clearCreativeData(accountId, localCreativeIds);

    // 凤巢返回回来的计划实体id
    List<Long> campaignIds = new ArrayList<>(campaignEntities.size());

    for (CampaignDTO campaignEntity : campaignEntities) {
      campaignEntity.setAccountId(acid);
      campaignIds.add(campaignEntity.getCampaignId());
    }
    campaignDAO.save(campaignEntities);

    List<AdgroupType> adgroupTypeList = apiService.getAllAdGroup(campaignIds);

    List<AdgroupDTO> adgroupEntities = EntityConvertUtils.convertToAdEntity(adgroupTypeList);

    List<Long> adgroupIds = new ArrayList<>(adgroupEntities.size());
    for (AdgroupDTO adgroupEntity : adgroupEntities) {
      adgroupEntity.setAccountId(acid);
      adgroupIds.add(adgroupEntity.getAdgroupId());
    }
    adgroupDAO.save(adgroupEntities);

    // 分批次请求关键词数据
    List<Long> subList = new ArrayList<>(4);
    for (int i = 1; i <= adgroupIds.size(); i++) {
      Long adgroupId = adgroupIds.get(i - 1);
      subList.add(adgroupId);

      if (i % 4 == 0) {
        List<KeywordType> keywordTypes = apiService.getAllKeyword(subList);
        List<KeywordDTO> keywordEntities = EntityConvertUtils.convertToKwEntity(keywordTypes);

        for (KeywordDTO keywordEntity : keywordEntities) {
          keywordEntity.setAccountId(acid);
        }
        keywordDAO.save(keywordEntities);
        subList.clear();
      }
    }

    if (!subList.isEmpty()) {
      List<KeywordType> keywordTypes = apiService.getAllKeyword(subList);
      List<KeywordDTO> keywordEntities = EntityConvertUtils.convertToKwEntity(keywordTypes);

      for (KeywordDTO keywordEntity : keywordEntities) {
        keywordEntity.setAccountId(acid);
      }
      keywordDAO.save(keywordEntities);
      subList.clear();
    }

    List<CreativeType> creativeTypes = apiService.getAllCreative(adgroupIds);

    List<CreativeDTO> creativeEntityList = EntityConvertUtils.convertToCrEntity(creativeTypes);

    for (CreativeDTO creativeEntity : creativeEntityList) {
      creativeEntity.setAccountId(acid);
    }
    creativeDAO.save(creativeEntityList);
  }
  @Override
  public void updateAccountData(String userName, long accountId) {
    SystemUserDTO systemUserDTO = getSystemUser(userName);
    if (systemUserDTO == null) {
      return;
    }

    List<BaiduAccountInfoDTO> baiduAccountInfoDTOList = systemUserDTO.getBaiduAccounts();

    if (baiduAccountInfoDTOList == null || baiduAccountInfoDTOList.isEmpty()) {
      return;
    }

    // 清除当前账户所有数据
    clearAccountData(accountId);

    BaiduAccountInfoDTO _dto;
    for (BaiduAccountInfoDTO baiduAccountInfoDTO : baiduAccountInfoDTOList) {

      Long aid = baiduAccountInfoDTO.getId();
      if (aid != accountId) continue;

      CommonService commonService =
          BaiduServiceSupport.getCommonService(
              baiduAccountInfoDTO.getBaiduUserName(),
              baiduAccountInfoDTO.getBaiduPassword(),
              baiduAccountInfoDTO.getToken());
      BaiduApiService apiService = new BaiduApiService(commonService);

      // 初始化账户数据
      AccountInfoType accountInfoType = apiService.getAccountInfo();
      boolean isDefault = baiduAccountInfoDTO.isDfault();
      _dto = ObjectUtils.convert(accountInfoType, BaiduAccountInfoDTO.class);
      _dto.setId(accountInfoType.getUserid());
      _dto.setBaiduUserName(baiduAccountInfoDTO.getBaiduUserName());
      _dto.setBaiduPassword(baiduAccountInfoDTO.getBaiduPassword());
      _dto.setToken(baiduAccountInfoDTO.getToken());
      _dto.setDfault(isDefault);

      // 更新账户数据
      updateBaiduAccountInfo(userName, accountId, _dto);

      // 更新推广计划数据
      List<CampaignType> campaignTypes = apiService.getAllCampaign();
      List<CampaignDTO> campaignEntities = EntityConvertUtils.convertToCamEntity(campaignTypes);
      // 查询推广单元
      List<Long> camIds = new ArrayList<>(campaignEntities.size());
      for (CampaignDTO campaignEntity : campaignEntities) {
        campaignEntity.setAccountId(aid);
        camIds.add(campaignEntity.getCampaignId());
      }
      campaignDAO.save(campaignEntities);

      // 更新推广单元数据
      List<AdgroupType> adgroupTypeList = apiService.getAllAdGroup(camIds);
      List<AdgroupDTO> adgroupEntities = EntityConvertUtils.convertToAdEntity(adgroupTypeList);
      List<Long> adgroupIds = new ArrayList<>();
      for (AdgroupDTO adgroupEntity : adgroupEntities) {
        adgroupEntity.setAccountId(aid);
        adgroupIds.add(adgroupEntity.getAdgroupId());
      }
      adgroupDAO.save(adgroupEntities);

      // 分批次请求关键词数据
      List<Long> subList = new ArrayList<>(4);
      for (int i = 1; i <= adgroupIds.size(); i++) {
        Long adgroupId = adgroupIds.get(i - 1);
        subList.add(adgroupId);

        if (i % 4 == 0) {
          List<KeywordType> keywordTypes = apiService.getAllKeyword(subList);
          List<KeywordDTO> keywordEntities = EntityConvertUtils.convertToKwEntity(keywordTypes);

          for (KeywordDTO keywordEntity : keywordEntities) {
            keywordEntity.setAccountId(aid);
          }
          keywordDAO.save(keywordEntities);
          subList.clear();
        }
      }

      if (!subList.isEmpty()) {
        List<KeywordType> keywordTypes = apiService.getAllKeyword(subList);
        List<KeywordDTO> keywordEntities = EntityConvertUtils.convertToKwEntity(keywordTypes);

        for (KeywordDTO keywordEntity : keywordEntities) {
          keywordEntity.setAccountId(aid);
        }
        keywordDAO.save(keywordEntities);
        subList.clear();
      }

      List<CreativeType> creativeTypes = apiService.getAllCreative(adgroupIds);

      List<CreativeDTO> creativeEntityList = EntityConvertUtils.convertToCrEntity(creativeTypes);

      for (CreativeDTO creativeEntity : creativeEntityList) {
        creativeEntity.setAccountId(aid);
      }
      creativeDAO.save(creativeEntityList);
    }
  }