|
|
@@ -2,8 +2,11 @@ package com.usoftchina.saas.account.controller;
|
|
|
|
|
|
import com.usoftchina.saas.account.dto.AccountDTO;
|
|
|
import com.usoftchina.saas.account.dto.AccountRegDTO;
|
|
|
+import com.usoftchina.saas.account.dto.CompanyBaseDTO;
|
|
|
import com.usoftchina.saas.account.po.Account;
|
|
|
import com.usoftchina.saas.account.service.AccountService;
|
|
|
+import com.usoftchina.saas.account.service.CompanyService;
|
|
|
+import com.usoftchina.saas.account.vo.CompanyBaseVO;
|
|
|
import com.usoftchina.saas.base.Result;
|
|
|
import com.usoftchina.saas.exception.ExceptionCode;
|
|
|
import com.usoftchina.saas.utils.BeanMapper;
|
|
|
@@ -11,6 +14,8 @@ import com.usoftchina.saas.utils.RegexpUtils;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
/**
|
|
|
* @author yingp
|
|
|
* @date 2018/10/2
|
|
|
@@ -22,6 +27,9 @@ public class AccountController {
|
|
|
@Autowired
|
|
|
private AccountService accountService;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private CompanyService companyService;
|
|
|
+
|
|
|
/**
|
|
|
* 注册
|
|
|
*
|
|
|
@@ -71,7 +79,12 @@ public class AccountController {
|
|
|
return Result.error(ExceptionCode.USER_PWD_ERROR);
|
|
|
}
|
|
|
|
|
|
- return Result.success();
|
|
|
+ AccountDTO accountDTO = BeanMapper.map(account, AccountDTO.class);
|
|
|
+ // 绑定的公司
|
|
|
+ List<CompanyBaseVO> companyBaseVOS = companyService.findBaseByAccountId(account.getId());
|
|
|
+ accountDTO.setCompanies(BeanMapper.mapList(companyBaseVOS, CompanyBaseDTO.class));
|
|
|
+
|
|
|
+ return Result.success(accountDTO);
|
|
|
}
|
|
|
|
|
|
private Account getAccountByUsername(String username) {
|
|
|
@@ -100,6 +113,62 @@ public class AccountController {
|
|
|
}
|
|
|
|
|
|
AccountDTO accountDTO = BeanMapper.map(account, AccountDTO.class);
|
|
|
+ // 绑定的公司
|
|
|
+ List<CompanyBaseVO> companyBaseVOS = companyService.findBaseByAccountId(account.getId());
|
|
|
+ accountDTO.setCompanies(BeanMapper.mapList(companyBaseVOS, CompanyBaseDTO.class));
|
|
|
+
|
|
|
return Result.success(accountDTO);
|
|
|
}
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 个人账户绑定公司
|
|
|
+ *
|
|
|
+ * @param accountId
|
|
|
+ * @param companyId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @PostMapping("/bind/company")
|
|
|
+ public Result bindCompany(@RequestParam long accountId, @RequestParam long companyId) {
|
|
|
+ accountService.bindCompany(accountId, companyId);
|
|
|
+ return Result.success();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 解除个人账户绑定公司
|
|
|
+ *
|
|
|
+ * @param accountId
|
|
|
+ * @param companyId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @PostMapping("/unbind/company")
|
|
|
+ public Result unbindCompany(@RequestParam long accountId, @RequestParam long companyId) {
|
|
|
+ accountService.unbindCompany(accountId, companyId);
|
|
|
+ return Result.success();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 个人账户绑定角色
|
|
|
+ *
|
|
|
+ * @param accountId
|
|
|
+ * @param roleId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @PostMapping("/bind/role")
|
|
|
+ public Result bindRole(@RequestParam long accountId, @RequestParam long roleId) {
|
|
|
+ accountService.bindRole(accountId, roleId);
|
|
|
+ return Result.success();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 解除个人账户绑定角色
|
|
|
+ *
|
|
|
+ * @param accountId
|
|
|
+ * @param roleId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @PostMapping("/unbind/role")
|
|
|
+ public Result unbindRole(@RequestParam long accountId, @RequestParam long roleId) {
|
|
|
+ accountService.unbindRole(accountId, roleId);
|
|
|
+ return Result.success();
|
|
|
+ }
|
|
|
}
|