Fix the bug of RedisAuthUser

Signed-off-by: loheagn <[email protected]>
bugitt · Sep 18, 2022 · 75cfcb5 · 75cfcb5
1 parent 0d4d5f9
commit 75cfcb5
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/internal/context/auth.go b/internal/context/auth.go
@@ -164,10 +164,16 @@ func RedisAuthUser(token string, createIfNotExist bool) (*db.User, error) {
 	if err != nil {
 		return nil, err
 	}
-	user, err := db.GetUserByStudentID(studentID)
+	user, err := db.GetUserByName(studentID)
 	if err != nil {
-		if _, ok := err.(db.ErrUserNotExist); ok && createIfNotExist {
-			return db.CreateDefaultUserByStudentID(studentID)
+		if _, ok := err.(db.ErrUserNotExist); ok {
+			if user, err = db.GetUserByStudentID(studentID); err != nil {
+				if _, ok := err.(db.ErrUserNotExist); ok && createIfNotExist {
+					return db.CreateDefaultUserByStudentID(studentID)
+				}
+				return nil, err
+			}
+			return user, nil
 		}
 		return nil, err
 	}