wehub-resource-sync
e071084ebe
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
43 lines
814 B
Go
43 lines
814 B
Go
package grpc
|
|
|
|
import (
|
|
"runtime/debug"
|
|
|
|
"go-micro.dev/v6/errors"
|
|
"go-micro.dev/v6/logger"
|
|
"go-micro.dev/v6/transport"
|
|
pb "go-micro.dev/v6/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
|
|
}
|