fork-it/gpt-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 分组管理内存集群同步缓存

Browse Source
This commit is contained in:
tbphp
2025-07-09 23:09:25 +08:00
parent ae4644f287
commit 01fff0d70c
5 changed files with 107 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/handler/group_handler.go
View File

@@ -210,6 +210,9 @@ func (s *Server) CreateGroup(c *gin.Context) {
return
}
if err := s.GroupManager.Invalidate(); err != nil {
logrus.WithContext(c.Request.Context()).WithError(err).Error("failed to invalidate group cache")
}
response.Success(c, s.newGroupResponse(&group))
}
@@ -341,6 +344,9 @@ func (s *Server) UpdateGroup(c *gin.Context) {
return
}
if err := s.GroupManager.Invalidate(); err != nil {
logrus.WithContext(c.Request.Context()).WithError(err).Error("failed to invalidate group cache")
}
response.Success(c, s.newGroupResponse(&group))
}
@@ -471,6 +477,9 @@ func (s *Server) DeleteGroup(c *gin.Context) {
return
}
if err := s.GroupManager.Invalidate(); err != nil {
logrus.WithContext(c.Request.Context()).WithError(err).Error("failed to invalidate group cache")
}
response.Success(c, gin.H{"message": "Group and associated keys deleted successfully"})
}

3
internal/handler/handler.go
View File

@@ -20,6 +20,7 @@ type Server struct {
DB *gorm.DB
config types.ConfigManager
SettingsManager *config.SystemSettingsManager
GroupManager *services.GroupManager
KeyManualValidationService *services.KeyManualValidationService
TaskService *services.TaskService
KeyService *services.KeyService
@@ -32,6 +33,7 @@ type NewServerParams struct {
DB *gorm.DB
Config types.ConfigManager
SettingsManager *config.SystemSettingsManager
GroupManager *services.GroupManager
KeyManualValidationService *services.KeyManualValidationService
TaskService *services.TaskService
KeyService *services.KeyService
@@ -44,6 +46,7 @@ func NewServer(params NewServerParams) *Server {
DB: params.DB,
config: params.Config,
SettingsManager: params.SettingsManager,
GroupManager: params.GroupManager,
KeyManualValidationService: params.KeyManualValidationService,
TaskService: params.TaskService,
KeyService: params.KeyService,