micro/api/handler/options.go

71 lines
1.5 KiB
Go
Raw Normal View History

2019-06-03 18:44:43 +01:00
package handler
import (
"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
)
// DefaultMaxRecvSize specifies max recv size for handler
var DefaultMaxRecvSize int64 = 1024 * 1024 * 100 // 10Mb
// Options struct holds handler options
2019-06-03 18:44:43 +01:00
type Options struct {
Router router.Router
2020-04-12 14:29:38 +01:00
Client client.Client
Logger logger.Logger
Namespace string
MaxRecvSize int64
2019-06-03 18:44:43 +01:00
}
// Option func signature
2019-06-03 18:44:43 +01:00
type Option func(o *Options)
// NewOptions creates new options struct and fills it
2019-06-03 18:44:43 +01:00
func NewOptions(opts ...Option) Options {
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
}
}
// 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
}
}
// WithMaxRecvSize specifies max body size
func WithMaxRecvSize(size int64) Option {
return func(o *Options) {
o.MaxRecvSize = size
}
}