|
@@ -3,9 +3,11 @@ package cn.minbb.job.controller.rest;
|
|
|
import cn.minbb.job.data.Const;
|
|
|
import cn.minbb.job.data.ResponseResult;
|
|
|
import cn.minbb.job.enumerate.Role;
|
|
|
+import cn.minbb.job.model.Company;
|
|
|
import cn.minbb.job.model.User;
|
|
|
import cn.minbb.job.model.UserRole;
|
|
|
import cn.minbb.job.model.vo.UserVo;
|
|
|
+import cn.minbb.job.service.CompanyService;
|
|
|
import cn.minbb.job.service.UserRoleService;
|
|
|
import cn.minbb.job.service.UserService;
|
|
|
import com.alibaba.fastjson.JSON;
|
|
@@ -24,10 +26,12 @@ import java.util.Set;
|
|
|
@RestController
|
|
|
public class SignController {
|
|
|
|
|
|
+ private final CompanyService companyService;
|
|
|
private final UserService userService;
|
|
|
private final UserRoleService userRoleService;
|
|
|
|
|
|
- public SignController(UserService userService, UserRoleService userRoleService) {
|
|
|
+ public SignController(CompanyService companyService, UserService userService, UserRoleService userRoleService) {
|
|
|
+ this.companyService = companyService;
|
|
|
this.userService = userService;
|
|
|
this.userRoleService = userRoleService;
|
|
|
}
|
|
@@ -41,14 +45,7 @@ public class SignController {
|
|
|
if (userService.existsByUsername(username)) {
|
|
|
return ResponseResult.<UserVo>ok(false).code(Const.ReturnCode.FAILED).message("用户名已存在").build();
|
|
|
} else {
|
|
|
- User user = new User();
|
|
|
- user.setName(name);
|
|
|
- user.setUsername(username);
|
|
|
- user.setPassword(password);
|
|
|
- user.setIsAccountNonExpired(Boolean.TRUE);
|
|
|
- user.setIsAccountNonLocked(Boolean.TRUE);
|
|
|
- user.setIsCredentialsNonExpired(Boolean.TRUE);
|
|
|
- user.setIsEnabled(Boolean.TRUE);
|
|
|
+ User user = new User(name, username, password);
|
|
|
Set<UserRole> userRoleSet = new HashSet<>(1);
|
|
|
userRoleSet.add(userRoleService.findOneByName(Role.USER.name()));
|
|
|
user.setUserRoleSet(userRoleSet);
|
|
@@ -61,6 +58,25 @@ public class SignController {
|
|
|
@PostMapping("/sign-up-company")
|
|
|
public ResponseResult<UserVo> signUpCompany(@RequestBody Map<String, String> params) {
|
|
|
log.info("注册企业用户入参:{}", params);
|
|
|
- return null;
|
|
|
+ String name = params.get("name");
|
|
|
+ String introduction = params.get("introduction");
|
|
|
+ String username = params.get("username");
|
|
|
+ String password = params.get("password");
|
|
|
+ if (userService.existsByUsername(username)) {
|
|
|
+ return ResponseResult.<UserVo>ok(false).code(Const.ReturnCode.FAILED).message("用户名已存在").build();
|
|
|
+ } else {
|
|
|
+ User user = new User(name, username, password);
|
|
|
+ Set<UserRole> userRoleSet = new HashSet<>(1);
|
|
|
+ userRoleSet.add(userRoleService.findOneByName(Role.COMPANY.name()));
|
|
|
+ user.setUserRoleSet(userRoleSet);
|
|
|
+ Company company = new Company();
|
|
|
+ company.setName(name);
|
|
|
+ company.setIntroduction(introduction);
|
|
|
+ company.setIsEnabled(Boolean.FALSE);
|
|
|
+ user.setCompany(company);
|
|
|
+ user = userService.saveOne(user);
|
|
|
+ UserVo userVo = JSON.parseObject(JSON.toJSONString(user), UserVo.class);
|
|
|
+ return ResponseResult.<UserVo>ok(true).code(Const.ReturnCode.SUCCESS).message(Const.ReturnMessage.SUCCESS).data(userVo).build();
|
|
|
+ }
|
|
|
}
|
|
|
}
|