2016-12-14 15:41:48 +00:00
|
|
|
// Package cmd is an interface for parsing the command line
|
2015-02-14 23:33:21 +00:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
2015-08-30 02:24:16 +01:00
|
|
|
"fmt"
|
2016-01-01 02:45:15 +00:00
|
|
|
"io"
|
2015-12-05 19:25:36 +00:00
|
|
|
"math/rand"
|
2015-04-27 22:21:56 +01:00
|
|
|
"os"
|
2015-05-16 00:34:02 +01:00
|
|
|
"strings"
|
2015-12-05 19:25:36 +00:00
|
|
|
"time"
|
2015-02-14 23:33:21 +00:00
|
|
|
|
2016-01-01 01:16:21 +00:00
|
|
|
"github.com/micro/cli"
|
2018-12-06 18:19:05 +00:00
|
|
|
"github.com/micro/go-log"
|
2016-04-26 18:49:02 +01:00
|
|
|
"github.com/micro/go-micro/client"
|
|
|
|
"github.com/micro/go-micro/server"
|
|
|
|
|
|
|
|
// brokers
|
2015-11-20 16:17:33 +00:00
|
|
|
"github.com/micro/go-micro/broker"
|
2016-04-26 18:49:02 +01:00
|
|
|
"github.com/micro/go-micro/broker/http"
|
|
|
|
|
|
|
|
// registries
|
2015-11-20 16:17:33 +00:00
|
|
|
"github.com/micro/go-micro/registry"
|
2016-04-26 18:49:02 +01:00
|
|
|
"github.com/micro/go-micro/registry/consul"
|
2018-12-04 17:22:20 +00:00
|
|
|
"github.com/micro/go-micro/registry/gossip"
|
2016-04-27 18:21:05 +01:00
|
|
|
"github.com/micro/go-micro/registry/mdns"
|
2016-04-26 18:49:02 +01:00
|
|
|
|
|
|
|
// selectors
|
2015-12-09 19:32:10 +00:00
|
|
|
"github.com/micro/go-micro/selector"
|
2016-04-26 18:49:02 +01:00
|
|
|
|
|
|
|
// transports
|
2015-11-20 16:17:33 +00:00
|
|
|
"github.com/micro/go-micro/transport"
|
2016-04-26 18:49:02 +01:00
|
|
|
thttp "github.com/micro/go-micro/transport/http"
|
2015-02-14 23:33:21 +00:00
|
|
|
)
|
|
|
|
|
2016-01-01 01:16:21 +00: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 19:12:17 +00:00
|
|
|
Init(opts ...Option) error
|
2016-01-01 01:16:21 +00:00
|
|
|
// Options set within this command
|
|
|
|
Options() Options
|
|
|
|
}
|
|
|
|
|
|
|
|
type cmd struct {
|
|
|
|
opts Options
|
|
|
|
app *cli.App
|
|
|
|
}
|
|
|
|
|
|
|
|
type Option func(o *Options)
|
|
|
|
|
2015-02-14 23:33:21 +00:00
|
|
|
var (
|
2016-01-01 01:16:21 +00:00
|
|
|
DefaultCmd = newCmd()
|
2015-12-23 22:45:40 +00:00
|
|
|
|
2016-01-01 01:16:21 +00:00
|
|
|
DefaultFlags = []cli.Flag{
|
2016-11-18 17:29:26 +00:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "client",
|
|
|
|
EnvVar: "MICRO_CLIENT",
|
|
|
|
Usage: "Client for go-micro; rpc",
|
|
|
|
},
|
2016-03-29 17:18:39 +01:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "client_request_timeout",
|
|
|
|
EnvVar: "MICRO_CLIENT_REQUEST_TIMEOUT",
|
|
|
|
Usage: "Sets the client request timeout. e.g 500ms, 5s, 1m. Default: 5s",
|
|
|
|
},
|
|
|
|
cli.IntFlag{
|
|
|
|
Name: "client_retries",
|
|
|
|
EnvVar: "MICRO_CLIENT_RETRIES",
|
|
|
|
Usage: "Sets the client retries. Default: 1",
|
|
|
|
},
|
2016-06-07 00:46:14 +01:00
|
|
|
cli.IntFlag{
|
|
|
|
Name: "client_pool_size",
|
|
|
|
EnvVar: "MICRO_CLIENT_POOL_SIZE",
|
2017-10-24 15:35:25 +01:00
|
|
|
Usage: "Sets the client connection pool size. Default: 1",
|
2016-06-07 00:46:14 +01:00
|
|
|
},
|
|
|
|
cli.StringFlag{
|
|
|
|
Name: "client_pool_ttl",
|
|
|
|
EnvVar: "MICRO_CLIENT_POOL_TTL",
|
|
|
|
Usage: "Sets the client connection pool ttl. e.g 500ms, 5s, 1m. Default: 1m",
|
|
|
|
},
|
2018-04-06 14:03:39 +01:00
|
|
|
cli.IntFlag{
|
|
|
|
Name: "register_ttl",
|
|
|
|
EnvVar: "MICRO_REGISTER_TTL",
|
|
|
|
Usage: "Register TTL in seconds",
|
|
|
|
},
|
|
|
|
cli.IntFlag{
|
|
|
|
Name: "register_interval",
|
|
|
|
EnvVar: "MICRO_REGISTER_INTERVAL",
|
|
|
|
Usage: "Register interval in seconds",
|
|
|
|
},
|
2018-09-13 19:02:08 +03:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "server",
|
|
|
|
EnvVar: "MICRO_SERVER",
|
|
|
|
Usage: "Server for go-micro; rpc",
|
|
|
|
},
|
2015-05-26 22:39:48 +01:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "server_name",
|
|
|
|
EnvVar: "MICRO_SERVER_NAME",
|
|
|
|
Usage: "Name of the server. go.micro.srv.example",
|
|
|
|
},
|
2015-11-08 01:48:48 +00:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "server_version",
|
|
|
|
EnvVar: "MICRO_SERVER_VERSION",
|
|
|
|
Usage: "Version of the server. 1.1.0",
|
|
|
|
},
|
2015-05-26 22:39:48 +01:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "server_id",
|
|
|
|
EnvVar: "MICRO_SERVER_ID",
|
|
|
|
Usage: "Id of the server. Auto-generated if not specified",
|
|
|
|
},
|
2015-05-16 00:34:02 +01:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "server_address",
|
|
|
|
EnvVar: "MICRO_SERVER_ADDRESS",
|
|
|
|
Usage: "Bind address for the server. 127.0.0.1:8080",
|
|
|
|
},
|
2015-11-11 18:22:04 +00: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-26 22:39:48 +01: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",
|
|
|
|
},
|
2015-05-16 00:34:02 +01:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "broker",
|
|
|
|
EnvVar: "MICRO_BROKER",
|
2015-10-18 12:12:41 +01:00
|
|
|
Usage: "Broker for pub/sub. http, nats, rabbitmq",
|
2015-05-16 00:34:02 +01:00
|
|
|
},
|
|
|
|
cli.StringFlag{
|
|
|
|
Name: "broker_address",
|
|
|
|
EnvVar: "MICRO_BROKER_ADDRESS",
|
|
|
|
Usage: "Comma-separated list of broker addresses",
|
|
|
|
},
|
|
|
|
cli.StringFlag{
|
|
|
|
Name: "registry",
|
|
|
|
EnvVar: "MICRO_REGISTRY",
|
2016-06-28 22:41:44 +01:00
|
|
|
Usage: "Registry for discovery. consul, mdns",
|
2015-05-16 00:34:02 +01:00
|
|
|
},
|
|
|
|
cli.StringFlag{
|
|
|
|
Name: "registry_address",
|
|
|
|
EnvVar: "MICRO_REGISTRY_ADDRESS",
|
|
|
|
Usage: "Comma-separated list of registry addresses",
|
|
|
|
},
|
2015-12-09 19:32:10 +00:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "selector",
|
|
|
|
EnvVar: "MICRO_SELECTOR",
|
2016-05-03 22:11:42 +01:00
|
|
|
Usage: "Selector used to pick nodes for querying",
|
2015-12-09 19:32:10 +00:00
|
|
|
},
|
2015-05-20 22:57:19 +01:00
|
|
|
cli.StringFlag{
|
|
|
|
Name: "transport",
|
|
|
|
EnvVar: "MICRO_TRANSPORT",
|
2016-06-28 22:41:44 +01:00
|
|
|
Usage: "Transport mechanism used; http",
|
2015-05-20 22:57:19 +01:00
|
|
|
},
|
|
|
|
cli.StringFlag{
|
|
|
|
Name: "transport_address",
|
|
|
|
EnvVar: "MICRO_TRANSPORT_ADDRESS",
|
|
|
|
Usage: "Comma-separated list of transport addresses",
|
|
|
|
},
|
2015-04-27 22:21:56 +01:00
|
|
|
}
|
2015-02-14 23:33:21 +00:00
|
|
|
|
2016-03-15 22:12:28 +00:00
|
|
|
DefaultBrokers = map[string]func(...broker.Option) broker.Broker{
|
2016-04-26 18:49:02 +01:00
|
|
|
"http": http.NewBroker,
|
2015-06-01 18:55:27 +01:00
|
|
|
}
|
2015-05-16 00:34:02 +01:00
|
|
|
|
2016-11-18 17:29:26 +00:00
|
|
|
DefaultClients = map[string]func(...client.Option) client.Client{
|
|
|
|
"rpc": client.NewClient,
|
|
|
|
}
|
|
|
|
|
2016-03-15 22:20:21 +00:00
|
|
|
DefaultRegistries = map[string]func(...registry.Option) registry.Registry{
|
2016-04-26 18:49:02 +01:00
|
|
|
"consul": consul.NewRegistry,
|
2018-12-04 17:22:20 +00:00
|
|
|
"gossip": gossip.NewRegistry,
|
2016-04-30 00:20:05 +01:00
|
|
|
"mdns": mdns.NewRegistry,
|
2015-05-16 00:34:02 +01:00
|
|
|
}
|
|
|
|
|
2016-01-01 01:16:21 +00:00
|
|
|
DefaultSelectors = map[string]func(...selector.Option) selector.Selector{
|
2016-05-05 21:14:57 +01:00
|
|
|
"default": selector.NewSelector,
|
2018-12-29 15:44:51 +00:00
|
|
|
"cache": selector.NewSelector,
|
2015-12-09 19:32:10 +00:00
|
|
|
}
|
|
|
|
|
2016-11-18 17:29:26 +00:00
|
|
|
DefaultServers = map[string]func(...server.Option) server.Server{
|
|
|
|
"rpc": server.NewServer,
|
|
|
|
}
|
|
|
|
|
2016-03-15 22:25:32 +00:00
|
|
|
DefaultTransports = map[string]func(...transport.Option) transport.Transport{
|
2016-04-26 18:49:02 +01:00
|
|
|
"http": thttp.NewTransport,
|
2015-06-01 18:55:27 +01:00
|
|
|
}
|
2016-01-27 01:13:31 +00:00
|
|
|
|
|
|
|
// used for default selection as the fall back
|
2016-11-18 17:29:26 +00:00
|
|
|
defaultClient = "rpc"
|
|
|
|
defaultServer = "rpc"
|
2016-01-27 01:13:31 +00:00
|
|
|
defaultBroker = "http"
|
|
|
|
defaultRegistry = "consul"
|
2017-10-24 15:35:25 +01:00
|
|
|
defaultSelector = "cache"
|
2016-01-27 01:13:31 +00:00
|
|
|
defaultTransport = "http"
|
2015-06-01 18:55:27 +01:00
|
|
|
)
|
2015-02-14 23:33:21 +00:00
|
|
|
|
2015-12-05 19:25:36 +00:00
|
|
|
func init() {
|
|
|
|
rand.Seed(time.Now().Unix())
|
2016-01-01 02:45:15 +00:00
|
|
|
help := cli.HelpPrinter
|
|
|
|
cli.HelpPrinter = func(writer io.Writer, templ string, data interface{}) {
|
|
|
|
help(writer, templ, data)
|
|
|
|
os.Exit(0)
|
|
|
|
}
|
2015-12-05 19:25:36 +00:00
|
|
|
}
|
|
|
|
|
2016-01-01 01:16:21 +00:00
|
|
|
func newCmd(opts ...Option) Cmd {
|
|
|
|
options := Options{
|
2016-01-02 00:38:57 +00:00
|
|
|
Broker: &broker.DefaultBroker,
|
|
|
|
Client: &client.DefaultClient,
|
|
|
|
Registry: ®istry.DefaultRegistry,
|
|
|
|
Server: &server.DefaultServer,
|
|
|
|
Selector: &selector.DefaultSelector,
|
|
|
|
Transport: &transport.DefaultTransport,
|
|
|
|
|
2016-01-01 01:16:21 +00:00
|
|
|
Brokers: DefaultBrokers,
|
2016-11-18 17:29:26 +00:00
|
|
|
Clients: DefaultClients,
|
2016-01-01 01:16:21 +00:00
|
|
|
Registries: DefaultRegistries,
|
|
|
|
Selectors: DefaultSelectors,
|
2016-11-18 17:29:26 +00:00
|
|
|
Servers: DefaultServers,
|
2016-01-01 01:16:21 +00:00
|
|
|
Transports: DefaultTransports,
|
|
|
|
}
|
2015-10-18 00:01:43 +01:00
|
|
|
|
2016-01-01 01:16:21 +00:00
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
|
|
|
}
|
2015-10-18 00:01:43 +01:00
|
|
|
|
2016-01-01 02:45:15 +00:00
|
|
|
if len(options.Description) == 0 {
|
|
|
|
options.Description = "a go-micro service"
|
|
|
|
}
|
|
|
|
|
2016-01-01 01:16:21 +00: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 02:45:15 +00:00
|
|
|
|
|
|
|
if len(options.Version) == 0 {
|
|
|
|
cmd.app.HideVersion = true
|
|
|
|
}
|
|
|
|
|
2016-01-01 01:16:21 +00: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 {
|
2016-01-02 00:38:57 +00:00
|
|
|
// If flags are set then use them otherwise do nothing
|
|
|
|
var serverOpts []server.Option
|
|
|
|
var clientOpts []client.Option
|
|
|
|
|
2016-11-18 17:29:26 +00:00
|
|
|
// Set the client
|
|
|
|
if name := ctx.String("client"); len(name) > 0 {
|
2018-08-08 18:57:29 +01:00
|
|
|
// only change if we have the client and type differs
|
|
|
|
if cl, ok := c.opts.Clients[name]; ok && (*c.opts.Client).String() != name {
|
2016-11-18 17:29:26 +00:00
|
|
|
*c.opts.Client = cl()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set the server
|
|
|
|
if name := ctx.String("server"); len(name) > 0 {
|
2018-08-08 18:57:29 +01:00
|
|
|
// only change if we have the server and type differs
|
|
|
|
if s, ok := c.opts.Servers[name]; ok && (*c.opts.Server).String() != name {
|
2016-11-18 17:29:26 +00:00
|
|
|
*c.opts.Server = s()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// Set the broker
|
2018-08-08 18:57:29 +01:00
|
|
|
if name := ctx.String("broker"); len(name) > 0 && (*c.opts.Broker).String() != name {
|
|
|
|
b, ok := c.opts.Brokers[name]
|
|
|
|
if !ok {
|
2016-01-27 01:13:31 +00:00
|
|
|
return fmt.Errorf("Broker %s not found", name)
|
2016-01-02 00:38:57 +00:00
|
|
|
}
|
|
|
|
|
2018-08-08 18:57:29 +01:00
|
|
|
*c.opts.Broker = b()
|
2016-01-02 00:38:57 +00:00
|
|
|
serverOpts = append(serverOpts, server.Broker(*c.opts.Broker))
|
|
|
|
clientOpts = append(clientOpts, client.Broker(*c.opts.Broker))
|
2015-02-14 23:33:21 +00:00
|
|
|
}
|
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// Set the registry
|
2018-08-08 18:57:29 +01:00
|
|
|
if name := ctx.String("registry"); len(name) > 0 && (*c.opts.Registry).String() != name {
|
|
|
|
r, ok := c.opts.Registries[name]
|
|
|
|
if !ok {
|
2016-01-27 01:13:31 +00:00
|
|
|
return fmt.Errorf("Registry %s not found", name)
|
2016-01-02 00:38:57 +00:00
|
|
|
}
|
|
|
|
|
2018-08-08 18:57:29 +01:00
|
|
|
*c.opts.Registry = r()
|
2016-01-02 00:38:57 +00:00
|
|
|
serverOpts = append(serverOpts, server.Registry(*c.opts.Registry))
|
|
|
|
clientOpts = append(clientOpts, client.Registry(*c.opts.Registry))
|
2016-01-04 23:07:56 +00:00
|
|
|
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Selector).Init(selector.Registry(*c.opts.Registry)); err != nil {
|
|
|
|
log.Fatalf("Error configuring registry: %v", err)
|
|
|
|
}
|
|
|
|
|
2016-01-04 23:07:56 +00:00
|
|
|
clientOpts = append(clientOpts, client.Selector(*c.opts.Selector))
|
2016-01-20 15:22:44 +00:00
|
|
|
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Broker).Init(broker.Registry(*c.opts.Registry)); err != nil {
|
|
|
|
log.Fatalf("Error configuring broker: %v", err)
|
|
|
|
}
|
2015-06-01 18:55:27 +01:00
|
|
|
}
|
2015-05-20 22:57:19 +01:00
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// Set the selector
|
2018-08-08 18:57:29 +01:00
|
|
|
if name := ctx.String("selector"); len(name) > 0 && (*c.opts.Selector).String() != name {
|
|
|
|
s, ok := c.opts.Selectors[name]
|
|
|
|
if !ok {
|
2016-01-27 01:13:31 +00:00
|
|
|
return fmt.Errorf("Selector %s not found", name)
|
2016-01-02 00:38:57 +00:00
|
|
|
}
|
|
|
|
|
2018-08-08 18:57:29 +01:00
|
|
|
*c.opts.Selector = s(selector.Registry(*c.opts.Registry))
|
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// No server option here. Should there be?
|
|
|
|
clientOpts = append(clientOpts, client.Selector(*c.opts.Selector))
|
2015-12-09 19:32:10 +00:00
|
|
|
}
|
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// Set the transport
|
2018-08-08 18:57:29 +01:00
|
|
|
if name := ctx.String("transport"); len(name) > 0 && (*c.opts.Transport).String() != name {
|
|
|
|
t, ok := c.opts.Transports[name]
|
|
|
|
if !ok {
|
2016-01-27 01:13:31 +00:00
|
|
|
return fmt.Errorf("Transport %s not found", name)
|
2016-01-02 00:38:57 +00:00
|
|
|
}
|
|
|
|
|
2018-08-08 18:57:29 +01:00
|
|
|
*c.opts.Transport = t()
|
2016-01-02 00:38:57 +00:00
|
|
|
serverOpts = append(serverOpts, server.Transport(*c.opts.Transport))
|
|
|
|
clientOpts = append(clientOpts, client.Transport(*c.opts.Transport))
|
2015-02-14 23:33:21 +00:00
|
|
|
}
|
2015-04-27 22:21:56 +01:00
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// Parse the server options
|
2015-05-26 22:39:48 +01:00
|
|
|
metadata := make(map[string]string)
|
2016-01-01 01:16:21 +00:00
|
|
|
for _, d := range ctx.StringSlice("server_metadata") {
|
2015-05-26 22:39:48 +01: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 00:38:57 +00:00
|
|
|
if len(metadata) > 0 {
|
|
|
|
serverOpts = append(serverOpts, server.Metadata(metadata))
|
|
|
|
}
|
|
|
|
|
2018-08-08 18:57:29 +01:00
|
|
|
if len(ctx.String("broker_address")) > 0 {
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Broker).Init(broker.Addrs(strings.Split(ctx.String("broker_address"), ",")...)); err != nil {
|
|
|
|
log.Fatalf("Error configuring broker: %v", err)
|
|
|
|
}
|
2018-08-08 18:57:29 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if len(ctx.String("registry_address")) > 0 {
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Registry).Init(registry.Addrs(strings.Split(ctx.String("registry_address"), ",")...)); err != nil {
|
|
|
|
log.Fatalf("Error configuring registry: %v", err)
|
|
|
|
}
|
2018-08-08 18:57:29 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
if len(ctx.String("transport_address")) > 0 {
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Transport).Init(transport.Addrs(strings.Split(ctx.String("transport_address"), ",")...)); err != nil {
|
|
|
|
log.Fatalf("Error configuring transport: %v", err)
|
|
|
|
}
|
2018-08-08 18:57:29 +01:00
|
|
|
}
|
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
if len(ctx.String("server_name")) > 0 {
|
|
|
|
serverOpts = append(serverOpts, server.Name(ctx.String("server_name")))
|
|
|
|
}
|
2015-05-26 22:39:48 +01:00
|
|
|
|
2016-01-02 00:38:57 +00: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")))
|
|
|
|
}
|
|
|
|
|
2018-04-06 14:03:39 +01:00
|
|
|
if ttl := time.Duration(ctx.GlobalInt("register_ttl")); ttl > 0 {
|
|
|
|
serverOpts = append(serverOpts, server.RegisterTTL(ttl*time.Second))
|
|
|
|
}
|
|
|
|
|
2016-03-29 17:18:39 +01:00
|
|
|
// client opts
|
2018-07-20 14:20:23 +08:00
|
|
|
if r := ctx.Int("client_retries"); r >= 0 {
|
2016-03-29 17:18:39 +01:00
|
|
|
clientOpts = append(clientOpts, client.Retries(r))
|
|
|
|
}
|
|
|
|
|
|
|
|
if t := ctx.String("client_request_timeout"); len(t) > 0 {
|
|
|
|
d, err := time.ParseDuration(t)
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("failed to parse client_request_timeout: %v", t)
|
|
|
|
}
|
|
|
|
clientOpts = append(clientOpts, client.RequestTimeout(d))
|
|
|
|
}
|
|
|
|
|
2016-06-07 00:46:14 +01:00
|
|
|
if r := ctx.Int("client_pool_size"); r > 0 {
|
|
|
|
clientOpts = append(clientOpts, client.PoolSize(r))
|
|
|
|
}
|
|
|
|
|
|
|
|
if t := ctx.String("client_pool_ttl"); len(t) > 0 {
|
|
|
|
d, err := time.ParseDuration(t)
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("failed to parse client_pool_ttl: %v", t)
|
|
|
|
}
|
|
|
|
clientOpts = append(clientOpts, client.PoolTTL(d))
|
|
|
|
}
|
|
|
|
|
2016-01-02 00:38:57 +00:00
|
|
|
// We have some command line opts for the server.
|
|
|
|
// Lets set it up
|
|
|
|
if len(serverOpts) > 0 {
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Server).Init(serverOpts...); err != nil {
|
|
|
|
log.Fatalf("Error configuring server: %v", err)
|
|
|
|
}
|
2016-01-02 00:38:57 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Use an init option?
|
|
|
|
if len(clientOpts) > 0 {
|
2018-12-06 18:19:05 +00:00
|
|
|
if err := (*c.opts.Client).Init(clientOpts...); err != nil {
|
|
|
|
log.Fatalf("Error configuring client: %v", err)
|
|
|
|
}
|
2016-01-02 00:38:57 +00:00
|
|
|
}
|
2015-08-30 21:22:35 +01:00
|
|
|
|
2015-04-27 22:21:56 +01:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2016-01-02 19:12:17 +00:00
|
|
|
func (c *cmd) Init(opts ...Option) error {
|
2016-01-01 01:16:21 +00:00
|
|
|
for _, o := range opts {
|
|
|
|
o(&c.opts)
|
2015-04-27 22:21:56 +01:00
|
|
|
}
|
2016-01-01 01:16:21 +00:00
|
|
|
c.app.Name = c.opts.Name
|
|
|
|
c.app.Version = c.opts.Version
|
2017-05-09 15:49:59 +08:00
|
|
|
c.app.HideVersion = len(c.opts.Version) == 0
|
2016-01-01 01:16:21 +00:00
|
|
|
c.app.Usage = c.opts.Description
|
|
|
|
c.app.RunAndExitOnError()
|
2016-01-02 19:12:17 +00:00
|
|
|
return nil
|
2016-01-01 01:16:21 +00:00
|
|
|
}
|
2015-04-27 22:21:56 +01:00
|
|
|
|
2016-01-04 11:25:58 +00:00
|
|
|
func DefaultOptions() Options {
|
|
|
|
return DefaultCmd.Options()
|
|
|
|
}
|
|
|
|
|
|
|
|
func App() *cli.App {
|
|
|
|
return DefaultCmd.App()
|
|
|
|
}
|
|
|
|
|
2016-01-02 19:12:17 +00:00
|
|
|
func Init(opts ...Option) error {
|
|
|
|
return DefaultCmd.Init(opts...)
|
2016-01-01 01:16:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func NewCmd(opts ...Option) Cmd {
|
|
|
|
return newCmd(opts...)
|
2015-02-14 23:33:21 +00:00
|
|
|
}
|