package blobovniczatree import ( "context" "errors" "os" "path/filepath" "strings" "git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs" "go.uber.org/zap" ) var errFailedToChangeExtensionReadOnly = errors.New("failed to change blobovnicza extension: read only mode") // Open opens blobovnicza tree. func (b *Blobovniczas) Open(readOnly bool) error { b.readOnly = readOnly b.metrics.SetMode(readOnly) b.openManagers() return nil } // Init initializes blobovnicza tree. // // Should be called exactly once. func (b *Blobovniczas) Init() error { b.log.Debug(logs.BlobovniczatreeInitializingBlobovniczas) b.log.Debug(logs.BlobovniczaTreeFixingFileExtensions) if err := b.addDBExtensionToDBs(b.rootPath, 0); err != nil { b.log.Error(logs.BlobovniczaTreeFixingFileExtensionsFailed, zap.Error(err)) return err } b.log.Debug(logs.BlobovniczaTreeFixingFileExtensionsCompletedSuccessfully) if b.readOnly { b.log.Debug(logs.BlobovniczatreeReadonlyModeSkipBlobovniczasInitialization) return nil } return b.iterateLeaves(context.TODO(), func(p string) (bool, error) { shBlz := b.getBlobovniczaWithoutCaching(p) _, err := shBlz.Open() if err != nil { return true, err } defer shBlz.Close() b.log.Debug(logs.BlobovniczatreeBlobovniczaSuccessfullyInitializedClosing, zap.String("id", p)) return false, nil }) } func (b *Blobovniczas) openManagers() { b.commondbManager.Open() // order important b.activeDBManager.Open() b.dbCache.Open() } // Close implements common.Storage. func (b *Blobovniczas) Close() error { b.dbCache.Close() // order important b.activeDBManager.Close() b.commondbManager.Close() return nil } // returns blobovnicza with path p // // If blobovnicza is already cached, instance from cache is returned w/o changes. func (b *Blobovniczas) getBlobovnicza(p string) *sharedDB { return b.dbCache.GetOrCreate(p) } func (b *Blobovniczas) getBlobovniczaWithoutCaching(p string) *sharedDB { return b.commondbManager.GetByPath(p) } func (b *Blobovniczas) addDBExtensionToDBs(path string, depth uint64) error { entries, err := os.ReadDir(path) if os.IsNotExist(err) && depth == 0 { return nil } for _, entry := range entries { if entry.IsDir() { if err := b.addDBExtensionToDBs(filepath.Join(path, entry.Name()), depth+1); err != nil { return err } continue } if strings.HasSuffix(entry.Name(), dbExtension) { continue } if b.readOnly { return errFailedToChangeExtensionReadOnly } sourcePath := filepath.Join(path, entry.Name()) targetPath := filepath.Join(path, entry.Name()+dbExtension) b.log.Debug(logs.BlobovniczaTreeFixingFileExtensionForFile, zap.String("source", sourcePath), zap.String("target", targetPath)) if err := os.Rename(sourcePath, targetPath); err != nil { b.log.Error(logs.BlobovniczaTreeFixingFileExtensionFailed, zap.String("source", sourcePath), zap.String("target", targetPath), zap.Error(err)) return err } b.log.Debug(logs.BlobovniczaTreeFixingFileExtensionCompletedSuccessfully, zap.String("source", sourcePath), zap.String("target", targetPath)) } return nil }