diff --git a/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/dto/CreateUserDTO.java b/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/dto/CreateUserDTO.java index 040f05b6..b4a77973 100644 --- a/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/dto/CreateUserDTO.java +++ b/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/dto/CreateUserDTO.java @@ -80,6 +80,8 @@ public class CreateUserDTO { @ApiModelProperty(value = "是否审核") private Boolean isReview; + @ApiModelProperty(value = "套餐id") + private Long templateId; @Override diff --git a/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/impl/UserServiceImpl.java b/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/impl/UserServiceImpl.java index 07dbae05..71f10ee5 100644 --- a/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/impl/UserServiceImpl.java +++ b/manage/ad-platform-management/src/main/java/com/baiye/modules/system/service/impl/UserServiceImpl.java @@ -181,6 +181,7 @@ public class UserServiceImpl implements UserService { if (flag) { CompanyDto companyDto = new CompanyDto(); companyDto.setCompanyName(userDTO.getUsername()); + companyDto.setTemplateId(userDTO.getTemplateId()); companyDto.setStatus(DefaultNumberConstants.ZERO_NUMBER); companyId = companyService.createCompany(companyDto).getId(); }