micro/cmd/cmd.go

169 lines
4.0 KiB
Go
Raw Normal View History

package cmd
import (
"io"
"os"
"strings"
"text/tabwriter"
"text/template"
"github.com/codegangsta/cli"
"github.com/myodc/go-micro/broker"
2015-05-21 23:06:01 +01:00
"github.com/myodc/go-micro/client"
2015-05-05 19:05:06 +01:00
"github.com/myodc/go-micro/registry"
"github.com/myodc/go-micro/server"
"github.com/myodc/go-micro/store"
2015-05-20 22:57:19 +01:00
"github.com/myodc/go-micro/transport"
// brokers
"github.com/myodc/go-micro/broker/http"
"github.com/myodc/go-micro/broker/nats"
// registries
"github.com/myodc/go-micro/registry/consul"
"github.com/myodc/go-micro/registry/kubernetes"
// stores
sconsul "github.com/myodc/go-micro/store/consul"
"github.com/myodc/go-micro/store/etcd"
"github.com/myodc/go-micro/store/memcached"
"github.com/myodc/go-micro/store/memory"
// transport
thttp "github.com/myodc/go-micro/transport/http"
tnats "github.com/myodc/go-micro/transport/nats"
"github.com/myodc/go-micro/transport/rabbitmq"
)
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",
},
2015-05-20 22:57:19 +01:00
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")
2015-05-20 22:57:19 +01:00
bAddrs := strings.Split(c.String("broker_address"), ",")
switch c.String("broker") {
case "http":
broker.DefaultBroker = http.NewBroker(bAddrs)
case "nats":
broker.DefaultBroker = nats.NewBroker(bAddrs)
}
2015-05-20 22:57:19 +01:00
rAddrs := strings.Split(c.String("registry_address"), ",")
switch c.String("registry") {
case "kubernetes":
registry.DefaultRegistry = kubernetes.NewRegistry(rAddrs)
case "consul":
registry.DefaultRegistry = consul.NewRegistry(rAddrs)
}
2015-05-20 22:57:19 +01:00
sAddrs := strings.Split(c.String("store_address"), ",")
switch c.String("store") {
case "consul":
store.DefaultStore = sconsul.NewStore(sAddrs)
case "memcached":
store.DefaultStore = memcached.NewStore(sAddrs)
case "memory":
store.DefaultStore = memory.NewStore(sAddrs)
2015-03-06 18:59:24 +00:00
case "etcd":
store.DefaultStore = etcd.NewStore(sAddrs)
2015-05-20 22:57:19 +01:00
}
tAddrs := strings.Split(c.String("transport_address"), ",")
switch c.String("transport") {
case "http":
transport.DefaultTransport = thttp.NewTransport(tAddrs)
2015-05-20 22:57:19 +01:00
case "rabbitmq":
transport.DefaultTransport = rabbitmq.NewTransport(tAddrs)
2015-05-20 22:57:19 +01:00
case "nats":
transport.DefaultTransport = tnats.NewTransport(tAddrs)
}
client.DefaultClient = client.NewClient()
2015-05-21 23:06:01 +01:00
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()
}