router: add to service options; add dns and static implementations (#1733)

* config/cmd: add router to service options

* router/service: use micro client

router/dns/dns.go

@@ -0,0 +1,130 @@
+package dns
+
+import (
+	"fmt"
+	"net"
+	"strconv"
+
+	"github.com/micro/go-micro/v2/router"
+)
+
+// NewRouter returns an initialized dns router
+func NewRouter(opts ...router.Option) router.Router {
+	options := router.DefaultOptions()
+	for _, o := range opts {
+		o(&options)
+	}
+	if len(options.Network) == 0 {
+		options.Network = "micro"
+	}
+	return &dns{options, &table{options}}
+}
+
+type dns struct {
+	options router.Options
+	table   *table
+}
+
+func (d *dns) Init(opts ...router.Option) error {
+	for _, o := range opts {
+		o(&d.options)
+	}
+	d.table.options = d.options
+	return nil
+}
+
+func (d *dns) Options() router.Options {
+	return d.options
+}
+
+func (d *dns) Table() router.Table {
+	return d.table
+}
+
+func (d *dns) Advertise() (<-chan *router.Advert, error) {
+	return nil, nil
+}
+
+func (d *dns) Process(*router.Advert) error {
+	return nil
+}
+
+func (d *dns) Lookup(opts ...router.QueryOption) ([]router.Route, error) {
+	return d.table.Query(opts...)
+}
+
+func (d *dns) Watch(opts ...router.WatchOption) (router.Watcher, error) {
+	return nil, nil
+}
+
+func (d *dns) Close() error {
+	return nil
+}
+
+func (d *dns) String() string {
+	return "dns"
+}
+
+type table struct {
+	options router.Options
+}
+
+func (t *table) Create(router.Route) error {
+	return nil
+}
+
+func (t *table) Delete(router.Route) error {
+	return nil
+}
+
+func (t *table) Update(router.Route) error {
+	return nil
+}
+
+func (t *table) List() ([]router.Route, error) {
+	return nil, nil
+}
+
+func (t *table) Query(opts ...router.QueryOption) ([]router.Route, error) {
+	options := router.NewQuery(opts...)
+
+	// check to see if we have the port provided in the service, e.g. go-micro-srv-foo:8000
+	host, port, err := net.SplitHostPort(options.Service)
+	if err == nil {
+		// lookup the service using A records
+		ips, err := net.LookupHost(host)
+		if err != nil {
+			return nil, err
+		}
+
+		p, _ := strconv.Atoi(port)
+
+		// convert the ip addresses to routes
+		result := make([]router.Route, len(ips))
+		for i, ip := range ips {
+			result[i] = router.Route{
+				Service: options.Service,
+				Address: fmt.Sprintf("%s:%d", ip, uint16(p)),
+			}
+		}
+		return result, nil
+	}
+
+	// we didn't get the port so we'll lookup the service using SRV records. If we can't lookup the
+	// service using the SRV record, we return the error.
+	_, nodes, err := net.LookupSRV(options.Service, "tcp", t.options.Network)
+	if err != nil {
+		return nil, err
+	}
+
+	// convert the nodes (net services) to routes
+	result := make([]router.Route, len(nodes))
+	for i, n := range nodes {
+		result[i] = router.Route{
+			Service: options.Service,
+			Address: fmt.Sprintf("%s:%d", n.Target, n.Port),
+			Network: t.options.Network,
+		}
+	}
+	return result, nil
+}

router/options.go

@@ -1,8 +1,9 @@
 package router
 
 import (
+	"context"
+
 	"github.com/google/uuid"
-	"github.com/micro/go-micro/v2/client"
 	"github.com/micro/go-micro/v2/registry"
 )
 
@@ -20,8 +21,8 @@ type Options struct {
 	Registry registry.Registry
 	// Advertise is the advertising strategy
 	Advertise Strategy
-	// Client for calling router
-	Client client.Client
+	// Context for additional options
+	Context context.Context
 }
 
 // Id sets Router Id
@@ -38,13 +39,6 @@ func Address(a string) Option {
 	}
 }
 
