micro/cmd/cmd.go

350 lines
8.8 KiB
Go
Raw Normal View History

package cmd
import (
2015-08-30 04:24:16 +03:00
"flag"
"fmt"
2016-01-01 05:45:15 +03:00
"io"
2015-12-05 22:25:36 +03:00
"math/rand"
"os"
"strings"
2015-12-05 22:25:36 +03:00
"time"
2016-01-01 04:16:21 +03:00
"github.com/micro/cli"
2015-11-20 19:17:33 +03:00
"github.com/micro/go-micro/broker"
"github.com/micro/go-micro/client"
"github.com/micro/go-micro/registry"
2015-12-09 22:32:10 +03:00
"github.com/micro/go-micro/selector"
2015-11-20 19:17:33 +03:00
"github.com/micro/go-micro/server"
"github.com/micro/go-micro/transport"
)
2016-01-01 04:16:21 +03:00
type Cmd interface {
// The cli app within this cmd
App() *cli.App
// Adds options, parses flags and initialise
// exits on error
2016-01-02 22:12:17 +03:00
Init(opts ...Option) error
2016-01-01 04:16:21 +03:00
// Options set within this command
Options() Options
}
type cmd struct {
opts Options
app *cli.App
}
type Option func(o *Options)
var (
2016-01-01 04:16:21 +03:00
DefaultCmd = newCmd()
2015-12-24 01:45:40 +03:00
2016-01-01 04:16:21 +03:00
DefaultFlags = []cli.Flag{
2015-05-27 00:39:48 +03:00
cli.StringFlag{
Name: "server_name",
EnvVar: "MICRO_SERVER_NAME",
Usage: "Name of the server. go.micro.srv.example",
},
2015-11-08 04:48:48 +03:00
cli.StringFlag{
Name: "server_version",
EnvVar: "MICRO_SERVER_VERSION",
Usage: "Version of the server. 1.1.0",
},
2015-05-27 00:39:48 +03:00
cli.StringFlag{
Name: "server_id",
EnvVar: "MICRO_SERVER_ID",
Usage: "Id of the server. Auto-generated if not specified",
},
cli.StringFlag{
Name: "server_address",
EnvVar: "MICRO_SERVER_ADDRESS",
Usage: "Bind address for the server. 127.0.0.1:8080",
},
2015-11-11 21:22:04 +03:00
cli.StringFlag{
Name: "server_advertise",
EnvVar: "MICRO_SERVER_ADVERTISE",
Usage: "Used instead of the server_address when registering with discovery. 127.0.0.1:8080",
},
2015-05-27 00:39:48 +03:00
cli.StringSliceFlag{
Name: "server_metadata",
EnvVar: "MICRO_SERVER_METADATA",
Value: &cli.StringSlice{},
Usage: "A list of key-value pairs defining metadata. version=1.0.0",
},
cli.StringFlag{
Name: "broker",
EnvVar: "MICRO_BROKER",
Usage: "Broker for pub/sub. http, nats, rabbitmq",
},
cli.StringFlag{
Name: "broker_address",
EnvVar: "MICRO_BROKER_ADDRESS",
Usage: "Comma-separated list of broker addresses",
},
cli.StringFlag{
Name: "registry",
EnvVar: "MICRO_REGISTRY",
Usage: "Registry for discovery. memory, consul, etcd, kubernetes",
},
cli.StringFlag{
Name: "registry_address",
EnvVar: "MICRO_REGISTRY_ADDRESS",
Usage: "Comma-separated list of registry addresses",
},
2015-12-09 22:32:10 +03:00
cli.StringFlag{
Name: "selector",
EnvVar: "MICRO_SELECTOR",
Usage: "Selector used to pick nodes for querying. random, roundrobin, blacklist",
},
2015-05-21 00:57:19 +03:00
cli.StringFlag{
Name: "transport",
EnvVar: "MICRO_TRANSPORT",
Usage: "Transport mechanism used; http, rabbitmq, nats",
2015-05-21 00:57:19 +03:00
},
cli.StringFlag{
Name: "transport_address",
EnvVar: "MICRO_TRANSPORT_ADDRESS",
Usage: "Comma-separated list of transport addresses",
},
2015-08-30 04:24:16 +03:00
// logging flags
cli.BoolFlag{
Name: "logtostderr",
Usage: "log to standard error instead of files",
},
cli.BoolFlag{
Name: "alsologtostderr",
Usage: "log to standard error as well as files",
},
cli.StringFlag{
Name: "log_dir",
Usage: "log files will be written to this directory instead of the default temporary directory",
},
cli.StringFlag{
Name: "stderrthreshold",
Usage: "logs at or above this threshold go to stderr",
},
cli.StringFlag{
Name: "v",
Usage: "log level for V logs",
},
cli.StringFlag{
Name: "vmodule",
Usage: "comma-separated list of pattern=N settings for file-filtered logging",
},
cli.StringFlag{
Name: "log_backtrace_at",
Usage: "when logging hits line file:N, emit a stack trace",
},
}
2016-01-01 04:16:21 +03:00
DefaultBrokers = map[string]func([]string, ...broker.Option) broker.Broker{
2015-11-25 03:17:15 +03:00
"http": broker.NewBroker,
}
2016-01-01 04:16:21 +03:00
DefaultRegistries = map[string]func([]string, ...registry.Option) registry.Registry{
2015-11-25 03:17:15 +03:00
"consul": registry.NewRegistry,
}
2016-01-01 04:16:21 +03:00
DefaultSelectors = map[string]func(...selector.Option) selector.Selector{
2015-12-09 22:32:10 +03:00
"random": selector.NewSelector,
}
2016-01-01 04:16:21 +03:00
DefaultTransports = map[string]func([]string, ...transport.Option) transport.Transport{
2015-11-25 03:17:15 +03:00
"http": transport.NewTransport,
}
)
2015-12-05 22:25:36 +03:00
func init() {
rand.Seed(time.Now().Unix())
2016-01-01 05:45:15 +03:00
help := cli.HelpPrinter
cli.HelpPrinter = func(writer io.Writer, templ string, data interface{}) {
help(writer, templ, data)
os.Exit(0)
}
2015-12-05 22:25:36 +03:00
}
2016-01-01 04:16:21 +03:00
func newCmd(opts ...Option) Cmd {
options := Options{
2016-01-02 03:38:57 +03:00
Broker: &broker.DefaultBroker,
Client: &client.DefaultClient,
Registry: &registry.DefaultRegistry,
Server: &server.DefaultServer,
Selector: &selector.DefaultSelector,
Transport: &transport.DefaultTransport,
2016-01-01 04:16:21 +03:00
Brokers: DefaultBrokers,
Registries: DefaultRegistries,
Selectors: DefaultSelectors,
Transports: DefaultTransports,
}
2016-01-01 04:16:21 +03:00
for _, o := range opts {
o(&options)
}
2016-01-01 05:45:15 +03:00
if len(options.Description) == 0 {
options.Description = "a go-micro service"
}
2016-01-01 04:16:21 +03:00
cmd := new(cmd)
cmd.opts = options
cmd.app = cli.NewApp()
cmd.app.Name = cmd.opts.Name
cmd.app.Version = cmd.opts.Version
cmd.app.Usage = cmd.opts.Description
cmd.app.Before = cmd.Before
cmd.app.Flags = DefaultFlags
cmd.app.Action = func(c *cli.Context) {}
2016-01-01 05:45:15 +03:00
if len(options.Version) == 0 {
cmd.app.HideVersion = true
}
2016-01-01 04:16:21 +03:00
return cmd
}
func (c *cmd) App() *cli.App {
return c.app
}
func (c *cmd) Options() Options {
return c.opts
}
func (c *cmd) Before(ctx *cli.Context) error {
// Due to logger issues with glog, we need to do this
os.Args = os.Args[:1]
flag.Set("logtostderr", fmt.Sprintf("%v", ctx.Bool("logtostderr")))
flag.Set("alsologtostderr", fmt.Sprintf("%v", ctx.Bool("alsologtostderr")))
flag.Set("stderrthreshold", ctx.String("stderrthreshold"))
flag.Set("log_backtrace_at", ctx.String("log_backtrace_at"))
flag.Set("log_dir", ctx.String("log_dir"))
flag.Set("vmodule", ctx.String("vmodule"))
flag.Set("v", ctx.String("v"))
flag.Parse()
2016-01-02 03:38:57 +03:00
// If flags are set then use them otherwise do nothing
var serverOpts []server.Option
var clientOpts []client.Option
// Set the broker
if len(ctx.String("broker")) > 0 {
if b, ok := c.opts.Brokers[ctx.String("broker")]; ok {
n := b(strings.Split(ctx.String("broker_address"), ","))
c.opts.Broker = &n
} else {
return fmt.Errorf("Broker %s not found", ctx.String("broker"))
}
serverOpts = append(serverOpts, server.Broker(*c.opts.Broker))
clientOpts = append(clientOpts, client.Broker(*c.opts.Broker))
}
2016-01-02 03:38:57 +03:00
// Set the registry
if len(ctx.String("registry")) > 0 {
if r, ok := c.opts.Registries[ctx.String("registry")]; ok {
n := r(strings.Split(ctx.String("registry_address"), ","))
c.opts.Registry = &n
} else {
return fmt.Errorf("Registry %s not found", ctx.String("registry"))
}
serverOpts = append(serverOpts, server.Registry(*c.opts.Registry))
clientOpts = append(clientOpts, client.Registry(*c.opts.Registry))
}
2015-05-21 00:57:19 +03:00
2016-01-02 03:38:57 +03:00
// Set the selector
if len(ctx.String("selector")) > 0 {
if s, ok := c.opts.Selectors[ctx.String("selector")]; ok {
n := s(selector.Registry(*c.opts.Registry))
c.opts.Selector = &n
} else {
return fmt.Errorf("Selector %s not found", ctx.String("selector"))
}
// No server option here. Should there be?
clientOpts = append(clientOpts, client.Selector(*c.opts.Selector))
2015-12-09 22:32:10 +03:00
}
2016-01-02 03:38:57 +03:00
// Set the transport
if len(ctx.String("transport")) > 0 {
if t, ok := c.opts.Transports[ctx.String("transport")]; ok {
n := t(strings.Split(ctx.String("transport_address"), ","))
c.opts.Transport = &n
} else {
return fmt.Errorf("Transport %s not found", ctx.String("transport"))
}
serverOpts = append(serverOpts, server.Transport(*c.opts.Transport))
clientOpts = append(clientOpts, client.Transport(*c.opts.Transport))
}
2016-01-02 03:38:57 +03:00
// Parse the server options
2015-05-27 00:39:48 +03:00
metadata := make(map[string]string)
2016-01-01 04:16:21 +03:00
for _, d := range ctx.StringSlice("server_metadata") {
2015-05-27 00:39:48 +03:00
var key, val string
parts := strings.Split(d, "=")
key = parts[0]
if len(parts) > 1 {
val = strings.Join(parts[1:], "=")
}
metadata[key] = val
}
2016-01-02 03:38:57 +03:00
if len(metadata) > 0 {
serverOpts = append(serverOpts, server.Metadata(metadata))
}
if len(ctx.String("server_name")) > 0 {
serverOpts = append(serverOpts, server.Name(ctx.String("server_name")))
}
2015-05-27 00:39:48 +03:00
2016-01-02 03:38:57 +03:00
if len(ctx.String("server_version")) > 0 {
serverOpts = append(serverOpts, server.Version(ctx.String("server_version")))
}
if len(ctx.String("server_id")) > 0 {
serverOpts = append(serverOpts, server.Id(ctx.String("server_id")))
}
if len(ctx.String("server_address")) > 0 {
serverOpts = append(serverOpts, server.Address(ctx.String("server_address")))
}
if len(ctx.String("server_advertise")) > 0 {
serverOpts = append(serverOpts, server.Advertise(ctx.String("server_advertise")))
}
// We have some command line opts for the server.
// Lets set it up
if len(serverOpts) > 0 {
(*c.opts.Server).Init(serverOpts...)
}
// Use an init option?
if len(clientOpts) > 0 {
2016-01-02 22:12:17 +03:00
(*c.opts.Client).Init(clientOpts...)
2016-01-02 03:38:57 +03:00
}
2015-08-30 23:22:35 +03:00
return nil
}
2016-01-02 22:12:17 +03:00
func (c *cmd) Init(opts ...Option) error {
2016-01-01 04:16:21 +03:00
for _, o := range opts {
o(&c.opts)
}
2016-01-01 04:16:21 +03:00
c.app.Name = c.opts.Name
c.app.Version = c.opts.Version
c.app.Usage = c.opts.Description
c.app.RunAndExitOnError()
2016-01-02 22:12:17 +03:00
return nil
2016-01-01 04:16:21 +03:00
}
2016-01-02 22:12:17 +03:00
func Init(opts ...Option) error {
return DefaultCmd.Init(opts...)
2016-01-01 04:16:21 +03:00
}
func NewCmd(opts ...Option) Cmd {
return newCmd(opts...)
}