93690baf4e
Bumps [github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus](https://github.com/grpc-ecosystem/go-grpc-middleware) from 1.0.0 to 1.0.1. - [Release notes](https://github.com/grpc-ecosystem/go-grpc-middleware/releases) - [Commits](https://github.com/grpc-ecosystem/go-grpc-middleware/compare/v1.0.0...providers/prometheus/v1.0.1) --- updated-dependencies: - dependency-name: github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus dependency-type: direct:production update-type: version-update:semver-patch ... Signed-off-by: dependabot[bot] <support@github.com>
67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
// Copyright (c) The go-grpc-middleware Authors.
|
|
// Licensed under the Apache License 2.0.
|
|
|
|
package interceptors
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func splitFullMethodName(fullMethod string) (string, string) {
|
|
fullMethod = strings.TrimPrefix(fullMethod, "/") // remove leading slash
|
|
if i := strings.Index(fullMethod, "/"); i >= 0 {
|
|
return fullMethod[:i], fullMethod[i+1:]
|
|
}
|
|
return "unknown", "unknown"
|
|
}
|
|
|
|
type CallMeta struct {
|
|
ReqOrNil any
|
|
Typ GRPCType
|
|
Service string
|
|
Method string
|
|
IsClient bool
|
|
}
|
|
|
|
func NewClientCallMeta(fullMethod string, streamDesc *grpc.StreamDesc, reqOrNil any) CallMeta {
|
|
c := CallMeta{IsClient: true, ReqOrNil: reqOrNil, Typ: Unary}
|
|
if streamDesc != nil {
|
|
c.Typ = clientStreamType(streamDesc)
|
|
}
|
|
c.Service, c.Method = splitFullMethodName(fullMethod)
|
|
return c
|
|
}
|
|
|
|
func NewServerCallMeta(fullMethod string, streamInfo *grpc.StreamServerInfo, reqOrNil any) CallMeta {
|
|
c := CallMeta{IsClient: false, ReqOrNil: reqOrNil, Typ: Unary}
|
|
if streamInfo != nil {
|
|
c.Typ = serverStreamType(streamInfo)
|
|
}
|
|
c.Service, c.Method = splitFullMethodName(fullMethod)
|
|
return c
|
|
}
|
|
func (c CallMeta) FullMethod() string {
|
|
return fmt.Sprintf("/%s/%s", c.Service, c.Method)
|
|
}
|
|
|
|
func clientStreamType(desc *grpc.StreamDesc) GRPCType {
|
|
if desc.ClientStreams && !desc.ServerStreams {
|
|
return ClientStream
|
|
} else if !desc.ClientStreams && desc.ServerStreams {
|
|
return ServerStream
|
|
}
|
|
return BidiStream
|
|
}
|
|
|
|
func serverStreamType(info *grpc.StreamServerInfo) GRPCType {
|
|
if info.IsClientStream && !info.IsServerStream {
|
|
return ClientStream
|
|
} else if !info.IsClientStream && info.IsServerStream {
|
|
return ServerStream
|
|
}
|
|
return BidiStream
|
|
}
|