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