mirror of
https://github.com/go-micro/go-micro.git
synced 2024-12-06 08:16:03 +02:00
218 lines
4.1 KiB
Go
218 lines
4.1 KiB
Go
package web
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/micro/cli"
|
|
"github.com/micro/go-micro"
|
|
"github.com/micro/go-micro/registry"
|
|
)
|
|
|
|
type Options struct {
|
|
Name string
|
|
Version string
|
|
Id string
|
|
Metadata map[string]string
|
|
Address string
|
|
Advertise string
|
|
|
|
Action func(*cli.Context)
|
|
Flags []cli.Flag
|
|
|
|
RegisterTTL time.Duration
|
|
RegisterInterval time.Duration
|
|
|
|
Server *http.Server
|
|
Handler http.Handler
|
|
|
|
// Alternative Options
|
|
Context context.Context
|
|
|
|
Registry registry.Registry
|
|
Service micro.Service
|
|
|
|
Secure bool
|
|
TLSConfig *tls.Config
|
|
BeforeStart []func() error
|
|
BeforeStop []func() error
|
|
AfterStart []func() error
|
|
AfterStop []func() error
|
|
|
|
// Static directory
|
|
StaticDir string
|
|
}
|
|
|
|
func newOptions(opts ...Option) Options {
|
|
opt := Options{
|
|
Name: DefaultName,
|
|
Version: DefaultVersion,
|
|
Id: DefaultId,
|
|
Address: DefaultAddress,
|
|
RegisterTTL: DefaultRegisterTTL,
|
|
RegisterInterval: DefaultRegisterInterval,
|
|
StaticDir: DefaultStaticDir,
|
|
Service: micro.NewService(),
|
|
Context: context.TODO(),
|
|
}
|
|
|
|
for _, o := range opts {
|
|
o(&opt)
|
|
}
|
|
|
|
return opt
|
|
}
|
|
|
|
// Server name
|
|
func Name(n string) Option {
|
|
return func(o *Options) {
|
|
o.Name = n
|
|
}
|
|
}
|
|
|
|
// Unique server id
|
|
func Id(id string) Option {
|
|
return func(o *Options) {
|
|
o.Id = id
|
|
}
|
|
}
|
|
|
|
// Version of the service
|
|
func Version(v string) Option {
|
|
return func(o *Options) {
|
|
o.Version = v
|
|
}
|
|
}
|
|
|
|
// Metadata associated with the service
|
|
func Metadata(md map[string]string) Option {
|
|
return func(o *Options) {
|
|
o.Metadata = md
|
|
}
|
|
}
|
|
|
|
// Address to bind to - host:port
|
|
func Address(a string) Option {
|
|
return func(o *Options) {
|
|
o.Address = a
|
|
}
|
|
}
|
|
|
|
// The address to advertise for discovery - host:port
|
|
func Advertise(a string) Option {
|
|
return func(o *Options) {
|
|
o.Advertise = a
|
|
}
|
|
}
|
|
|
|
// Context specifies a context for the service.
|
|
// Can be used to signal shutdown of the service.
|
|
// Can be used for extra option values.
|
|
func Context(ctx context.Context) Option {
|
|
return func(o *Options) {
|
|
o.Context = ctx
|
|
}
|
|
}
|
|
|
|
func Registry(r registry.Registry) Option {
|
|
return func(o *Options) {
|
|
o.Registry = r
|
|
}
|
|
}
|
|
|
|
func RegisterTTL(t time.Duration) Option {
|
|
return func(o *Options) {
|
|
o.RegisterTTL = t
|
|
}
|
|
}
|
|
|
|
func RegisterInterval(t time.Duration) Option {
|
|
return func(o *Options) {
|
|
o.RegisterInterval = t
|
|
}
|
|
}
|
|
|
|
func Handler(h http.Handler) Option {
|
|
return func(o *Options) {
|
|
o.Handler = h
|
|
}
|
|
}
|
|
|
|
func Server(srv *http.Server) Option {
|
|
return func(o *Options) {
|
|
o.Server = srv
|
|
}
|
|
}
|
|
|
|
// MicroService sets the micro.Service used internally
|
|
func MicroService(s micro.Service) Option {
|
|
return func(o *Options) {
|
|
o.Service = s
|
|
}
|
|
}
|
|
|
|
// Flags sets the command flags.
|
|
func Flags(flags ...cli.Flag) Option {
|
|
return func(o *Options) {
|
|
o.Flags = append(o.Flags, flags...)
|
|
}
|
|
}
|
|
|
|
// Action sets the command action.
|
|
func Action(a func(*cli.Context)) Option {
|
|
return func(o *Options) {
|
|
o.Action = a
|
|
}
|
|
}
|
|
|
|
// BeforeStart is executed before the server starts.
|
|
func BeforeStart(fn func() error) Option {
|
|
return func(o *Options) {
|
|
o.BeforeStart = append(o.BeforeStart, fn)
|
|
}
|
|
}
|
|
|
|
// BeforeStop is executed before the server stops.
|
|
func BeforeStop(fn func() error) Option {
|
|
return func(o *Options) {
|
|
o.BeforeStop = append(o.BeforeStop, fn)
|
|
}
|
|
}
|
|
|
|
// AfterStart is executed after server start.
|
|
func AfterStart(fn func() error) Option {
|
|
return func(o *Options) {
|
|
o.AfterStart = append(o.AfterStart, fn)
|
|
}
|
|
}
|
|
|
|
// AfterStop is executed after server stop.
|
|
func AfterStop(fn func() error) Option {
|
|
return func(o *Options) {
|
|
o.AfterStop = append(o.AfterStop, fn)
|
|
}
|
|
}
|
|
|
|
// Secure Use secure communication. If TLSConfig is not specified we use InsecureSkipVerify and generate a self signed cert
|
|
func Secure(b bool) Option {
|
|
return func(o *Options) {
|
|
o.Secure = b
|
|
}
|
|
}
|
|
|
|
// TLSConfig to be used for the transport.
|
|
func TLSConfig(t *tls.Config) Option {
|
|
return func(o *Options) {
|
|
o.TLSConfig = t
|
|
}
|
|
}
|
|
|
|
// StaticDir sets the static file directory. This defaults to ./html
|
|
func StaticDir(d string) Option {
|
|
return func(o *Options) {
|
|
o.StaticDir = d
|
|
}
|
|
}
|