|
|
|
@ -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
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|