Ejemplo n.º 1
0
func BenchmarkRequests(b *testing.B) {
	tcpAddr, _ := net.ResolveTCPAddr("tcp", ":0")
	l, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		b.Fatal("Failed to create TCP listener:", err)
	}
	tcpAddr = l.Addr().(*net.TCPAddr)
	err = l.Close()
	if err != nil {
		b.Fatal("Failed to create TCP listener:", err)
	}
	worker := New(fmt.Sprintf("tcp://*:%d", tcpAddr.Port), 10)
	worker.RegisterWorkerFunction("test_func", worker.MakeWorkerFunction(func(i *SampleType) *gototo.Response {
		return gototo.CreateSuccessResponse(i)
	}))
	err = worker.Start()
	if err != nil {
		b.Fatal("Failed to start worker:", err)
	}
	time.Sleep(100 * time.Millisecond)
	defer worker.Stop()
	addr := fmt.Sprintf("tcp://127.0.0.1:%d", tcpAddr.Port)
	connection := NewConnection(addr)
	connection.RegisterResponseType("test_func", &SampleType{}, true)
	err = connection.Start()
	if err != nil {
		b.Fatal("Failed to start connection:", err)
	}
	defer connection.Stop()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		connection.Call("test_func", &SampleType{String: ""})
	}
}
Ejemplo n.º 2
0
func BenchmarkRequests(b *testing.B) {
	worker := New(":0")
	worker.RegisterWorkerFunction("test_func", worker.MakeWorkerFunction(func(i *SampleType) *gototo.Response {
		return gototo.CreateSuccessResponse(i)
	}))
	err := worker.Start()
	tcpAddr := worker.ConnectedAddress().(*net.TCPAddr)
	if err != nil {
		b.Fatal("Failed to start worker:", err)
	}
	time.Sleep(100 * time.Millisecond)
	defer worker.Stop()
	addr := fmt.Sprintf("http://127.0.0.1:%d", tcpAddr.Port)
	connection := NewConnection(addr)
	connection.RegisterResponseType("test_func", &SampleType{}, true)
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		connection.Call("test_func", &SampleType{String: ""})
	}
}
Ejemplo n.º 3
0
func TestCall(t *testing.T) {
	tcpAddr, _ := net.ResolveTCPAddr("tcp", ":0")
	l, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		t.Fatal("Failed to create TCP listener:", err)
	}
	tcpAddr = l.Addr().(*net.TCPAddr)
	err = l.Close()
	if err != nil {
		t.Fatal("Failed to create TCP listener:", err)
	}
	worker := New(fmt.Sprintf("tcp://*:%d", tcpAddr.Port), 10)
	worker.SetLogMetrics(true)
	var _ gototo.Worker = worker
	worker.RegisterWorkerFunction("test_func", worker.MakeWorkerFunction(func(i *SampleValidatedType) *gototo.Response {
		if i.String != "fail" {
			return gototo.CreateSuccessResponse(i)
		}
		return gototo.CreateErrorResponse(gototo.NewCodedError("Empty string!", 42))
	}))
	worker.RegisterWorkerFunction("slow_test_func", worker.MakeWorkerFunction(func(i *WaitRequest) *gototo.Response {
		time.Sleep(i.Timeout)
		return gototo.CreateSuccessResponse(i)
	}))
	err = worker.Start()
	if err != nil {
		t.Fatal("Failed to start worker:", err)
	}
	time.Sleep(100 * time.Millisecond)
	defer worker.Stop()
	addr := fmt.Sprintf("tcp://127.0.0.1:%d", tcpAddr.Port)
	connection := NewConnection(addr)
	var _ gototo.WorkerConnection = connection
	connection.RegisterResponseType("test_func", &SampleValidatedType{}, true)
	connection.RegisterResponseType("slow_test_func", &WaitRequest{}, true)
	connection.RegisterDefaultOptions("slow_test_func", &gototo.RequestOptions{Timeout: 100 * time.Millisecond, RetryCount: 3})
	err = connection.Start()
	if err != nil {
		t.Fatal("Failed to start connection:", err)
	}
	err = connection.Disconnect(addr)
	if err != nil {
		t.Fatal("Failed to disconnect:", err)
	}
	if len(connection.GetEndpoints()) > 0 {
		t.Fatal("Expected no connections")
	}
	err = connection.Connect(addr)
	if err != nil {
		t.Fatal("Failed to reconnect:", err)
	}
	if len(connection.GetEndpoints()) != 1 {
		t.Fatal("Expected one connections")
	}
	err = connection.SetEndpoints()
	if err != nil {
		t.Fatal("Failed to disconnect:", err)
	}
	if len(connection.GetEndpoints()) > 0 {
		t.Fatal("Expected no connections")
	}
	err = connection.SetEndpoints(addr)
	if err != nil {
		t.Fatal("Failed to reconnect:", err)
	}
	if len(connection.GetEndpoints()) != 1 {
		t.Fatal("Expected one connections")
	}
	err = connection.Connect(addr)
	if err != nil {
		t.Fatal("Failed to reconnect:", err)
	}
	if len(connection.GetEndpoints()) != 1 {
		t.Fatal("Expected one connections")
	}
	defer connection.Stop()
	resp, err := connection.Call("test_func", &SampleValidatedType{String: "test request string"})
	if err != nil {
		t.Fatal("Failed to decode response:", err)
	}
	if svt, ok := resp.(*SampleValidatedType); !ok {
		t.Fatalf("Bad response: %#v\n", resp)
	} else if svt.String != "test request string" {
		t.Fatalf("Bad response: %#v\n", resp)
	}
	resp, err = connection.Call("test_func", &SampleValidatedType{String: ""})
	if err == nil {
		t.Fatalf("Expected error: %#v %#v\n", resp, err)
	} else {
		if responseError, ok := err.(*gototo.ResponseError); !ok {
			t.Fatalf("Expected *ResponseError: %#v\n", err)
		} else if responseError.Error() != "Validation failed: Empty String field" {
			t.Fatalf("Expected 'Validation failed: Empty String field': %#v\n", err)
		} else if responseError.Code() != gototo.CodeValidationFailed {
			log.Printf("%#v", responseError)
			t.Fatalf("Expected validation failed code, found %d", responseError.Code())
		}
	}
	resp, err = connection.Call("test_func", &SampleValidatedType{String: "fail"})
	if err == nil {
		t.Fatalf("Expected error: %#v %#v\n", resp, err)
	} else {
		if responseError, ok := err.(*gototo.ResponseError); !ok {
			t.Fatalf("Expected *ResponseError: %#v\n", err)
		} else if responseError.Error() != "Empty string!" {
			t.Fatalf("Expected 'Empty string!': %#v\n", err)
		} else if responseError.Code() != 42 {
			t.Fatalf("Expected error code 42, found %d", responseError.Code())
		}
	}
	resp, err = connection.Call("slow_test_func", &WaitRequest{Timeout: 200 * time.Millisecond})
	if err != nil {
		t.Error("Unexpected response error:", err)
	}
	resp, err = connection.Call("slow_test_func", &WaitRequest{Timeout: 1 * time.Second})
	if err != gototo.ErrTimeout {
		t.Error("Expected timeout, found:", err, resp)
	}
}