move transport (#1967)

network/transport/grpc/handler.go

@@ -0,0 +1,43 @@
+package grpc
+
+import (
+	"runtime/debug"
+
+	"github.com/micro/go-micro/v3/errors"
+	"github.com/micro/go-micro/v3/logger"
+	"github.com/micro/go-micro/v3/network/transport"
+	pb "github.com/micro/go-micro/v3/network/transport/grpc/proto"
+	"google.golang.org/grpc/peer"
+)
+
+// microTransport satisfies the pb.TransportServer inteface
+type microTransport struct {
+	addr string
+	fn   func(transport.Socket)
+}
+
+func (m *microTransport) Stream(ts pb.Transport_StreamServer) (err error) {
+
+	sock := &grpcTransportSocket{
+		stream: ts,
+		local:  m.addr,
+	}
+
+	p, ok := peer.FromContext(ts.Context())
+	if ok {
+		sock.remote = p.Addr.String()
+	}
+
+	defer func() {
+		if r := recover(); r != nil {
+			logger.Error(r, string(debug.Stack()))
+			sock.Close()
+			err = errors.InternalServerError("go.micro.transport", "panic recovered: %v", r)
+		}
+	}()
+
+	// execute socket func
+	m.fn(sock)
+
+	return err
+}