milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Not remove rocksdb folder when close and init (#5974) 

Signed-off-by: fishpenguin <kun.yu@zilliz.com>
pull/5980/head^2
yukun 2021-06-22 16:20:10 +08:00 committed by GitHub
parent d2c8462dac
commit 282d2624e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
internal/util/rocksmq/server/rocksmq/global_rmq.go
View File

@ -12,7 +12,6 @@
package rocksmq
import (
"os"
"sync"
"github.com/milvus-io/milvus/internal/allocator"
@ -33,9 +32,6 @@ func InitRocksMQ(rocksdbName string) error {
var err error
once.Do(func() {
kvname := rocksdbName + "_kv"
if _, err := os.Stat(kvname); !os.IsNotExist(err) {
_ = os.RemoveAll(kvname)
}
rocksdbKV, err := rocksdbkv.NewRocksdbKV(kvname)
if err != nil {
panic(err)
@ -43,9 +39,6 @@ func InitRocksMQ(rocksdbName string) error {
idAllocator := allocator.NewGlobalIDAllocator("rmq_id", rocksdbKV)
_ = idAllocator.Initialize()
if _, err := os.Stat(rocksdbName); !os.IsNotExist(err) {
_ = os.RemoveAll(rocksdbName)
}
Rmq, err = NewRocksMQ(rocksdbName, idAllocator)
if err != nil {
panic(err)
@ -57,9 +50,5 @@ func InitRocksMQ(rocksdbName string) error {
func CloseRocksMQ() {
if Rmq != nil && Rmq.store != nil {
Rmq.store.Close()
rocksdbName := Rmq.store.Name()
_ = os.RemoveAll(rocksdbName)
kvname := rocksdbName + "_kv"
os.RemoveAll(kvname)
}
}