mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
6a8f9a7aea
- Use RWMutex for disk cache reads and singleflight for WebP cache misses - Cache migration read-only state and file access whitelist with pub/sub invalidation - Add w_upload_stats incremental counters updated on upload/delete - Add w_uploads composite indexes and goose backfill migrations - Document performance analysis in docs/PERFORMANCE.md - Fix RegisterCustomRoutes to accept apiV1Router parameter
66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package upload
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
"github.com/Rain-kl/Wavelet/internal/storage"
|
|
"github.com/Rain-kl/Wavelet/pkg/logger"
|
|
)
|
|
|
|
// StorageReadOnly checks if the storage system is in read-only maintenance mode.
|
|
func StorageReadOnly(ctx context.Context) bool {
|
|
state := loadMigrationAccessState(ctx)
|
|
if state.loadErr != nil {
|
|
logger.ErrorF(ctx, "读取存储维护状态失败: %v", state.loadErr)
|
|
return true
|
|
}
|
|
return state.readOnly
|
|
}
|
|
|
|
func openStoredObject(ctx context.Context, upload *model.Upload) (*storage.Object, error) {
|
|
driver := storage.Driver(upload.StorageDriver)
|
|
if driver == "" {
|
|
driver = storage.DriverLocal
|
|
}
|
|
backend, err := backendForStoredDriver(ctx, driver)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return backend.Get(ctx, upload.FilePath)
|
|
}
|
|
|
|
func backendForStoredDriver(ctx context.Context, driver storage.Driver) (storage.Backend, error) {
|
|
backend, err := storage.ForDriver(ctx, driver)
|
|
if err == nil {
|
|
return backend, nil
|
|
}
|
|
|
|
target, ok, targetErr := currentMigrationTargetConfig(ctx)
|
|
if targetErr != nil {
|
|
return nil, targetErr
|
|
}
|
|
if ok && target.Driver == driver {
|
|
return storage.NewBackend(ctx, target, driver)
|
|
}
|
|
return nil, fmt.Errorf("storage configuration for driver %q is unavailable", driver)
|
|
}
|
|
|
|
func currentMigrationTargetConfig(ctx context.Context) (storage.Config, bool, error) {
|
|
state := loadMigrationAccessState(ctx)
|
|
if state.loadErr != nil {
|
|
return storage.Config{}, false, state.loadErr
|
|
}
|
|
if state.targetErr != nil {
|
|
return storage.Config{}, false, state.targetErr
|
|
}
|
|
if !state.hasTarget {
|
|
return storage.Config{}, false, nil
|
|
}
|
|
return state.target, true, nil
|
|
}
|