修改推广码

src/main/java/com/edgec/browserbackend/account/controller/AdministratorController.java

@@ -331,6 +331,25 @@ public class AdministratorController {
         }
         return resultDto;
     }
+
+    //获取用户企业认证的信息
+    @PreAuthorize(Securitys.ADMIN_EL)
+    @RequestMapping(path = "/0xadministrator/promotioncode/add", method = RequestMethod.PUT)
+    public ResultDto addPromotionCode(Principal principal, @RequestParam("username") String username, @RequestParam("promotionCode") String promotionCode) {
+        ResultDto resultDto = new ResultDto();
+        try {
+            administratorService.addPromotionCode(username, promotionCode);
+            resultDto.setStatus(0);
+        }catch (ClientRequestException e) {
+            resultDto.setStatus(-1);
+            Map<String, Object> statusInfo = new HashMap<>();
+            statusInfo.put("code", e.getErrorCode());
+            statusInfo.put("message", e.getMessage());
+            resultDto.setStatusInfo(statusInfo);
+        }
+        return resultDto;
+    }
+
 }
 
 

src/main/java/com/edgec/browserbackend/account/service/AdministratorService.java

@@ -48,4 +48,5 @@ public interface AdministratorService {
     //迁移userbilling到userprepaidbilling
     boolean userbillingTransfer();
 
+    void addPromotionCode(String username, String promotionCode);
 }

src/main/java/com/edgec/browserbackend/account/service/impl/AdministratorServiceImpl.java

@@ -298,6 +298,15 @@ public class AdministratorServiceImpl implements AdministratorService {
         userPrePaidBillingRepository.save(userPrePaidBilling);
 
     }
+
+    @Override
+    public void addPromotionCode(String username, String promotionCode) {
+        Account account = accountRepository.findByName(username);
+        if (account == null)
+            throw new ClientRequestException(AccountErrorCode.NAMENOTEXIST);
+        account.setPromotionCode(promotionCode);
+        accountRepository.save(account);
+    }
 }