frostfs-node/pkg/local_object_storage/metabase/version.go
Evgenii Stratonikov ae1dab29bc [#1482] metabase: Encode database keys in binary
Signed-off-by: Evgenii Stratonikov <evgeniy@morphbits.ru>
2022-09-15 10:28:26 +03:00

41 lines
955 B
Go

package meta
import (
"encoding/binary"
"fmt"
"go.etcd.io/bbolt"
)
// version contains current metabase version.
const version = 2
var versionKey = []byte("version")
func checkVersion(tx *bbolt.Tx, initialized bool) error {
b := tx.Bucket(shardInfoBucket)
if b != nil {
data := b.Get(versionKey)
if len(data) == 8 {
stored := binary.LittleEndian.Uint64(data)
if stored != version {
return fmt.Errorf("invalid version: expected=%d, stored=%d", version, stored)
}
}
}
if !initialized { // new database, write version
return updateVersion(tx, version)
}
return nil // return error here after the first version increase
}
func updateVersion(tx *bbolt.Tx, version uint64) error {
data := make([]byte, 8)
binary.LittleEndian.PutUint64(data, version)
b, err := tx.CreateBucketIfNotExists(shardInfoBucket)
if err != nil {
return fmt.Errorf("can't create auxilliary bucket: %w", err)
}
return b.Put(versionKey, data)
}