func NewTCPClient(hostport string) (*Client, error) {
cli, err := rpc.NewClient(anet.TCPDialer(hostport))
if err != nil {
return nil, err
}
return &Client{cli}, nil
}
func BenchmarkTCPClientCreate(b *testing.B) {
srv := NewServer()
defer srv.Close()
srv.TCPListen(hostport)
b.ResetTimer()
for i := 0; i < b.N; i++ {
NewClient(anet.TCPDialer(hostport))
}
b.StopTimer()
}
func BenchmarkTCPNormRPC(b *testing.B) {
srv := NewServer()
srv.Register(serverPrefix, &testServer{})
defer srv.Close()
srv.TCPListen(hostport)
cli, err := NewClient(anet.TCPDialer(hostport))
if err != nil {
b.Fatal(err)
}
rangeRPCBench(b, cli)
}
return NewMemStore(), func() {}
}},
{"disk", func(t *testing.T) (BlkStore, func()) {
te := testutil.NewTestEnv("disk", t)
return NewDiskStore(te.Root()), func() { te.Teardown() }
}},
{"remote", func(t *testing.T) (BlkStore, func()) {
const hostport = "localhost:9000"
te := testutil.NewTestEnv("remote", t)
srv, err := remote.NewTCPServer(te.Root(), hostport)
if err != nil {
t.Errorf("setup error: %v", err)
}
bs, err := NewRemoteStore(anet.TCPDialer(hostport))
if err != nil {
t.Errorf("setup error: %v", err)
}
return bs, func() {
srv.Close()
te.Teardown()
}
}},
}
func testPutGet(t *testing.T, test testCase) {
bs, cleanup := test.setup(t)
defer cleanup()
