func (engine *RocksDbEngine) Init(config *proto.DBConfigs) error {
if config == nil {
return proto.ErrNoEngineConfig
}
if config.RocksDbConfigs == nil {
config.RocksDbConfigs = DefaultRocksDbConf
}
options := ratgo.NewOptions()
// options.SetCreateIfMissing(config.RocksDbConfigs.CreateIfMissing)
options.SetCreateIfMissing(true)
options.SetParanoidChecks(config.RocksDbConfigs.ParanoidCheck)
if config.RocksDbConfigs.LRUCacheSize > 0 {
options.SetCache(ratgo.NewLRUCache(config.RocksDbConfigs.LRUCacheSize))
}
if config.RocksDbConfigs.BloomFilterLength > 0 {
options.SetFilterPolicy(ratgo.NewBloomFilter(config.RocksDbConfigs.BloomFilterLength))
}
engine.config = config
engine.dbOptions = options
db, err := ratgo.Open(engine.config.DataPath, engine.dbOptions)
if err != nil {
return err
}
engine.db = db
return nil
}
func TestRocksEngine(t *testing.T) {
// init
conf := new(proto.DBConfigs)
conf.DataPath = "test"
engine := NewRocksDbEngine()
err := engine.Init(conf)
if engine.Name() != "rocksdb" {
t.Fatalf("want engine name RocksDb, but result is %v\n", engine.Name())
}
if err != nil {
t.Fatalf("Init engine error, %v\n", err.Error())
}
var connection proto.DbConnection
connection, _ = engine.Connect(&proto.DbConnectParams{DbName: "test", UserName: "******"})
var result []byte
var results [][]byte
k1 := []byte("user1")
k2 := []byte("user2")
k3 := []byte("user3")
k4 := []byte("user4")
// k5 := []byte("user5")
v1 := []byte("value1")
v2 := []byte("value2")
v3 := []byte("value3")
// v4 := []byte("value4")
// v5 := []byte("value5")
wo := new(proto.DbWriteOptions)
ro := new(proto.DbReadOptions)
// Put
err = connection.Put(wo, k1, v1)
if err != nil {
t.Errorf("Put Key : %v error, %v\n", string(k1), err.Error())
}
err = connection.Puts(wo, [][]byte{k2, k3}, [][]byte{v2, v3})
if err != nil {
t.Errorf("Puts Key error, %v\n", err.Error())
}
// Get
result, err = connection.Get(ro, k1)
if err != nil {
t.Errorf("Get key : %v error, %v\n", string(k1), err.Error())
}
if bytes.Compare(result, v1) != 0 {
t.Errorf("Expect %v=%v, but is %v\n", string(k1), string(result), string(v1))
}
results, err = connection.Gets(ro, [][]byte{k2, k3})
if err != nil {
t.Errorf("Gets error, %v\n", err.Error())
}
if bytes.Compare(results[0], v2) != 0 || bytes.Compare(results[1], v3) != 0 {
t.Error("Gets result error")
}
// Iterate
iter, err := connection.Iterate(ro, k2, k4, 0)
count := 0
for ; iter.Valid(); iter.Next() {
count++
}
if count != 2 {
t.Errorf("Iterate error, expect 2, result is %d\n", count)
}
iter.Close()
connection.Close()
// err = connection.Destroy()
// if err != nil {
// t.Fatal("Destory Db error")
// }
}