mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-12-30 14:00:57 +00:00
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
// Copyright 2016 Michal Witkowski. All Rights Reserved.
|
|
// See LICENSE for licensing terms.
|
|
|
|
package grpc_prometheus
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
)
|
|
|
|
type grpcType string
|
|
|
|
const (
|
|
Unary grpcType = "unary"
|
|
ClientStream grpcType = "client_stream"
|
|
ServerStream grpcType = "server_stream"
|
|
BidiStream grpcType = "bidi_stream"
|
|
)
|
|
|
|
var (
|
|
allCodes = []codes.Code{
|
|
codes.OK, codes.Canceled, codes.Unknown, codes.InvalidArgument, codes.DeadlineExceeded, codes.NotFound,
|
|
codes.AlreadyExists, codes.PermissionDenied, codes.Unauthenticated, codes.ResourceExhausted,
|
|
codes.FailedPrecondition, codes.Aborted, codes.OutOfRange, codes.Unimplemented, codes.Internal,
|
|
codes.Unavailable, codes.DataLoss,
|
|
}
|
|
)
|
|
|
|
func splitMethodName(fullMethodName string) (string, string) {
|
|
fullMethodName = strings.TrimPrefix(fullMethodName, "/") // remove leading slash
|
|
if i := strings.Index(fullMethodName, "/"); i >= 0 {
|
|
return fullMethodName[:i], fullMethodName[i+1:]
|
|
}
|
|
return "unknown", "unknown"
|
|
}
|
|
|
|
func typeFromMethodInfo(mInfo *grpc.MethodInfo) grpcType {
|
|
if !mInfo.IsClientStream && !mInfo.IsServerStream {
|
|
return Unary
|
|
}
|
|
if mInfo.IsClientStream && !mInfo.IsServerStream {
|
|
return ClientStream
|
|
}
|
|
if !mInfo.IsClientStream && mInfo.IsServerStream {
|
|
return ServerStream
|
|
}
|
|
return BidiStream
|
|
}
|