148 lines
3.5 KiB
Go
148 lines
3.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
"text/tabwriter"
|
|
"text/template"
|
|
|
|
"github.com/codegangsta/cli"
|
|
"github.com/myodc/go-micro/broker"
|
|
"github.com/myodc/go-micro/client"
|
|
"github.com/myodc/go-micro/registry"
|
|
"github.com/myodc/go-micro/server"
|
|
"github.com/myodc/go-micro/store"
|
|
"github.com/myodc/go-micro/transport"
|
|
)
|
|
|
|
var (
|
|
Flags = []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "server_address",
|
|
EnvVar: "MICRO_SERVER_ADDRESS",
|
|
Value: ":0",
|
|
Usage: "Bind address for the server. 127.0.0.1:8080",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "broker",
|
|
EnvVar: "MICRO_BROKER",
|
|
Value: "http",
|
|
Usage: "Broker for pub/sub. http, nats, etc",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "broker_address",
|
|
EnvVar: "MICRO_BROKER_ADDRESS",
|
|
Usage: "Comma-separated list of broker addresses",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "registry",
|
|
EnvVar: "MICRO_REGISTRY",
|
|
Value: "consul",
|
|
Usage: "Registry for discovery. kubernetes, consul, etc",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "registry_address",
|
|
EnvVar: "MICRO_REGISTRY_ADDRESS",
|
|
Usage: "Comma-separated list of registry addresses",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "store",
|
|
EnvVar: "MICRO_STORE",
|
|
Value: "consul",
|
|
Usage: "Store used as a basic key/value store using consul, memcached, etc",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "store_address",
|
|
EnvVar: "MICRO_STORE_ADDRESS",
|
|
Usage: "Comma-separated list of store addresses",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "transport",
|
|
EnvVar: "MICRO_TRANSPORT",
|
|
Value: "http",
|
|
Usage: "Transport mechanism used; http, rabbitmq, etc",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "transport_address",
|
|
EnvVar: "MICRO_TRANSPORT_ADDRESS",
|
|
Usage: "Comma-separated list of transport addresses",
|
|
},
|
|
}
|
|
)
|
|
|
|
func Setup(c *cli.Context) error {
|
|
server.Address = c.String("server_address")
|
|
|
|
bAddrs := strings.Split(c.String("broker_address"), ",")
|
|
|
|
switch c.String("broker") {
|
|
case "http":
|
|
broker.DefaultBroker = broker.NewHttpBroker(bAddrs)
|
|
case "nats":
|
|
broker.DefaultBroker = broker.NewNatsBroker(bAddrs)
|
|
}
|
|
|
|
rAddrs := strings.Split(c.String("registry_address"), ",")
|
|
|
|
switch c.String("registry") {
|
|
case "kubernetes":
|
|
registry.DefaultRegistry = registry.NewKubernetesRegistry(rAddrs)
|
|
case "consul":
|
|
registry.DefaultRegistry = registry.NewConsulRegistry(rAddrs)
|
|
}
|
|
|
|
sAddrs := strings.Split(c.String("store_address"), ",")
|
|
|
|
switch c.String("store") {
|
|
case "memcached":
|
|
store.DefaultStore = store.NewMemcacheStore(sAddrs)
|
|
case "memory":
|
|
store.DefaultStore = store.NewMemoryStore(sAddrs)
|
|
case "etcd":
|
|
store.DefaultStore = store.NewEtcdStore(sAddrs)
|
|
}
|
|
|
|
tAddrs := strings.Split(c.String("transport_address"), ",")
|
|
|
|
switch c.String("transport") {
|
|
case "http":
|
|
transport.DefaultTransport = transport.NewHttpTransport(tAddrs)
|
|
case "rabbitmq":
|
|
transport.DefaultTransport = transport.NewRabbitMQTransport(tAddrs)
|
|
case "nats":
|
|
transport.DefaultTransport = transport.NewNatsTransport(tAddrs)
|
|
}
|
|
|
|
client.DefaultClient = client.New()
|
|
|
|
return nil
|
|
}
|
|
|
|
func Init() {
|
|
cli.AppHelpTemplate = `
|
|
GLOBAL OPTIONS:
|
|
{{range .Flags}}{{.}}
|
|
{{end}}
|
|
`
|
|
|
|
cli.HelpPrinter = func(writer io.Writer, templ string, data interface{}) {
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 8, 1, '\t', 0)
|
|
t := template.Must(template.New("help").Parse(templ))
|
|
err := t.Execute(w, data)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
w.Flush()
|
|
os.Exit(2)
|
|
}
|
|
|
|
app := cli.NewApp()
|
|
app.HideVersion = true
|
|
app.Usage = "a go micro app"
|
|
app.Action = func(c *cli.Context) {}
|
|
app.Before = Setup
|
|
app.Flags = Flags
|
|
app.RunAndExitOnError()
|
|
}
|