2019-06-03 18:44:43 +01:00
|
|
|
package handler
|
|
|
|
|
|
|
|
import (
|
2021-10-02 19:55:07 +03:00
|
|
|
"go.unistack.org/micro/v3/api/router"
|
|
|
|
"go.unistack.org/micro/v3/client"
|
|
|
|
"go.unistack.org/micro/v3/logger"
|
2019-06-03 18:44:43 +01:00
|
|
|
)
|
|
|
|
|
2021-04-27 08:32:47 +03:00
|
|
|
// DefaultMaxRecvSize specifies max recv size for handler
|
|
|
|
var DefaultMaxRecvSize int64 = 1024 * 1024 * 100 // 10Mb
|
2020-03-26 14:29:28 +03:00
|
|
|
|
2021-02-14 16:16:01 +03:00
|
|
|
// Options struct holds handler options
|
2019-06-03 18:44:43 +01:00
|
|
|
type Options struct {
|
2020-03-26 14:29:28 +03:00
|
|
|
Router router.Router
|
2020-04-12 14:29:38 +01:00
|
|
|
Client client.Client
|
2021-02-13 15:35:56 +03:00
|
|
|
Logger logger.Logger
|
2021-03-06 19:45:13 +03:00
|
|
|
Namespace string
|
|
|
|
MaxRecvSize int64
|
2019-06-03 18:44:43 +01:00
|
|
|
}
|
|
|
|
|
2021-02-14 16:16:01 +03:00
|
|
|
// Option func signature
|
2019-06-03 18:44:43 +01:00
|
|
|
type Option func(o *Options)
|
|
|
|
|
2021-02-14 16:16:01 +03:00
|
|
|
// NewOptions creates new options struct and fills it
|
2019-06-03 18:44:43 +01:00
|
|
|
func NewOptions(opts ...Option) Options {
|
2021-02-13 15:35:56 +03:00
|
|
|
options := Options{
|
|
|
|
Client: client.DefaultClient,
|
|
|
|
Router: router.DefaultRouter,
|
|
|
|
Logger: logger.DefaultLogger,
|
|
|
|
MaxRecvSize: DefaultMaxRecvSize,
|
|
|
|
}
|
2019-06-03 18:44:43 +01:00
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
|
|
|
|
|
|
|
// set namespace if blank
|
|
|
|
if len(options.Namespace) == 0 {
|
|
|
|
WithNamespace("go.micro.api")(&options)
|
|
|
|
}
|
|
|
|
|
|
|
|
return options
|
|
|
|
}
|
|
|
|
|
|
|
|
// WithNamespace specifies the namespace for the handler
|
|
|
|
func WithNamespace(s string) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.Namespace = s
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// WithRouter specifies a router to be used by the handler
|
|
|
|
func WithRouter(r router.Router) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.Router = r
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-02-14 16:16:01 +03:00
|
|
|
// WithClient specifies client to be used by the handler
|
2020-04-12 14:29:38 +01:00
|
|
|
func WithClient(c client.Client) Option {
|
2019-06-03 18:44:43 +01:00
|
|
|
return func(o *Options) {
|
2020-04-12 14:29:38 +01:00
|
|
|
o.Client = c
|
2019-06-03 18:44:43 +01:00
|
|
|
}
|
|
|
|
}
|
2020-03-26 14:29:28 +03:00
|
|
|
|
2020-07-16 23:33:11 +08:00
|
|
|
// WithMaxRecvSize specifies max body size
|
2020-03-26 14:29:28 +03:00
|
|
|
func WithMaxRecvSize(size int64) Option {
|
|
|
|
return func(o *Options) {
|
|
|
|
o.MaxRecvSize = size
|
|
|
|
}
|
|
|
|
}
|