package meta import ( "strings" "go.etcd.io/bbolt" ) func (db *DB) CleanUp() error { return db.boltDB.Update(func(tx *bbolt.Tx) error { return tx.ForEach(func(name []byte, b *bbolt.Bucket) error { switch { case isFKBTBucket(name): cleanUpFKBT(tx, name, b) case isListBucket(name): cleanUpListBucket(tx, name, b) default: cleanUpUniqueBucket(tx, name, b) } return nil }) }) } func isFKBTBucket(name []byte) bool { bucketName := string(name) switch { case strings.Contains(bucketName, userAttributePostfix), strings.Contains(bucketName, ownerPostfix): return true default: return false } } func isListBucket(name []byte) bool { bucketName := string(name) switch { case strings.Contains(bucketName, payloadHashPostfix), strings.Contains(bucketName, parentPostfix): return true default: return false } } func cleanUpUniqueBucket(tx *bbolt.Tx, name []byte, b *bbolt.Bucket) { if b.Stats().KeyN == 0 { _ = tx.DeleteBucket(name) // ignore error, best effort there } } func cleanUpFKBT(tx *bbolt.Tx, name []byte, b *bbolt.Bucket) { removedBuckets := 0 remainingBuckets := b.Stats().BucketN - 1 _ = b.ForEach(func(k, _ []byte) error { fkbtRoot := b.Bucket(k) if fkbtRoot == nil { return nil } if fkbtRoot.Stats().KeyN == 0 { err := b.DeleteBucket(k) if err == nil { removedBuckets++ } } return nil }) if remainingBuckets == removedBuckets { _ = tx.DeleteBucket(name) // ignore error, best effort there } } func cleanUpListBucket(tx *bbolt.Tx, name []byte, b *bbolt.Bucket) { cleanUpUniqueBucket(tx, name, b) }