Merge remote-tracking branch 'origin/feat/1.2.0/img' into test

This commit is contained in:
LinkinStars
2023-10-11 15:05:35 +08:00
3 changed files with 19 additions and 0 deletions
+2
View File
@@ -216,6 +216,8 @@ func (uc *UserController) UserLogout(ctx *gin.Context) {
}
_ = uc.authService.RemoveUserCacheInfo(ctx, accessToken)
_ = uc.authService.RemoveAdminUserCacheInfo(ctx, accessToken)
visitToken, _ := ctx.Cookie(constant.UserVisitCookiesCacheKey)
_ = uc.authService.RemoveUserVisitCacheInfo(ctx, visitToken)
handler.HandleResponse(ctx, nil, nil)
}
+9
View File
@@ -86,6 +86,15 @@ func (ar *authRepo) RemoveUserCacheInfo(ctx context.Context, accessToken string)
return nil
}
// RemoveUserVisitCacheInfo remove visit token cache
func (ar *authRepo) RemoveUserVisitCacheInfo(ctx context.Context, visitToken string) (err error) {
err = ar.data.Cache.Del(ctx, constant.UserVisitTokenCacheKey+visitToken)
if err != nil {
return errors.InternalServer(reason.DatabaseError).WithError(err).WithStack()
}
return nil
}
// SetUserStatus set user status
func (ar *authRepo) SetUserStatus(ctx context.Context, userID string, userInfo *entity.UserCacheInfo) (err error) {
userInfoCache, err := json.Marshal(userInfo)
+8
View File
@@ -14,6 +14,7 @@ type AuthRepo interface {
SetUserCacheInfo(ctx context.Context, accessToken, visitToken string, userInfo *entity.UserCacheInfo) error
GetUserVisitCacheInfo(ctx context.Context, visitToken string) (accessToken string, err error)
RemoveUserCacheInfo(ctx context.Context, accessToken string) (err error)
RemoveUserVisitCacheInfo(ctx context.Context, visitToken string) (err error)
SetUserStatus(ctx context.Context, userID string, userInfo *entity.UserCacheInfo) (err error)
GetUserStatus(ctx context.Context, userID string) (userInfo *entity.UserCacheInfo, err error)
RemoveUserStatus(ctx context.Context, userID string) (err error)
@@ -96,6 +97,13 @@ func (as *AuthService) RemoveUserCacheInfo(ctx context.Context, accessToken stri
return as.authRepo.RemoveUserCacheInfo(ctx, accessToken)
}
func (as *AuthService) RemoveUserVisitCacheInfo(ctx context.Context, visitToken string) (err error) {
if len(visitToken) > 0 {
return as.authRepo.RemoveUserVisitCacheInfo(ctx, visitToken)
}
return nil
}
// AddUserTokenMapping add user token mapping
func (as *AuthService) AddUserTokenMapping(ctx context.Context, userID, accessToken string) (err error) {
return as.authRepo.AddUserTokenMapping(ctx, userID, accessToken)