[fix] #95

homenoc · Jul 6, 2021 · 7653dd5 · 7653dd5
1 parent 2d00ffd
commit 7653dd5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pkg/api/core/group/info/v0/info.go b/pkg/api/core/group/info/v0/info.go
@@ -25,6 +25,7 @@ func Get(c *gin.Context) {
 	authResult := auth.UserAuthentication(core.Token{UserToken: userToken, AccessToken: accessToken})
 	if authResult.Err != nil {
 		c.JSON(http.StatusUnauthorized, common.Error{Error: authResult.Err.Error()})
+		return
 	}
 
 	// User

diff --git a/pkg/api/core/token/v0/token.go b/pkg/api/core/token/v0/token.go
@@ -110,6 +110,11 @@ func Delete(c *gin.Context) {
 		return
 	}
 
+	if len(result.Token) == 0 {
+		c.JSON(http.StatusUnauthorized, common.Error{Error: "Error: Unauthorized..."})
+		return
+	}
+
 	logging.WriteLog(strconv.Itoa(int(result.Token[0].User.ID))+"-"+result.Token[0].User.Name, "Logout")
 
 	if err := dbToken.Delete(&core.Token{Model: gorm.Model{ID: result.Token[0].ID}}); err != nil {