You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.3 KiB
44 lines
1.3 KiB
package telegram |
|
|
|
import ( |
|
"context" |
|
|
|
"github.com/gotd/td/bin" |
|
"github.com/gotd/td/tg" |
|
) |
|
|
|
// InvokeFunc implements tg.Invoker as function. |
|
type InvokeFunc func(ctx context.Context, input bin.Encoder, output bin.Decoder) error |
|
|
|
// Invoke implements tg.Invoker. |
|
func (i InvokeFunc) Invoke(ctx context.Context, input bin.Encoder, output bin.Decoder) error { |
|
return i(ctx, input, output) |
|
} |
|
|
|
// Middleware returns new InvokeFunc for next invoker. |
|
type Middleware interface { |
|
Handle(next tg.Invoker) InvokeFunc |
|
} |
|
|
|
// MiddlewareFunc implements Middleware as function. |
|
type MiddlewareFunc func(next tg.Invoker) InvokeFunc |
|
|
|
// Handle implements Middleware. |
|
func (m MiddlewareFunc) Handle(next tg.Invoker) InvokeFunc { return m(next) } |
|
|
|
// chainMiddlewares composes new invoker in such order that first element in |
|
// chain is called first, and latest "next" argument will be "invoker". |
|
// |
|
// E.g. we have invoker and two middlewares, so order will be following: |
|
// 0, 1, (invoker), 1 (after "next"), 0 (after "next"). |
|
// |
|
// See TestMiddlewareOrder or ExampleMiddleware. |
|
func chainMiddlewares(invoker tg.Invoker, chain ...Middleware) tg.Invoker { |
|
if len(chain) == 0 { |
|
return invoker |
|
} |
|
for i := len(chain) - 1; i >= 0; i-- { |
|
invoker = chain[i].Handle(invoker) |
|
} |
|
return invoker |
|
}
|
|
|