func (em *Engine) Connect(params *proto.DbConnectParams) (conn *DBConnection, err error) {
var c proto.DbConnection
c, err = em.dbEngine.Connect(params)
log.Info("Come to a new client %v, id is %d", params.UserName, c.Id())
if conn != nil && err != nil {
conn.Close()
return nil, err
}
conn = &DBConnection{connectionParams: params, DbConnection: c}
return conn, 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")
// }
}
