linka-cloud/grpc-toolkit
2
0
Fork 0
mirror of https://github.com/linka-cloud/grpc.git synced 2025-06-22 09:12:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

interceptors: migrate to otel and add logging interceptor

Browse Source 
Signed-off-by: Adphi <philippe.adrien.nousse@gmail.com>
This commit is contained in:
Adphi
2024-10-17 17:15:05 +02:00
parent 9591a64e09
commit 3a3d77169c
41 changed files with 2817 additions and 852 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
service/service.go
View File

@ -30,6 +30,7 @@ import (
"google.golang.org/grpc/health/grpc_health_v1"
greflect "google.golang.org/grpc/reflection"
"go.linka.cloud/grpc-toolkit/internal/injectlogger"
"go.linka.cloud/grpc-toolkit/logger"
"go.linka.cloud/grpc-toolkit/registry"
"go.linka.cloud/grpc-toolkit/registry/noop"
@ -81,19 +82,35 @@ func newService(opts ...Option) (*service, error) {
for _, f := range opts {
f(s.opts)
}
s.opts.ctx, s.cancel = context.WithCancel(s.opts.ctx)
md := md(s.opts)
if md != nil {
s.opts.unaryServerInterceptors = append([]grpc.UnaryServerInterceptor{md.UnaryServerInterceptor()}, s.opts.unaryServerInterceptors...)
s.opts.unaryClientInterceptors = append([]grpc.UnaryClientInterceptor{md.UnaryClientInterceptor()}, s.opts.unaryClientInterceptors...)
s.opts.streamServerInterceptors = append([]grpc.StreamServerInterceptor{md.StreamServerInterceptor()}, s.opts.streamServerInterceptors...)
s.opts.streamClientInterceptors = append([]grpc.StreamClientInterceptor{md.StreamClientInterceptor()}, s.opts.streamClientInterceptors...)
s.opts.unaryServerInterceptors = append([]grpc.UnaryServerInterceptor{
md.UnaryServerInterceptor(),
injectlogger.New(s.opts.ctx).UnaryServerInterceptor()},
s.opts.unaryServerInterceptors...,
)
s.opts.unaryClientInterceptors = append([]grpc.UnaryClientInterceptor{
md.UnaryClientInterceptor(),
injectlogger.New(s.opts.ctx).UnaryClientInterceptor()},
s.opts.unaryClientInterceptors...,
)
s.opts.streamServerInterceptors = append([]grpc.StreamServerInterceptor{
md.StreamServerInterceptor(),
injectlogger.New(s.opts.ctx).StreamServerInterceptor()},
s.opts.streamServerInterceptors...,
)
s.opts.streamClientInterceptors = append([]grpc.StreamClientInterceptor{
md.StreamClientInterceptor(),
injectlogger.New(s.opts.ctx).StreamClientInterceptor()},
s.opts.streamClientInterceptors...,
)
}
if s.opts.error != nil {
return nil, s.opts.error
}
s.opts.ctx, s.cancel = context.WithCancel(s.opts.ctx)
go func() {
for {
select {