micro/transport/grpc/handler.go

44 lines
851 B
Go
Raw Normal View History

2019-05-24 19:15:59 +03:00
package grpc
import (
"runtime/debug"
"github.com/micro/go-micro/errors"
2019-07-09 20:41:26 +03:00
"github.com/micro/go-micro/transport"
pb "github.com/micro/go-micro/transport/grpc/proto"
2019-06-03 21:30:43 +03:00
"github.com/micro/go-micro/util/log"
2019-05-24 19:15:59 +03:00
"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) {
2019-05-24 19:15:59 +03:00
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)
2019-05-24 19:15:59 +03:00
}
}()
// execute socket func
m.fn(sock)
return err
2019-05-24 19:15:59 +03:00
}