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()