3

src/main/java/com/book/demo/controller/DemoController.java

@@ -1,33 +0,0 @@
-package com.book.demo.controller;
-
-import com.book.demo.entity.User;
-import com.book.demo.mapper.UserMapper;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.*;
-
-@RestController
-public class DemoController {
-    @Autowired
-     private UserMapper userMapper;
-
-    //返回JSON
-    @ResponseBody
-    @RequestMapping("/query")
-    public User queryUserByusername(@RequestParam("username") String username){
-        User user=userMapper.queryUserByusername(username);
-        return user;
-    }
-
-    @ResponseBody
-    @PostMapping("/login")
-    public Integer login(@RequestBody User user){
-        System.out.println(user);
-        User user1=userMapper.queryUserByusername(user.getUsername());
-        if(user1==null)
-            return -1;
-        if(!user1.getPassword().trim().equals(user.getPassword().trim()))
-            return -2;
-        else
-            return 0;
-    }
-}

src/main/java/com/book/demo/controller/UserController.java

@@ -0,0 +1,88 @@
+package com.book.demo.controller;
+
+import com.book.demo.entity.User;
+import com.book.demo.mapper.UserMapper;
+import com.book.demo.services.Result;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.*;
+
+@Controller
+public class UserController {
+    @Autowired
+     private UserMapper userMapper;
+    @Autowired
+    private Result res;
+
+    //返回JSON
+    @ResponseBody
+    @RequestMapping("/query")
+    public Result queryUserByusername(@RequestParam("username") String username){
+        User user=userMapper.queryUserByusername(username);
+        if(user!=null) {
+            res.setCode(0);
+            res.setMsg("查找成功");
+            res.data.add(user);
+        }
+        else {
+            res.setCode(-1);
+            res.setMsg("数据非法,无此用户");
+            res.data=null;
+        }
+        return res;
+    }
+
+    @ResponseBody
+    @PostMapping("/login")
+    public Result login(@RequestBody User user){
+        System.out.println(user);
+        User user1=userMapper.queryUserByusername(user.getUsername());
+        if(user1==null){
+            res.setCode(-1);
+            res.setMsg("用户不存在");
+            res.data=null;
+        }
+        else if(!user1.getPassword().trim().equals(user.getPassword().trim()))
+        {
+            res.setCode(-2);
+            res.setMsg("密码错误");
+            res.data=null;
+        }
+        else if(user.getKind()==0){
+            res.setCode(0);
+            res.setMsg("用户登录");
+            res.data.add(user1);
+        }
+        else{
+            res.setCode(1);
+            res.setMsg("管理员登录");
+            res.data.add(user1);
+        }
+        return res;
+    }
+
+
+    @ResponseBody
+    @PostMapping("/register")
+    public Result register(@RequestBody User user) {
+        User user1=userMapper.queryUserByusername(user.getUsername());
+        if(user1!=null){
+            res.setCode(-1);
+            res.setMsg("用户已存在");
+            res.data=null;
+        }
+        else{
+            user1=user;
+            user1.setKind(0);
+            if(userMapper.InsertUser(user1)>0){
+                res.setCode(1);
+                res.setMsg("注册成功");
+            }else{
+                res.setCode(-2);
+                res.setMsg("注册失败，稍后再试");
+            }
+            res.data=null;
+        }
+        return res;
+    }
+}

src/main/java/com/book/demo/mapper/UserMapper.java

@@ -1,6 +1,7 @@
 package com.book.demo.mapper;
 
 import com.book.demo.entity.User;
+import org.apache.ibatis.annotations.Insert;
 import org.apache.ibatis.annotations.Mapper;
 import org.apache.ibatis.annotations.Select;
 import org.springframework.stereotype.Repository;
@@ -12,4 +13,7 @@ public interface UserMapper {
     //通过username查询用户信息
     @Select("select * from user where username=#{username}")
     User queryUserByusername(String username);
+
+    @Insert("insert into user values(#{username},#{password},#{kind})")
+    int InsertUser(User user);
 }

src/main/java/com/book/demo/services/Result.java

@@ -0,0 +1,42 @@
+package com.book.demo.services;
+
+import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+import java.util.ArrayList;
+import java.util.List;
+
+@Service
+public class Result {
+    public int code;
+    public String msg;
+    public List data;
+
+    Result(){
+        data=new ArrayList();
+    }
+
+    public int getCode() {
+        return code;
+    }
+
+    public void setCode(int code) {
+        this.code = code;
+    }
+
+    public String getMsg() {
+        return msg;
+    }
+
+    public void setMsg(String msg) {
+        this.msg = msg;
+    }
+
+    public List getData() {
+        return data;
+    }
+
+    public void setData(List data) {
+        this.data = data;
+    }
+}