diff --git a/components.go b/components.go index e9557a0..2b0ce1b 100644 --- a/components.go +++ b/components.go @@ -192,13 +192,13 @@ func (c *Components) Health(context context.Context) HealthInfoMap { return result } -func (c *Components) Wrapper() server.HandlerWrapper { +func (c *Components) WrapHandler() server.HandlerWrapper { return func(h server.HandlerFunc) server.HandlerFunc { return func(ctx context.Context, req server.Request, rsp interface{}) error { ctx = context.WithValue(ctx, ComponentsKey{}, c) for _, component := range c.components { - if err := component.WrapperFunc(ctx, req, rsp); err != nil { + if err := component.WrapHandlerFunc(ctx, req, rsp); err != nil { return err } } diff --git a/components/logrus/logrus.go b/components/logrus/logrus.go index 094adcd..a582099 100644 --- a/components/logrus/logrus.go +++ b/components/logrus/logrus.go @@ -103,7 +103,7 @@ func (c *LogrusComponet) Health(context context.Context) error { return nil } -func (c *LogrusComponet) WrapperFunc(ctx context.Context, req server.Request, rsp interface{}) error { +func (c *LogrusComponet) WrapHandlerFunc(ctx context.Context, req server.Request, rsp interface{}) error { return nil } diff --git a/interface.go b/interface.go index a65cf1c..b88b172 100644 --- a/interface.go +++ b/interface.go @@ -16,5 +16,5 @@ type Component interface { MergeFlags(r *Components, flags []cli.Flag) []cli.Flag Health(context context.Context) error - WrapperFunc(ctx context.Context, req server.Request, rsp interface{}) error + WrapHandlerFunc(ctx context.Context, req server.Request, rsp interface{}) error }