feat: update cache #204

model/cache.go

@@ -168,7 +168,11 @@ func CacheUpdateUserQuota(id int) error {
 	if err != nil {
 	if err != nil {
 		return err
 		return err
 	}
 	}
-	err = common.RedisSet(fmt.Sprintf("user_quota:%d", id), fmt.Sprintf("%d", quota), time.Duration(UserId2QuotaCacheSeconds)*time.Second)
+	return CacheSetUserQuota(id, quota)
+}
+
+func CacheSetUserQuota(id int, quota int) error {
+	err := common.RedisSet(fmt.Sprintf("user_quota:%d", id), fmt.Sprintf("%d", quota), time.Duration(UserId2QuotaCacheSeconds)*time.Second)
 	return err
 	return err
 }
 }
 
 

model/token.go

@@ -102,6 +102,9 @@ func GetTokenById(id int) (*Token, error) {
 	token := Token{Id: id}
 	token := Token{Id: id}
 	var err error = nil
 	var err error = nil
 	err = DB.First(&token, "id = ?", id).Error
 	err = DB.First(&token, "id = ?", id).Error
+	if err != nil {
+		go cacheSetToken(&token)
+	}
 	return &token, err
 	return &token, err
 }
 }
 
 

model/user.go

@@ -410,6 +410,11 @@ func ValidateAccessToken(token string) (user *User) {
 
 
 func GetUserQuota(id int) (quota int, err error) {
 func GetUserQuota(id int) (quota int, err error) {
 	err = DB.Model(&User{}).Where("id = ?", id).Select("quota").Find(&quota).Error
 	err = DB.Model(&User{}).Where("id = ?", id).Select("quota").Find(&quota).Error
+	if err != nil {
+		if common.RedisEnabled {
+			go CacheSetUserQuota(id, quota)
+		}
+	}
 	return quota, err
 	return quota, err
 }
 }