66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"sync"
|
|
|
|
"kra/internal/biz"
|
|
"kra/internal/conf"
|
|
datastorage "kra/internal/data/storage"
|
|
)
|
|
|
|
type reloadableStorage struct {
|
|
mu sync.RWMutex
|
|
current biz.FileStorage
|
|
}
|
|
|
|
func NewFileStorage(data *Data) (biz.FileStorage, error) {
|
|
storage, err := buildFileStorage(data.runtime.Admin())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
wrapper := &reloadableStorage{current: storage}
|
|
data.storage = wrapper
|
|
return wrapper, nil
|
|
}
|
|
|
|
func buildFileStorage(config *conf.AdminBackend) (biz.FileStorage, error) {
|
|
return datastorage.New(config)
|
|
}
|
|
func (s *reloadableStorage) replace(storage biz.FileStorage) {
|
|
s.mu.Lock()
|
|
s.current = storage
|
|
s.mu.Unlock()
|
|
}
|
|
func (s *reloadableStorage) Put(ctx context.Context, name string, reader io.Reader) (*biz.StoredFile, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.current.Put(ctx, name, reader)
|
|
}
|
|
func (s *reloadableStorage) Open(ctx context.Context, name string) (io.ReadCloser, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.current.Open(ctx, name)
|
|
}
|
|
func (s *reloadableStorage) Delete(ctx context.Context, name string) error {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.current.Delete(ctx, name)
|
|
}
|
|
func (s *reloadableStorage) Compose(ctx context.Context, names []string, destination string) (*biz.StoredFile, string, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.current.Compose(ctx, names, destination)
|
|
}
|
|
func (s *reloadableStorage) DeletePrefix(ctx context.Context, prefix string) error {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.current.DeletePrefix(ctx, prefix)
|
|
}
|
|
func (s *reloadableStorage) List(ctx context.Context, prefix, cursor string, limit int) ([]*biz.StoredFile, string, bool, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
return s.current.List(ctx, prefix, cursor, limit)
|
|
}
|