client: Expose close function (#311)

centrifuge · Nov 23, 2022 · ccecb66 · ccecb66
1 parent 91238f3
commit ccecb66
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/client/client.go b/client/client.go
@@ -38,6 +38,8 @@ type Client interface {
 		*gethrpc.ClientSubscription, error)
 
 	URL() string
+
+	Close()
 }
 
 type client struct {
@@ -51,6 +53,10 @@ func (c client) URL() string {
 	return c.url
 }
 
+func (c client) Close() {
+	c.Client.Close()
+}
+
 // Connect connects to the provided url
 func Connect(url string) (Client, error) {
 	log.Printf("Connecting to %v...", url)

diff --git a/client/mocks/Client.go b/client/mocks/Client.go