-// Client to call router service
-func Client(c client.Client) Option {
-	return func(o *Options) {
-		o.Client = c
-	}
-}
-
 // Gateway sets network gateway
 func Gateway(g string) Option {
 	return func(o *Options) {
@@ -81,5 +75,6 @@ func DefaultOptions() Options {
 		Network:   DefaultNetwork,
 		Registry:  registry.DefaultRegistry,
 		Advertise: AdvertiseLocal,
+		Context:   context.Background(),
 	}
 }

router/registry/registry.go

@@ -0,0 +1,8 @@
+package registry
+
+import "github.com/micro/go-micro/v2/router"
+
+// NewRouter returns an initialised registry router
+func NewRouter(opts ...router.Option) router.Router {
+	return router.NewRouter(opts...)
+}

router/service/options.go

@@ -0,0 +1,22 @@
+package service
+
+import (
+	"context"
+
+	"github.com/micro/go-micro/v2/client"
+	"github.com/micro/go-micro/v2/router"
+)
+
+type clientKey struct{}
+
+// Client to call router service
+func Client(c client.Client) router.Option {
+	return func(o *router.Options) {
+		if o.Context == nil {
+			o.Context = context.WithValue(context.Background(), clientKey{}, c)
+			return
+		}
+
+		o.Context = context.WithValue(o.Context, clientKey{}, c)
+	}
+}

router/service/service.go

@@ -36,9 +36,10 @@ func NewRouter(opts ...router.Option) router.Router {
 	// NOTE: might need some client opts here
 	cli := client.DefaultClient
 
-	// set options client
-	if options.Client != nil {
-		cli = options.Client
+	// get options client from the context. We set this in the context to prevent an import loop, as
+	// the client depends on the router
+	if c, ok := options.Context.Value(clientKey{}).(client.Client); ok {
+		cli = c
 	}
 
 	// NOTE: should we have Client/Service option in router.Options?

router/static/static.go

@@ -0,0 +1,88 @@
+package static
+
+import "github.com/micro/go-micro/v2/router"
+
+// NewRouter returns an initialized static router
+func NewRouter(opts ...router.Option) router.Router {
+	options := router.DefaultOptions()
+	for _, o := range opts {
+		o(&options)
+	}
+	return &static{options, new(table)}
+}
+
+type static struct {
+	options router.Options
+	table   router.Table
+}
+
+func (s *static) Init(opts ...router.Option) error {
+	for _, o := range opts {
+		o(&s.options)
+	}
+	return nil
+}
+
+func (s *static) Options() router.Options {
+	return s.options
+}
+
+func (s *static) Table() router.Table {
+	return nil
+}
+
+func (s *static) Advertise() (<-chan *router.Advert, error) {
+	return nil, nil
+}
+
+func (s *static) Process(*router.Advert) error {
+	return nil
+}
+
+func (s *static) Lookup(...router.QueryOption) ([]router.Route, error) {
+	return nil, nil
+}
+
+func (s *static) Watch(opts ...router.WatchOption) (router.Watcher, error) {
+	return nil, nil
+}
+
+func (s *static) Close() error {
+	return nil
+}
+
+func (s *static) String() string {
+	return "static"
+}
+
+type table struct{}
+
+func (t *table) Create(router.Route) error {
+	return nil
+}
+
+func (t *table) Delete(router.Route) error {
+	return nil
+}
+
+func (t *table) Update(router.Route) error {
+	return nil
+}
+
+func (t *table) List() ([]router.Route, error) {
+	return nil, nil
+}
+
+func (t *table) Query(opts ...router.QueryOption) ([]router.Route, error) {
+	options := router.NewQuery(opts...)
+
+	return []router.Route{
+		router.Route{
+			Address: options.Service,
+			Service: options.Address,
+			Gateway: options.Gateway,
+			Network: options.Network,
+			Router:  options.Router,
+		},
+	}, nil
+}