mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-08 12:01:20 +00:00
46 lines
1007 B
Go
46 lines
1007 B
Go
|
package utils
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"os"
|
||
|
|
||
|
"github.com/ledgerwatch/turbo-geth/ethdb"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
TypeLMDB = "lmdb"
|
||
|
TypeMDBX = "mdbx"
|
||
|
)
|
||
|
|
||
|
var ErrUnsupported error = errors.New("unsupported KV type")
|
||
|
|
||
|
func RmTmpFiles(dbType string, snapshotPath string) error {
|
||
|
switch dbType {
|
||
|
case TypeLMDB:
|
||
|
return rmLmdbLock(snapshotPath)
|
||
|
case TypeMDBX:
|
||
|
return rmMdbxLock(snapshotPath)
|
||
|
default:
|
||
|
return ErrUnsupported
|
||
|
}
|
||
|
}
|
||
|
func rmLmdbLock(snapshotPath string) error {
|
||
|
err := os.Remove(snapshotPath + "/lock.mdb")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
return os.Remove(snapshotPath + "/LOCK")
|
||
|
}
|
||
|
func rmMdbxLock(path string) error {
|
||
|
return os.Remove(path + "/mdbx.lck")
|
||
|
}
|
||
|
|
||
|
func OpenSnapshotKV(dbType string, configsFunc ethdb.BucketConfigsFunc, path string) ethdb.RwKV {
|
||
|
if dbType == TypeLMDB {
|
||
|
return ethdb.NewLMDB().WithBucketsConfig(configsFunc).Path(path).MustOpen()
|
||
|
} else if dbType == TypeMDBX {
|
||
|
return ethdb.NewMDBX().WithBucketsConfig(configsFunc).Path(path).MustOpen()
|
||
|
}
|
||
|
panic(ErrUnsupported.Error())
|
||
|
}
|