Files
geo/server/internal/shared/objectstorage/reloadable.go
T
root 618399f86d feat(server): hot-reload config store and reloadable infra clients
- Replace single Load() with a watching Store that re-reads config files
  (and config.local.yaml) and fans out a ReloadEvent with a per-field diff
  so consumers can decide whether the change is hot-applicable or requires
  a process restart.
- Wrap llm, retrieval, vector store, and object storage clients in
  Reloadable* shells so the bootstrap can swap their underlying impls when
  config changes without re-instantiating handlers.
- Make jwt.Manager and ops TokenIssuer mutable under a lock so secrets and
  TTLs can be rotated live; thread default plan code through a setter on
  the ops AdminUserService.
- Wire ConfigStore through bootstrap and every cmd/main.go, scheduler /
  worker / tenant-api / ops-api start the watcher; services and handlers
  take a config.Provider so they always read current values for things
  like generation.stream_enabled, scheduler dispatch, retrieval, etc.
- Switch shared/config decoding off viper to a Kratos-derived runtime
  package so env placeholders (\${VAR:default}) resolve consistently and
  the same source machinery powers both the loader and the watcher.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-01 16:01:23 +08:00

61 lines
1.4 KiB
Go

package objectstorage
import (
"context"
"sync"
)
type ReloadableClient struct {
mu sync.RWMutex
current Client
}
func NewReloadableClient(current Client) *ReloadableClient {
return &ReloadableClient{current: current}
}
func (c *ReloadableClient) Update(next Client) {
if c == nil || next == nil {
return
}
c.mu.Lock()
defer c.mu.Unlock()
c.current = next
}
func (c *ReloadableClient) Validate() error {
return c.snapshot().Validate()
}
func (c *ReloadableClient) PutBytes(ctx context.Context, objectKey string, content []byte, contentType string) error {
return c.snapshot().PutBytes(ctx, objectKey, content, contentType)
}
func (c *ReloadableClient) GetBytes(ctx context.Context, objectKey string) ([]byte, error) {
return c.snapshot().GetBytes(ctx, objectKey)
}
func (c *ReloadableClient) Exists(ctx context.Context, objectKey string) (bool, error) {
return c.snapshot().Exists(ctx, objectKey)
}
func (c *ReloadableClient) Delete(ctx context.Context, objectKey string) error {
return c.snapshot().Delete(ctx, objectKey)
}
func (c *ReloadableClient) PublicURL(objectKey string) (string, error) {
return c.snapshot().PublicURL(objectKey)
}
func (c *ReloadableClient) snapshot() Client {
if c == nil {
return disabledClient{reason: "object storage client is nil"}
}
c.mu.RLock()
defer c.mu.RUnlock()
if c.current == nil {
return disabledClient{reason: "object storage client is not initialized"}
}
return c.current
}