Mkdir rocksmq_path if not exist (#6134)

Signed-off-by: kun yu <kun.yu@zilliz.com>
pull/6136/head
yukun 2021-06-26 12:28:07 +08:00 committed by GitHub
parent 8db95a63f6
commit e94a27856e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions

View File

@ -12,6 +12,7 @@
package rocksmq
import (
"os"
"sync"
"github.com/milvus-io/milvus/internal/allocator"
@ -31,6 +32,15 @@ func InitRmq(rocksdbName string, idAllocator allocator.GIDAllocator) error {
func InitRocksMQ(rocksdbName string) error {
var err error
once.Do(func() {
_, err := os.Stat(rocksdbName)
if os.IsNotExist(err) {
err = os.MkdirAll(rocksdbName, os.ModePerm)
if err != nil {
errMsg := "Create dir " + rocksdbName + " failed"
panic(errMsg)
}
}
kvname := rocksdbName + "_kv"
rocksdbKV, err := rocksdbkv.NewRocksdbKV(kvname)
if err != nil {