69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package chain
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/encoding/gbase64"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/os/gcfg"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
/**
|
|
* 超时链
|
|
* @author dc.To
|
|
* @version 20250409
|
|
*/
|
|
func ClientTimeoutInvokerChain(ctx context.Context, method string, req, res interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
|
|
ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
|
|
defer cancel()
|
|
return invoker(ctx, method, req, res, cc, opts...)
|
|
}
|
|
|
|
/**
|
|
* 上下文调用链
|
|
* @param void
|
|
* @author dc.To
|
|
* @version 20250424
|
|
*/
|
|
func ClientContextInvokerChain(ctx context.Context, method string, req, res interface{}, cc *grpc.ClientConn, invoker grpc.UnaryInvoker, opts ...grpc.CallOption) error {
|
|
return invoker(ctx, method, req, nil, cc, opts...)
|
|
}
|
|
|
|
/**
|
|
* Context Unary Interceptor
|
|
* @param void
|
|
* @author dc.To
|
|
* @version 20250425
|
|
*/
|
|
func ClientContextUnaryChain(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
|
|
return handler(ctx, req)
|
|
}
|
|
|
|
/**
|
|
* Metadata Unary Interceptor
|
|
* @author dc.To
|
|
* @version 20250418
|
|
*/
|
|
func ClientMetadataUnaryChain(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
|
|
md, ok := metadata.FromIncomingContext(ctx)
|
|
if ok {
|
|
for k, v := range md {
|
|
if strings.HasPrefix(k, "x-cfg-") && len(v) > 0 {
|
|
c := strings.Join(v, "")
|
|
if len(c) > 0 {
|
|
adapter, err := gcfg.NewAdapterContent(gbase64.MustDecodeToString(v[0]))
|
|
if err != nil {
|
|
g.Log("unary").Error(ctx, err)
|
|
}
|
|
g.Cfg(strings.TrimPrefix(k, "x-cfg-")).SetAdapter(adapter)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return handler(ctx, req)
|
|
}
|