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