1

registry/consul_registry.go

@@ -350,3 +350,7 @@ func (c *consulRegistry) String() string {
 func (c *consulRegistry) Options() Options {
 	return c.opts
 }
+
+func (c *consulRegistry) GetConsulClient() *consul.Client {
+	return c.Client
+}

registry/etcdv3/etcdv3.go

@@ -6,6 +6,7 @@ import (
 	"crypto/tls"
 	"encoding/json"
 	"errors"
+	consul "github.com/hashicorp/consul/api"
 	"path"
 	"strings"
 	"sync"
@@ -307,3 +308,7 @@ func NewRegistry(opts ...registry.Option) registry.Registry {
 	configure(e, opts...)
 	return e
 }
+
+func (e *etcdv3Registry) GetConsulClient() *consul.Client {
+	return nil
+}

registry/mock/mock.go

@@ -1,7 +1,9 @@
 package mock
 
-import ()
-import "github.com/huyangv/vmqant/registry"
+import (
+	consul "github.com/hashicorp/consul/api"
+	"github.com/huyangv/vmqant/registry"
+)
 
 type mockRegistry struct {
 	Services map[string][]*registry.Service
@@ -111,3 +113,7 @@ func NewRegistry(opts ...registry.Options) registry.Registry {
 	m.init()
 	return m
 }
+
+func (m *mockRegistry) GetConsulClient() *consul.Client {
+	return nil
+}

registry/registry.go

@@ -3,6 +3,7 @@ package registry
 
 import (
 	"errors"
+	consul "github.com/hashicorp/consul/api"
 )
 
 // Registry The registry provides an interface for service discovery
@@ -17,6 +18,7 @@ type Registry interface {
 	ListServices() ([]*Service, error)
 	Watch(...WatchOption) (Watcher, error)
 	String() string
+	GetConsulClient() *consul.Client
 }
 
 // Option Option