Compare commits
14 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6ed8f159e7 | ||
|
|
c9133edfd9 | ||
|
|
fed21bfb86 | ||
|
|
981e1e0cf0 | ||
|
|
7db81a88e4 | ||
|
|
ffdfa9c94f | ||
|
|
fe25f60b5d | ||
|
|
ccf5df87e7 | ||
|
|
c680bd2d5e | ||
|
|
1c5ad5f20b | ||
|
|
0110dee434 | ||
|
|
db0378c430 | ||
|
|
a569f303f3 | ||
|
|
e3ec5b5e99 |
70
apapter/atomic.go
Normal file
70
apapter/atomic.go
Normal file
@ -0,0 +1,70 @@
|
||||
package apapter
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"sync/atomic"
|
||||
|
||||
"github.com/gogf/gf/v2/container/gvar"
|
||||
"github.com/gogf/gf/v2/encoding/gbase64"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"google.golang.org/grpc/metadata"
|
||||
)
|
||||
|
||||
/**
|
||||
* 原子钟接口配置
|
||||
* @author dc.To
|
||||
* @version 20250807
|
||||
*/
|
||||
type AtomicAdapter struct {
|
||||
store atomic.Value
|
||||
adapter string
|
||||
}
|
||||
|
||||
func NewAtomicAdapter(ctx context.Context, adapter string) *AtomicAdapter {
|
||||
a := &AtomicAdapter{adapter: adapter}
|
||||
md, ok := metadata.FromIncomingContext(ctx)
|
||||
if ok {
|
||||
if c := md.Get(a.adapter); len(c) > 0 {
|
||||
a.SetContent(gjson.New(gbase64.MustDecodeToString(c[0])).Map())
|
||||
} else {
|
||||
g.Log("cfg").Error(ctx, "No Metadata from incoming cfg context")
|
||||
}
|
||||
}
|
||||
return a
|
||||
}
|
||||
|
||||
func (a *AtomicAdapter) SetContent(content map[string]interface{}) error {
|
||||
a.store.Store(content)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *AtomicAdapter) Available(ctx context.Context, resource ...string) (ok bool) {
|
||||
return a.store.Load() != nil
|
||||
}
|
||||
|
||||
func (a *AtomicAdapter) Get(ctx context.Context, pattern string) (value interface{}, err error) {
|
||||
raw := a.store.Load()
|
||||
if raw == nil {
|
||||
return nil, errors.New("config not loaded")
|
||||
}
|
||||
data := raw.(map[string]interface{})
|
||||
if v, ok := data[pattern]; ok {
|
||||
return gvar.New(v), nil
|
||||
}
|
||||
return gvar.New(nil), nil
|
||||
}
|
||||
|
||||
func (a *AtomicAdapter) Data(ctx context.Context) (data map[string]interface{}, err error) {
|
||||
raw := a.store.Load()
|
||||
if raw == nil {
|
||||
return nil, errors.New("config not loaded")
|
||||
}
|
||||
original := raw.(map[string]interface{})
|
||||
clone := make(map[string]interface{}, len(original))
|
||||
for k, v := range original {
|
||||
clone[k] = v
|
||||
}
|
||||
return clone, nil
|
||||
}
|
||||
50
apapter/context.go
Normal file
50
apapter/context.go
Normal file
@ -0,0 +1,50 @@
|
||||
package apapter
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/gogf/gf/v2/container/gvar"
|
||||
"github.com/gogf/gf/v2/encoding/gbase64"
|
||||
"github.com/gogf/gf/v2/encoding/gjson"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"google.golang.org/grpc/metadata"
|
||||
)
|
||||
|
||||
/**
|
||||
* @Description: ContextAdapter
|
||||
* @author dc.To
|
||||
* @version 20250807
|
||||
*/
|
||||
type ContextAdapter struct {
|
||||
ctx context.Context
|
||||
adapter string
|
||||
}
|
||||
|
||||
func NewContextAdapter(ctx context.Context, adapter string) *ContextAdapter {
|
||||
return &ContextAdapter{ctx: ctx, adapter: adapter}
|
||||
}
|
||||
|
||||
func (a *ContextAdapter) ctxWithConfig(ctx context.Context) *gvar.Var {
|
||||
md, ok := metadata.FromIncomingContext(ctx)
|
||||
if ok {
|
||||
if c := md.Get(a.adapter); len(c) > 0 {
|
||||
return gvar.New(gbase64.MustDecodeToString(c[0]))
|
||||
}
|
||||
}
|
||||
g.Log("cfg").Error(ctx, "No Metadata from ["+a.adapter+"] cfg context")
|
||||
|
||||
return gvar.New(nil)
|
||||
}
|
||||
|
||||
func (a *ContextAdapter) Available(ctx context.Context, resource ...string) (ok bool) {
|
||||
return a.ctxWithConfig(ctx) != nil
|
||||
}
|
||||
|
||||
func (a *ContextAdapter) Get(ctx context.Context, pattern string) (value interface{}, err error) {
|
||||
|
||||
return gjson.New(a.ctxWithConfig(ctx)).Get(pattern).Val(), nil
|
||||
}
|
||||
|
||||
func (a *ContextAdapter) Data(ctx context.Context) (data map[string]interface{}, err error) {
|
||||
return gjson.New(a.ctxWithConfig(ctx)).Map(), nil
|
||||
}
|
||||
@ -3,11 +3,9 @@ package chain
|
||||
import (
|
||||
"context"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/v2/encoding/gbase64"
|
||||
"git.blueorigin.work/blueorigin/invoke/apapter"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gcfg"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/metadata"
|
||||
)
|
||||
@ -18,19 +16,16 @@ import (
|
||||
* @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, 3*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...)
|
||||
}
|
||||
|
||||
@ -41,7 +36,6 @@ func ClientContextInvokerChain(ctx context.Context, method string, req, res inte
|
||||
* @version 20250425
|
||||
*/
|
||||
func ClientContextUnaryChain(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
|
||||
|
||||
return handler(ctx, req)
|
||||
}
|
||||
|
||||
@ -51,20 +45,16 @@ func ClientContextUnaryChain(ctx context.Context, req interface{}, info *grpc.Un
|
||||
* @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)
|
||||
}
|
||||
g.Cfg(strings.TrimPrefix(k, "x-cfg-")).SetAdapter(apapter.NewContextAdapter(ctx, k))
|
||||
}
|
||||
}
|
||||
} else {
|
||||
g.Log("unary").Error(ctx, "No Metadata from incoming cfg context")
|
||||
}
|
||||
return handler(ctx, req)
|
||||
}
|
||||
|
||||
@ -1,26 +1,62 @@
|
||||
package config
|
||||
package grpcx
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"git.blueorigin.work/blueorigin/invoke/chain"
|
||||
"git.blueorigin.work/blueorigin/protobuf"
|
||||
"github.com/gogf/gf/contrib/registry/etcd/v2"
|
||||
"github.com/gogf/gf/contrib/rpc/grpcx/v2"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/reflection"
|
||||
)
|
||||
|
||||
var Endpoint = os.Getenv("ETCD_ENDPOINTS")
|
||||
var Endpoints = os.Getenv("ETCD_ENDPOINTS")
|
||||
|
||||
var GrpcConfig *grpcx.GrpcServerConfig = grpcx.Server.NewConfig()
|
||||
|
||||
func Etcd() *etcd.Registry {
|
||||
return etcd.New(Endpoint)
|
||||
return etcd.New(Endpoints)
|
||||
}
|
||||
|
||||
func Registry(f ...func()) {
|
||||
if Endpoint != "" {
|
||||
func SetOptions(o ...grpc.ServerOption) *grpcx.GrpcServerConfig {
|
||||
|
||||
GrpcConfig.Options = append(GrpcConfig.Options, o...)
|
||||
return GrpcConfig
|
||||
}
|
||||
|
||||
/**
|
||||
* 服务注册
|
||||
* @param void
|
||||
* @author dc.To
|
||||
* @version 20250427
|
||||
*/
|
||||
func Registry(f ...func(s *grpcx.GrpcServer)) *grpcx.GrpcServer {
|
||||
GrpcConfig.Options = append(GrpcConfig.Options, []grpc.ServerOption{
|
||||
grpcx.Server.ChainUnary(
|
||||
grpcx.Server.UnaryValidate,
|
||||
chain.ClientContextUnaryChain,
|
||||
chain.ClientMetadataUnaryChain,
|
||||
)}...,
|
||||
)
|
||||
if Endpoints != "" {
|
||||
grpcx.Resolver.Register(Etcd())
|
||||
}
|
||||
for _, fn := range f {
|
||||
fn()
|
||||
|
||||
if GrpcConfig.Name == "default" || GrpcConfig.Name == "" {
|
||||
GrpcConfig.Name = filepath.Base(os.Args[0])
|
||||
}
|
||||
|
||||
s := grpcx.Server.New(GrpcConfig)
|
||||
|
||||
for _, fn := range f {
|
||||
fn(s)
|
||||
}
|
||||
|
||||
Reflect(s.Server)
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
/**
|
||||
@ -29,7 +65,7 @@ func Registry(f ...func()) {
|
||||
* @version 20250427
|
||||
*/
|
||||
func Reflect(s reflection.GRPCServer, f ...func()) {
|
||||
if Endpoint == "" {
|
||||
if Endpoints == "" {
|
||||
reflection.Register(s)
|
||||
}
|
||||
for _, fn := range f {
|
||||
@ -37,9 +73,11 @@ func Reflect(s reflection.GRPCServer, f ...func()) {
|
||||
}
|
||||
}
|
||||
|
||||
func Resolve[T any](value T, f func(T) T) T {
|
||||
if f == nil {
|
||||
return value
|
||||
}
|
||||
return f(value)
|
||||
/**
|
||||
* BSS 双向通信流
|
||||
* @author dc.To
|
||||
* @version 20250410
|
||||
*/
|
||||
func BssClient() protobuf.BssClient {
|
||||
return protobuf.NewBssClient(grpcx.Client.MustNewGrpcClientConn("api"))
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user