move transport back

transport/grpc/socket.go

@@ -0,0 +1,97 @@
+package grpc
+
+import (
+	"github.com/micro/go-micro/transport"
+	pb "github.com/micro/go-micro/transport/grpc/proto"
+	"google.golang.org/grpc"
+)
+
+type grpcTransportClient struct {
+	conn   *grpc.ClientConn
+	stream pb.Transport_StreamClient
+
+	local  string
+	remote string
+}
+
+type grpcTransportSocket struct {
+	stream pb.Transport_StreamServer
+	local  string
+	remote string
+}
+
+func (g *grpcTransportClient) Local() string {
+	return g.local
+}
+
+func (g *grpcTransportClient) Remote() string {
+	return g.remote
+}
+
+func (g *grpcTransportClient) Recv(m *transport.Message) error {
+	if m == nil {
+		return nil
+	}
+
+	msg, err := g.stream.Recv()
+	if err != nil {
+		return err
+	}
+
+	m.Header = msg.Header
+	m.Body = msg.Body
+	return nil
+}
+
+func (g *grpcTransportClient) Send(m *transport.Message) error {
+	if m == nil {
+		return nil
+	}
+
+	return g.stream.Send(&pb.Message{
+		Header: m.Header,
+		Body:   m.Body,
+	})
+}
+
+func (g *grpcTransportClient) Close() error {
+	return g.conn.Close()
+}
+
+func (g *grpcTransportSocket) Local() string {
+	return g.local
+}
+
+func (g *grpcTransportSocket) Remote() string {
+	return g.remote
+}
+
+func (g *grpcTransportSocket) Recv(m *transport.Message) error {
+	if m == nil {
+		return nil
+	}
+
+	msg, err := g.stream.Recv()
+	if err != nil {
+		return err
+	}
+
+	m.Header = msg.Header
+	m.Body = msg.Body
+	return nil
+}
+
+func (g *grpcTransportSocket) Send(m *transport.Message) error {
+	if m == nil {
+		return nil
+	}
+
+	return g.stream.Send(&pb.Message{
+		Header: m.Header,
+		Body:   m.Body,
+	})
+}
+
+func (g *grpcTransportSocket) Close() error {
+	return nil
+}