mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-02 06:56:36 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package storage provides upload storage backend operations and migration state.
|
|
package storage
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/plugins/domain/upload/shared"
|
|
"github.com/Rain-kl/Wavelet/plugins/drivers/driver_asynq_worker"
|
|
"github.com/Rain-kl/Wavelet/plugins/infra/storage/objectstore"
|
|
)
|
|
|
|
// MigrationAccessState captures cached migration maintenance state.
|
|
type MigrationAccessState struct {
|
|
ReadOnly bool
|
|
Target objectstore.Config
|
|
HasTarget bool
|
|
TargetErr error
|
|
LoadErr error
|
|
}
|
|
|
|
var (
|
|
migrationAccessMu sync.RWMutex
|
|
migrationAccessCached MigrationAccessState
|
|
migrationAccessValid bool
|
|
migrationAccessCheckedAt time.Time
|
|
)
|
|
|
|
// ResetMigrationAccessCache clears the in-process migration access cache.
|
|
func ResetMigrationAccessCache() {
|
|
migrationAccessMu.Lock()
|
|
migrationAccessValid = false
|
|
migrationAccessMu.Unlock()
|
|
}
|
|
|
|
// LoadMigrationAccessState returns cached migration maintenance state.
|
|
func LoadMigrationAccessState(ctx context.Context) MigrationAccessState {
|
|
migrationAccessMu.RLock()
|
|
if migrationAccessValid && time.Since(migrationAccessCheckedAt) < time.Duration(shared.AccessCacheTTL)*time.Second {
|
|
state := migrationAccessCached
|
|
migrationAccessMu.RUnlock()
|
|
return state
|
|
}
|
|
migrationAccessMu.RUnlock()
|
|
|
|
migrationAccessMu.Lock()
|
|
defer migrationAccessMu.Unlock()
|
|
|
|
if migrationAccessValid && time.Since(migrationAccessCheckedAt) < time.Duration(shared.AccessCacheTTL)*time.Second {
|
|
return migrationAccessCached
|
|
}
|
|
|
|
migrationAccessCached = buildMigrationAccessState(ctx)
|
|
migrationAccessValid = true
|
|
migrationAccessCheckedAt = time.Now()
|
|
return migrationAccessCached
|
|
}
|
|
|
|
func buildMigrationAccessState(ctx context.Context) MigrationAccessState {
|
|
execution, ok, err := LatestMigrationExecution(ctx)
|
|
if err != nil {
|
|
return MigrationAccessState{LoadErr: err, ReadOnly: true}
|
|
}
|
|
if !ok {
|
|
return MigrationAccessState{}
|
|
}
|
|
|
|
state := MigrationAccessState{
|
|
ReadOnly: execution.Status != driver_asynq_worker.TaskExecutionStatusSucceeded,
|
|
}
|
|
if execution.Status == driver_asynq_worker.TaskExecutionStatusSucceeded {
|
|
return state
|
|
}
|
|
|
|
target, err := ParseMigrationTargetConfig(ctx, []byte(execution.Payload))
|
|
if err != nil {
|
|
state.TargetErr = err
|
|
return state
|
|
}
|
|
|
|
state.Target = target
|
|
state.HasTarget = true
|
|
return state
|
|
}
|