Don't add to defaults in func init, just add them to cmd

This commit is contained in:
Asim 2016-04-26 18:49:02 +01:00
parent f7c57fd4f4
commit 3d3044404e
7 changed files with 26 additions and 35 deletions

View File

@ -2,13 +2,8 @@ package http
import (
"github.com/micro/go-micro/broker"
"github.com/micro/go-micro/cmd"
)
func init() {
cmd.DefaultBrokers["http"] = NewBroker
}
func NewBroker(opts ...broker.Option) broker.Broker {
return broker.NewBroker(opts...)
}

View File

@ -9,13 +9,28 @@ import (
"time"
"github.com/micro/cli"
"github.com/micro/go-micro/broker"
"github.com/micro/go-micro/broker/mqtt"
"github.com/micro/go-micro/client"
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/selector"
"github.com/micro/go-micro/server"
// brokers
"github.com/micro/go-micro/broker"
"github.com/micro/go-micro/broker/http"
"github.com/micro/go-micro/broker/mqtt"
// registries
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/registry/consul"
// selectors
"github.com/micro/go-micro/selector"
"github.com/micro/go-micro/selector/blacklist"
"github.com/micro/go-micro/selector/cache"
"github.com/micro/go-micro/selector/random"
"github.com/micro/go-micro/selector/roundrobin"
// transports
"github.com/micro/go-micro/transport"
thttp "github.com/micro/go-micro/transport/http"
)
type Cmd interface {
@ -118,20 +133,23 @@ var (
}
DefaultBrokers = map[string]func(...broker.Option) broker.Broker{
"http": broker.NewBroker,
"http": http.NewBroker,
"mqtt": mqtt.NewBroker,
}
DefaultRegistries = map[string]func(...registry.Option) registry.Registry{
"consul": registry.NewRegistry,
"consul": consul.NewRegistry,
}
DefaultSelectors = map[string]func(...selector.Option) selector.Selector{
"random": selector.NewSelector,
"cache": cache.NewSelector,
"random": random.NewSelector,
"roundrobin": roundrobin.NewSelector,
"blacklist": blacklist.NewSelector,
}
DefaultTransports = map[string]func(...transport.Option) transport.Transport{
"http": transport.NewTransport,
"http": thttp.NewTransport,
}
// used for default selection as the fall back

View File

@ -1,14 +1,9 @@
package consul
import (
"github.com/micro/go-micro/cmd"
"github.com/micro/go-micro/registry"
)
func init() {
cmd.DefaultRegistries["consul"] = NewRegistry
}
func NewRegistry(opts ...registry.Option) registry.Registry {
return registry.NewRegistry(opts...)
}

View File

@ -5,7 +5,6 @@ import (
"sync"
"time"
"github.com/micro/go-micro/cmd"
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/selector"
)
@ -27,7 +26,6 @@ type blackListSelector struct {
}
func init() {
cmd.DefaultSelectors["blacklist"] = NewSelector
rand.Seed(time.Now().Unix())
}

View File

@ -1,14 +1,9 @@
package random
import (
"github.com/micro/go-micro/cmd"
"github.com/micro/go-micro/selector"
)
func init() {
cmd.DefaultSelectors["random"] = NewSelector
}
func NewSelector(opts ...selector.Option) selector.Selector {
return selector.NewSelector(opts...)
}

View File

@ -3,7 +3,6 @@ package roundrobin
import (
"sync"
"github.com/micro/go-micro/cmd"
"github.com/micro/go-micro/registry"
"github.com/micro/go-micro/selector"
)
@ -12,10 +11,6 @@ type roundRobinSelector struct {
so selector.Options
}
func init() {
cmd.DefaultSelectors["roundrobin"] = NewSelector
}
func (r *roundRobinSelector) Init(opts ...selector.Option) error {
for _, o := range opts {
o(&r.so)

View File

@ -1,14 +1,9 @@
package http
import (
"github.com/micro/go-micro/cmd"
"github.com/micro/go-micro/transport"
)
func init() {
cmd.DefaultTransports["http"] = NewTransport
}
func NewTransport(opts ...transport.Option) transport.Transport {
return transport.NewTransport(opts...)
}