56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/micro/go-micro"
|
||
|
|
"github.com/micro/go-micro/api/router"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Options struct {
|
||
|
|
Namespace string
|
||
|
|
Router router.Router
|
||
|
|
Service micro.Service
|
||
|
|
}
|
||
|
|
|
||
|
|
type Option func(o *Options)
|
||
|
|
|
||
|
|
// NewOptions fills in the blanks
|
||
|
|
func NewOptions(opts ...Option) Options {
|
||
|
|
var options Options
|
||
|
|
for _, o := range opts {
|
||
|
|
o(&options)
|
||
|
|
}
|
||
|
|
|
||
|
|
// create service if its blank
|
||
|
|
if options.Service == nil {
|
||
|
|
WithService(micro.NewService())(&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
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithService specifies a micro.Service
|
||
|
|
func WithService(s micro.Service) Option {
|
||
|
|
return func(o *Options) {
|
||
|
|
o.Service = s
|
||
|
|
}
|
||
|
|
}
|