unistack-org/micro-wrapper-requestid
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 27 Wiki Activity

update for latest micro
Some checks failed
build / test (push) Failing after 1m17s
Details
codeql / analyze (go) (push) Failing after 1m37s
Details
build / lint (push) Successful in 9m15s
Details

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2024-04-23 10:24:09 +03:00
parent 376dd1ad3a
commit 8edb53d141
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
requestid.go
View File

@ -81,11 +81,11 @@ var DefaultMetadataFunc = func(ctx context.Context) (context.Context, error) {
return ctx, nil return ctx, nil
} }
type Hook struct{} type hook struct{}
var Wrapper = &Hook{} var Hook = &hook{}
func (w *Hook) ServerSubscriber(next server.FuncSubHandler) server.FuncSubHandler { func (w *hook) ServerSubscriber(next server.FuncSubHandler) server.FuncSubHandler {
return func(ctx context.Context, msg server.Message) error { return func(ctx context.Context, msg server.Message) error {
var err error var err error
if xid, ok := msg.Header()[DefaultMetadataKey]; ok { if xid, ok := msg.Header()[DefaultMetadataKey]; ok {
@ -98,7 +98,7 @@ func (w *Hook) ServerSubscriber(next server.FuncSubHandler) server.FuncSubHandle
} }
} }
func (w *Hook) ServerHandler(next server.FuncHandler) server.FuncHandler { func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler {
return func(ctx context.Context, req server.Request, rsp interface{}) error { return func(ctx context.Context, req server.Request, rsp interface{}) error {
var err error var err error
if ctx, err = DefaultMetadataFunc(ctx); err != nil { if ctx, err = DefaultMetadataFunc(ctx); err != nil {
@ -108,7 +108,7 @@ func (w *Hook) ServerHandler(next server.FuncHandler) server.FuncHandler {
} }
} }
func (w *Hook) ClientBatchPublish(next client.FuncBatchPublish) client.FuncBatchPublish { func (w *hook) ClientBatchPublish(next client.FuncBatchPublish) client.FuncBatchPublish {
return func(ctx context.Context, msgs []client.Message, opts ...client.PublishOption) error { return func(ctx context.Context, msgs []client.Message, opts ...client.PublishOption) error {
var err error var err error
if ctx, err = DefaultMetadataFunc(ctx); err != nil { if ctx, err = DefaultMetadataFunc(ctx); err != nil {
@ -118,7 +118,7 @@ func (w *Hook) ClientBatchPublish(next client.FuncBatchPublish) client.FuncBatch
} }
} }
func (w *Hook) ClientPublish(next client.FuncPublish) client.FuncPublish { func (w *hook) ClientPublish(next client.FuncPublish) client.FuncPublish {
return func(ctx context.Context, msg client.Message, opts ...client.PublishOption) error { return func(ctx context.Context, msg client.Message, opts ...client.PublishOption) error {
var err error var err error
if ctx, err = DefaultMetadataFunc(ctx); err != nil { if ctx, err = DefaultMetadataFunc(ctx); err != nil {
@ -128,7 +128,7 @@ func (w *Hook) ClientPublish(next client.FuncPublish) client.FuncPublish {
} }
} }
func (w *Hook) ClientCall(next client.FuncCall) client.FuncCall { func (w *hook) ClientCall(next client.FuncCall) client.FuncCall {
return func(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error { return func(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
var err error var err error
if ctx, err = DefaultMetadataFunc(ctx); err != nil { if ctx, err = DefaultMetadataFunc(ctx); err != nil {
@ -138,7 +138,7 @@ func (w *Hook) ClientCall(next client.FuncCall) client.FuncCall {
} }
} }
func (w *Hook) ClientStream(next client.FuncStream) client.FuncStream { func (w *hook) ClientStream(next client.FuncStream) client.FuncStream {
return func(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) { return func(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) {
var err error var err error
if ctx, err = DefaultMetadataFunc(ctx); err != nil { if ctx, err = DefaultMetadataFunc(ctx); err != nil {