Compare commits
38 Commits
v3.8.3
...
a7a682cfc3
| Author | SHA1 | Date | |
|---|---|---|---|
| a7a682cfc3 | |||
| a2762f4202 | |||
| e376188d06 | |||
|
|
7d8fa279fa | ||
| c6b165fc27 | |||
|
|
cefd04c64c | ||
| c716595b1c | |||
|
|
5b78e842e4 | ||
| 69ac0c5bc0 | |||
|
|
c78998d57c | ||
| 53310928ad | |||
|
|
d2252ccede | ||
| 38e78fbde9 | |||
|
|
6748f177a3 | ||
| 971d9a46a3 | |||
|
|
ee9a96f2f6 | ||
| 737ff8146a | |||
|
|
e70b60ddee | ||
| dc391902e9 | |||
|
|
c3f4cf564c | ||
| 9859c80c5b | |||
|
|
130aa5fe61 | ||
|
|
269e8208df | ||
| ce5de2809c | |||
|
|
d881112f30 | ||
|
|
6616a1c448 | ||
| 982d1013b0 | |||
|
|
7b5e5505d0 | ||
| 648bd8bd73 | |||
|
|
f839ee7e79 | ||
| 93c4f593a3 | |||
|
|
2fe3bafe88 | ||
| 3b50129b45 | |||
|
|
1ce984d8da | ||
| 3da2958c4b | |||
|
|
c5968ca6a2 | ||
| 7f631a4f3d | |||
|
|
0a9274c9a3 |
2
.github/workflows/build.yml
vendored
2
.github/workflows/build.yml
vendored
@@ -34,7 +34,7 @@ jobs:
|
||||
- name: checkout
|
||||
uses: actions/checkout@v3
|
||||
- name: lint
|
||||
uses: golangci/golangci-lint-action@v3.3.1
|
||||
uses: golangci/golangci-lint-action@v3.4.0
|
||||
continue-on-error: true
|
||||
with:
|
||||
# Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version.
|
||||
|
||||
2
.github/workflows/dependabot-automerge.yml
vendored
2
.github/workflows/dependabot-automerge.yml
vendored
@@ -15,7 +15,7 @@ jobs:
|
||||
steps:
|
||||
- name: metadata
|
||||
id: metadata
|
||||
uses: dependabot/fetch-metadata@v1.3.5
|
||||
uses: dependabot/fetch-metadata@v1.3.6
|
||||
with:
|
||||
github-token: "${{ secrets.TOKEN }}"
|
||||
- name: merge
|
||||
|
||||
2
.github/workflows/pr.yml
vendored
2
.github/workflows/pr.yml
vendored
@@ -34,7 +34,7 @@ jobs:
|
||||
- name: checkout
|
||||
uses: actions/checkout@v3
|
||||
- name: lint
|
||||
uses: golangci/golangci-lint-action@v3.3.1
|
||||
uses: golangci/golangci-lint-action@v3.4.0
|
||||
continue-on-error: true
|
||||
with:
|
||||
# Required: the version of golangci-lint is required and must be specified without patch version: we always use the latest patch version.
|
||||
|
||||
7
go.mod
7
go.mod
@@ -2,4 +2,9 @@ module go.unistack.org/micro-wrapper-recovery/v3
|
||||
|
||||
go 1.16
|
||||
|
||||
require go.unistack.org/micro/v3 v3.9.14
|
||||
require (
|
||||
go.unistack.org/micro/v3 v3.10.66
|
||||
golang.org/x/sys v0.19.0 // indirect
|
||||
google.golang.org/genproto/googleapis/rpc v0.0.0-20240415180920-8c6c420018be // indirect
|
||||
google.golang.org/grpc v1.63.2 // indirect
|
||||
)
|
||||
|
||||
100
wrapper.go
100
wrapper.go
@@ -1,4 +1,4 @@
|
||||
package recovery // import "go.unistack.org/micro-wrapper-recovery/v3"
|
||||
package recovery
|
||||
|
||||
import (
|
||||
"context"
|
||||
@@ -8,11 +8,6 @@ import (
|
||||
"go.unistack.org/micro/v3/server"
|
||||
)
|
||||
|
||||
var (
|
||||
_ server.HandlerWrapper = NewServerHandlerWrapper()
|
||||
_ server.SubscriberWrapper = NewServerSubscriberWrapper()
|
||||
)
|
||||
|
||||
func NewOptions(opts ...Option) Options {
|
||||
options := Options{
|
||||
ServerHandlerFn: DefaultServerHandlerFn,
|
||||
@@ -31,13 +26,13 @@ type Options struct {
|
||||
|
||||
type Option func(*Options)
|
||||
|
||||
func ServerHandlerFn(fn func(context.Context, server.Request, interface{}, error) error) Option {
|
||||
func ServerHandlerFunc(fn func(context.Context, server.Request, interface{}, error) error) Option {
|
||||
return func(o *Options) {
|
||||
o.ServerHandlerFn = fn
|
||||
}
|
||||
}
|
||||
|
||||
func ServerSubscriberFn(fn func(context.Context, server.Message, error) error) Option {
|
||||
func ServerSubscriberFunc(fn func(context.Context, server.Message, error) error) Option {
|
||||
return func(o *Options) {
|
||||
o.ServerSubscriberFn = fn
|
||||
}
|
||||
@@ -52,16 +47,17 @@ var (
|
||||
}
|
||||
)
|
||||
|
||||
type wrapper struct {
|
||||
var Hook = NewHook()
|
||||
|
||||
type hook struct {
|
||||
opts Options
|
||||
}
|
||||
|
||||
func NewServerHandlerWrapper(opts ...Option) server.HandlerWrapper {
|
||||
handler := &wrapper{opts: NewOptions(opts...)}
|
||||
return handler.HandlerFunc
|
||||
func NewHook(opts ...Option) *hook {
|
||||
return &hook{opts: NewOptions(opts...)}
|
||||
}
|
||||
|
||||
func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc {
|
||||
func (w *hook) ServerHandler(next server.FuncHandler) server.FuncHandler {
|
||||
return func(ctx context.Context, req server.Request, rsp interface{}) (err error) {
|
||||
defer func() {
|
||||
r := recover()
|
||||
@@ -74,17 +70,12 @@ func (w *wrapper) HandlerFunc(fn server.HandlerFunc) server.HandlerFunc {
|
||||
err = w.opts.ServerHandlerFn(ctx, req, rsp, fmt.Errorf("%v", r))
|
||||
}
|
||||
}()
|
||||
err = fn(ctx, req, rsp)
|
||||
err = next(ctx, req, rsp)
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
func NewServerSubscriberWrapper(opts ...Option) server.SubscriberWrapper {
|
||||
handler := &wrapper{opts: NewOptions(opts...)}
|
||||
return handler.SubscriberFunc
|
||||
}
|
||||
|
||||
func (w *wrapper) SubscriberFunc(fn server.SubscriberFunc) server.SubscriberFunc {
|
||||
func (w *hook) ServerSubscriber(next server.FuncSubHandler) server.FuncSubHandler {
|
||||
return func(ctx context.Context, msg server.Message) (err error) {
|
||||
defer func() {
|
||||
r := recover()
|
||||
@@ -97,74 +88,7 @@ func (w *wrapper) SubscriberFunc(fn server.SubscriberFunc) server.SubscriberFunc
|
||||
err = w.opts.ServerSubscriberFn(ctx, msg, fmt.Errorf("%v", r))
|
||||
}
|
||||
}()
|
||||
err = fn(ctx, msg)
|
||||
err = next(ctx, msg)
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
func NewClientWrapper() client.Wrapper {
|
||||
return func(c client.Client) client.Client {
|
||||
handler := &wrapper{
|
||||
clientClient: c,
|
||||
}
|
||||
return handler
|
||||
}
|
||||
}
|
||||
|
||||
func NewCallWrapper() client.CallWrapper {
|
||||
return func(fn client.CallFunc) client.CallFunc {
|
||||
handler := &wrapper{
|
||||
clientCallFunc: fn,
|
||||
}
|
||||
return handler.CallFunc
|
||||
}
|
||||
}
|
||||
|
||||
func (w *wrapper) CallFunc(ctx context.Context, addr string, req client.Request, rsp interface{}, opts client.CallOptions) (err error) {
|
||||
defer func() {
|
||||
r := recover()
|
||||
switch verr := r.(type) {
|
||||
case nil:
|
||||
return
|
||||
case error:
|
||||
err = w.clientCallFunc(ctx, addr, req, rsp, opts, verr)
|
||||
default:
|
||||
err = w.clientCallFunc(ctx, addr, req, rsp, opts, fmt.Errorf("%v", r))
|
||||
}
|
||||
}()
|
||||
err = w.CallFunc(ctx, addr, req, rsp, opts)
|
||||
return err
|
||||
}
|
||||
|
||||
func (w *wrapper) Call(ctx context.Context, req client.Request, rsp interface{}, opts ...client.CallOption) error {
|
||||
defer func() {
|
||||
r := recover()
|
||||
switch verr := r.(type) {
|
||||
case nil:
|
||||
return
|
||||
case error:
|
||||
err = w.clientClient.Call(ctx, addr, req, rsp, opts, verr)
|
||||
default:
|
||||
err = w.clientClient.Call(ctx, addr, req, rsp, opts, fmt.Errorf("%v", r))
|
||||
}
|
||||
}()
|
||||
err = w.clientClient.Call(ctx, req, rsp, opts...)
|
||||
return err
|
||||
}
|
||||
|
||||
func (w *wrapper) Stream(ctx context.Context, req client.Request, opts ...client.CallOption) (client.Stream, error) {
|
||||
|
||||
stream, err := w.Client.Stream(ctx, req, opts...)
|
||||
|
||||
return stream, err
|
||||
}
|
||||
|
||||
func (w *wrapper) Publish(ctx context.Context, p client.Message, opts ...client.PublishOption) error {
|
||||
|
||||
err := w.Client.Publish(ctx, p, opts...)
|
||||
|
||||
|
||||
return err
|
||||
}
|
||||
*/
|
||||
|
||||
Reference in New Issue
Block a user