|
@@ -0,0 +1,87 @@
|
|
|
+package cn.minbb.iot.controller.rest;
|
|
|
+
|
|
|
+import cn.minbb.iot.data.ResponseResult;
|
|
|
+import cn.minbb.iot.model.User;
|
|
|
+import cn.minbb.iot.service.UserService;
|
|
|
+import com.alibaba.fastjson.JSONObject;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.web.bind.annotation.PostMapping;
|
|
|
+import org.springframework.web.bind.annotation.RequestBody;
|
|
|
+import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
+import org.springframework.web.bind.annotation.RestController;
|
|
|
+
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
|
+
|
|
|
+@RestController
|
|
|
+@RequestMapping("app/user")
|
|
|
+public class UserController {
|
|
|
+
|
|
|
+ private UserService userService;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ public UserController(UserService userService) {
|
|
|
+ this.userService = userService;
|
|
|
+ }
|
|
|
+
|
|
|
+ @PostMapping("login")
|
|
|
+ public ResponseResult<User> userLogin(@RequestBody String data) {
|
|
|
+ ResponseResult<User> result = null;
|
|
|
+ User userClient = JSONObject.parseObject(data, User.class);
|
|
|
+ if (null != userClient) {
|
|
|
+ User user = userService.findUserByUsername(userClient.getUsername());
|
|
|
+ if (user == null) {
|
|
|
+ // 用户不存在
|
|
|
+ result = ResponseResult.ok(false).code(0).message("用户不存在").dataNone();
|
|
|
+ } else {
|
|
|
+ if (userClient.getPassword().equals(user.getPassword())) {
|
|
|
+ // OK
|
|
|
+ result = ResponseResult.ok(true).code(1).message("登录成功").data(user);
|
|
|
+ } else {
|
|
|
+ // 密码错误
|
|
|
+ result = ResponseResult.ok(false).code(0).message("密码错误").dataNone();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+ }
|
|
|
+
|
|
|
+ @PostMapping(value = "register")
|
|
|
+ public ResponseResult<User> register(@RequestBody String userJSON, HttpServletRequest request) {
|
|
|
+ ResponseResult<User> result = new ResponseResult<>();
|
|
|
+ User userClient = JSONObject.parseObject(userJSON, User.class);
|
|
|
+ if (userClient != null) {
|
|
|
+ String username = userClient.getUsername();
|
|
|
+ if (userService.findUserByUsername(username) == null) {
|
|
|
+ // 用户不存在 - 可以注册
|
|
|
+ User u = new User(userClient.getName(), username, userClient.getPassword());
|
|
|
+// Set<UserRole> userRoleSet = new HashSet<>();
|
|
|
+// userRoleSet.add(userRoleService.findOneByRole(UserRole.Role.USER));
|
|
|
+// u.setUserRoleSet(userRoleSet);
|
|
|
+ User user = userService.saveOne(u);
|
|
|
+ if (user != null) {
|
|
|
+ // 注册成功
|
|
|
+ result.setCode(1);
|
|
|
+ result.setSuccess(true);
|
|
|
+ result.setMessage("注册成功");
|
|
|
+ result.setData(user);
|
|
|
+ } else {
|
|
|
+ // 注册失败
|
|
|
+ result.setCode(0);
|
|
|
+ result.setSuccess(false);
|
|
|
+ result.setMessage("注册失败");
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ // 用户存在 - 不可以注册
|
|
|
+ result.setCode(0);
|
|
|
+ result.setSuccess(false);
|
|
|
+ result.setMessage("用户已存在");
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ // 错误的注册请求
|
|
|
+ result.setCode(0);
|
|
|
+ result.setSuccess(false);
|
|
|
+ result.setMessage("错误的注册请求");
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+ }
|
|
|
+}
|