storj/pkg/rpc/rpcpeer/peer_grpc.go
2019-11-01 22:43:24 +01:00

32 lines
652 B
Go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
// +build grpc
package rpcpeer
import (
"context"
"google.golang.org/grpc/credentials"
"google.golang.org/grpc/peer"
)
// internalFromContext returns a peer from the context using grpc.
func internalFromContext(ctx context.Context) (*Peer, error) {
peer, ok := peer.FromContext(ctx)
if !ok {
return nil, Error.New("unable to get grpc peer from context")
}
tlsInfo, ok := peer.AuthInfo.(credentials.TLSInfo)
if !ok {
return nil, Error.New("peer AuthInfo is not credentials.TLSInfo")
}
return &Peer{
Addr: peer.Addr,
State: tlsInfo.State,
}, nil
}