unistack-org/micro
4
0
Fork 2
Code Issues 18 Pull Requests Actions Packages Projects Releases 331 Wiki Activity

Allow configurable addresses for everything

Browse Source
This commit is contained in:
Asim
2015-05-16 00:34:02 +01:00
parent c77be7c571
commit 0e7bd77f4c
13 changed files with 143 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
cmd/cmd.go
View File

@@ -2,10 +2,12 @@ package cmd
import (
"os"
"strings"
"text/tabwriter"
"text/template"
"github.com/codegangsta/cli"
"github.com/myodc/go-micro/broker"
"github.com/myodc/go-micro/registry"
"github.com/myodc/go-micro/server"
"github.com/myodc/go-micro/store"
@@ -13,27 +15,81 @@ import (
var (
Flags = []cli.Flag{
cli.StringFlag{Name: "bind_address", Value: ":0", Usage: "Bind address for the server. 127.0.0.1:8080"},
cli.StringFlag{Name: "registry", Value: "consul", Usage: "Registry for discovery. kubernetes, consul, etc"},
cli.StringFlag{Name: "store", Value: "consul", Usage: "Store used as a basic key/value store using consul, memcached, etc"},
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",
Value: ":0",
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",
Value: "127.0.0.1:8500",
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",
Value: "127.0.0.1:8500",
Usage: "Comma-separated list of store addresses",
},
}
)
func Setup(c *cli.Context) error {
server.Address = c.String("bind_address")
server.Address = c.String("server_address")
broker_addrs := strings.Split(c.String("broker_address"), ",")
switch c.String("broker") {
case "http":
broker.DefaultBroker = broker.NewHttpBroker(broker_addrs)
case "nats":
broker.DefaultBroker = broker.NewNatsBroker(broker_addrs)
}
registry_addrs := strings.Split(c.String("registry_address"), ",")
switch c.String("registry") {
case "kubernetes":
registry.DefaultRegistry = registry.NewKubernetesRegistry()
registry.DefaultRegistry = registry.NewKubernetesRegistry(registry_addrs)
case "consul":
registry.DefaultRegistry = registry.NewConsulRegistry(registry_addrs)
}
store_addrs := strings.Split(c.String("store_address"), ",")
switch c.String("store") {
case "memcached":
store.DefaultStore = store.NewMemcacheStore()
store.DefaultStore = store.NewMemcacheStore(store_addrs)
case "memory":
store.DefaultStore = store.NewMemoryStore()
store.DefaultStore = store.NewMemoryStore(store_addrs)
case "etcd":
store.DefaultStore = store.NewEtcdStore()
store.DefaultStore = store.NewEtcdStore(store_addrs)
}
return nil