add a http client

http_test.go

@@ -0,0 +1,280 @@
+package http
+
+import (
+	"bufio"
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"net"
+	"net/http"
+	"strconv"
+	"testing"
+
+	"github.com/micro/go-micro/client"
+	"github.com/micro/go-micro/registry"
+	"github.com/micro/go-micro/selector"
+	"github.com/micro/go-plugins/client/http/proto"
+	"github.com/micro/go-plugins/registry/memory"
+
+	"golang.org/x/net/context"
+)
+
+func TestHTTPClient(t *testing.T) {
+	r := memory.NewRegistry()
+	s := selector.NewSelector(selector.Registry(r))
+
+	l, err := net.Listen("tcp", "127.0.0.1:0")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer l.Close()
+
+	mux := http.NewServeMux()
+	mux.HandleFunc("/foo/bar", func(w http.ResponseWriter, r *http.Request) {
+		// only accept post
+		if r.Method != "POST" {
+			http.Error(w, "expect post method", 500)
+			return
+		}
+
+		// get codec
+		ct := r.Header.Get("Content-Type")
+		codec, ok := defaultHTTPCodecs[ct]
+		if !ok {
+			http.Error(w, "codec not found", 500)
+			return
+		}
+		b, err := ioutil.ReadAll(r.Body)
+		if err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+
+		// extract message
+		msg := new(test.Message)
+		if err := codec.Unmarshal(b, msg); err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+
+		// marshal response
+		b, err = codec.Marshal(msg)
+		if err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+
+		// write response
+		w.Write(b)
+	})
+	go http.Serve(l, mux)
+
+	host, sport, err := net.SplitHostPort(l.Addr().String())
+	if err != nil {
+		t.Fatal(err)
+	}
+	port, _ := strconv.Atoi(sport)
+
+	if err := r.Register(&registry.Service{
+		Name: "test.service",
+		Nodes: []*registry.Node{
+			{
+				Id:      "test.service.1",
+				Address: host,
+				Port:    port,
+			},
+		},
+	}); err != nil {
+		t.Fatal(err)
+	}
+
+	c := NewClient(client.Selector(s))
+
+	for i := 0; i < 10; i++ {
+		msg := &test.Message{
+			Seq:  int64(i),
+			Data: fmt.Sprintf("message %d", i),
+		}
+		req := c.NewRequest("test.service", "/foo/bar", msg)
+		rsp := new(test.Message)
+		err := c.Call(context.TODO(), req, rsp)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if rsp.Seq != msg.Seq {
+			t.Fatalf("invalid seq %d for %d", rsp.Seq, msg.Seq)
+		}
+	}
+}
+
+func TestHTTPClientStream(t *testing.T) {
+	r := memory.NewRegistry()
+	s := selector.NewSelector(selector.Registry(r))
+
+	l, err := net.Listen("tcp", "127.0.0.1:0")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer l.Close()
+
+	mux := http.NewServeMux()
+	mux.HandleFunc("/foo/bar", func(w http.ResponseWriter, r *http.Request) {
+		// only accept post
+		if r.Method != "POST" {
+			http.Error(w, "expect post method", 500)
+			return
+		}
+
+		// hijack the connection
+		hj, ok := w.(http.Hijacker)
+		if !ok {
+			http.Error(w, "could not hijack conn", 500)
+			return
+
+		}
+
+		// hijacked
+		conn, bufrw, err := hj.Hijack()
+		if err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+		defer conn.Close()
+
+		// read off the first request
+		// get codec
+		ct := r.Header.Get("Content-Type")
+		codec, ok := defaultHTTPCodecs[ct]
+		if !ok {
+			http.Error(w, "codec not found", 500)
+			return
+		}
+		b, err := ioutil.ReadAll(r.Body)
+		if err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+
+		// extract message
+		msg := new(test.Message)
+		if err := codec.Unmarshal(b, msg); err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+
+		// marshal response
+		b, err = codec.Marshal(msg)
+		if err != nil {
+			http.Error(w, err.Error(), 500)
+			return
+		}
+
+		// write response
+		rsp := &http.Response{
+			Header:        r.Header,
+			Body:          &buffer{bytes.NewBuffer(b)},
+			Status:        "200 OK",
+			StatusCode:    200,
+			Proto:         "HTTP/1.1",
+			ProtoMajor:    1,
+			ProtoMinor:    1,
+			ContentLength: int64(len(b)),
+		}
+
+		// write response
+		rsp.Write(bufrw)
+		bufrw.Flush()
+
+		reader := bufio.NewReader(conn)
+
+		for {
+			r, err := http.ReadRequest(reader)
+			if err != nil {
+				http.Error(w, err.Error(), 500)
+				return
+			}
+
+			b, err = ioutil.ReadAll(r.Body)
+			if err != nil {
+				http.Error(w, err.Error(), 500)
+				return
+			}
+
+			// extract message
+			msg := new(test.Message)
+			if err := codec.Unmarshal(b, msg); err != nil {
+				http.Error(w, err.Error(), 500)
+				return
+			}
+
+			// marshal response
+			b, err = codec.Marshal(msg)
+			if err != nil {
+				http.Error(w, err.Error(), 500)
+				return
+			}
+
+			rsp := &http.Response{
+				Header:        r.Header,
+				Body:          &buffer{bytes.NewBuffer(b)},
+				Status:        "200 OK",
+				StatusCode:    200,
+				Proto:         "HTTP/1.1",
+				ProtoMajor:    1,
+				ProtoMinor:    1,
+				ContentLength: int64(len(b)),
+			}
+
+			// write response
+			rsp.Write(bufrw)
+			bufrw.Flush()
+		}
+	})
+	go http.Serve(l, mux)
+
+	host, sport, err := net.SplitHostPort(l.Addr().String())
+	if err != nil {
+		t.Fatal(err)
+	}
+	port, _ := strconv.Atoi(sport)
+
+	if err := r.Register(&registry.Service{
+		Name: "test.service",
+		Nodes: []*registry.Node{
+			{
+				Id:      "test.service.1",
+				Address: host,
+				Port:    port,
+			},
+		},
+	}); err != nil {
+		t.Fatal(err)
+	}
+
+	c := NewClient(client.Selector(s))
+	req := c.NewRequest("test.service", "/foo/bar", new(test.Message))
+	stream, err := c.Stream(context.TODO(), req)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer stream.Close()
+
+	for i := 0; i < 10; i++ {
+		msg := &test.Message{
+			Seq:  int64(i),
+			Data: fmt.Sprintf("message %d", i),
+		}
+		err := stream.Send(msg)
+		if err != nil {
+			t.Fatal(err)
+		}
+		rsp := new(test.Message)
+		err = stream.Recv(rsp)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if rsp.Seq != msg.Seq {
+			t.Fatalf("invalid seq %d for %d", rsp.Seq, msg.Seq)
+		}
+	}
+}