package grpc import ( "runtime/debug" "github.com/micro/go-micro/v2/errors" "github.com/micro/go-micro/v2/transport" pb "github.com/micro/go-micro/v2/transport/grpc/proto" "github.com/micro/go-micro/v2/util/log" "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 { log.Log(r, string(debug.Stack())) sock.Close() err = errors.InternalServerError("go.micro.transport", "panic recovered: %v", r) } }() // execute socket func m.fn(sock) return err }