推广系统 创建企业和企业用户接口

picc-admin-server/src/main/java/com/jpsoft/picc/modules/base/controller/CompanyController.java

@@ -234,4 +234,56 @@ public class CompanyController {
 
         return msgResult;
     }
+
+
+    @ApiOperation(value="保存推广者信息")
+    @RequestMapping(value = "savePromot",method = RequestMethod.POST)
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "userName",value = "企业账户用户名", required = true, paramType = "query"),
+            @ApiImplicitParam(name = "promoter",value = "推广者帐号", required = true, paramType = "query"),
+    })
+    public MessageResult<Integer> savePromot(String userName,String promoter, @RequestAttribute String subject){
+        MessageResult<Integer> msgResult = new MessageResult<>();
+
+        try {
+            Company company = new Company();
+            company.setId(UUID.randomUUID().toString());
+            company.setName(promoter);
+            company.setCreateBy(subject);
+            company.setCreateTime(new Date());
+            company.setDelFlag(false);
+
+            int affectCount = companyService.insert(company);
+
+            int affectCount1 = 0;
+
+            if(affectCount>0){
+                CompanyUser companyUser = new CompanyUser();
+                companyUser.setId(UUID.randomUUID().toString());
+                companyUser.setUserName(userName);
+                companyUser.setCreateBy(subject);
+                companyUser.setCreateTime(new Date());
+                companyUser.setDelFlag(false);
+                companyUser.setCompanyId(company.getId());
+
+                affectCount1 = companyUserService.insert(companyUser);
+            }
+
+            if (affectCount1 > 0) {
+                msgResult.setResult(true);
+                msgResult.setData(affectCount1);
+            } else {
+                msgResult.setResult(false);
+                msgResult.setMessage("数据库添加失败");
+            }
+        }
+        catch(Exception ex){
+            logger.error(ex.getMessage(),ex);
+
+            msgResult.setResult(false);
+            msgResult.setMessage(ex.getMessage());
+        }
+
+        return msgResult;
+    }
 }