From 6ffb4ccf2ef9f12cf2c16672b2fef553ed5673a0 Mon Sep 17 00:00:00 2001
From: markma <750975972@qq.com>
Date: Fri, 28 Jan 2022 17:04:44 +0800
Subject: [PATCH] =?UTF-8?q?09=20=E6=9D=83=E9=99=90=E7=AE=A1=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/UserHandler.java               | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/markma/leave_manager_spb/controller/UserHandler.java b/src/main/java/com/markma/leave_manager_spb/controller/UserHandler.java
index dfa4e60..69f6c2f 100644
--- a/src/main/java/com/markma/leave_manager_spb/controller/UserHandler.java
+++ b/src/main/java/com/markma/leave_manager_spb/controller/UserHandler.java
@@ -1,9 +1,12 @@
 package com.markma.leave_manager_spb.controller;
 
-import com.markma.leave_manager_spb.repository.UserRepository;
 import com.markma.leave_manager_spb.entity.User;
+import com.markma.leave_manager_spb.repository.UserRepository;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.*;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
 
 import java.util.List;
 
@@ -29,12 +32,23 @@ public class UserHandler {
         for (int i = 0; i < users.size(); i++) {
             if (users.get(i).getUsername().equals(username)) {
                 if (users.get(i).getPassword().equals(password)) {
-                    return users.get(i).getId()+" "+users.get(i).getType();
-                } else{
+                    return users.get(i).getId() + " " + users.get(i).getType();
+                } else {
                     return "error:wrong password";
                 }
             }
         }
         return "error:wrong username";
     }
+
+    @GetMapping("/checkUseridAndType/{userid}/{type}")
+    public String checkUseridAndType(@PathVariable("userid") Integer userid, @PathVariable("type") String type) {
+        List<User> users = userRepository.findAll();
+        for (int i = 0; i < users.size(); i++) {
+            if (users.get(i).getType().equals(type) && users.get(i).getId().equals(userid)) {
+                return "success";
+            }
+        }
+        return "failed";
+    }
 }