go.mod

@@ -1,5 +1,8 @@
-module git.selly.red/Selly-Modules/zookeeper
+module github.com/Selly-Modules/zookeeper
 
 go 1.16
 
-require github.com/samuel/go-zookeeper v0.0.0-20201211165307-7117e9ea2414
+require (
+	github.com/logrusorgru/aurora v2.0.3+incompatible
+	github.com/samuel/go-zookeeper v0.0.0-20201211165307-7117e9ea2414
+)

go.sum

@@ -1,2 +1,4 @@
+github.com/logrusorgru/aurora v2.0.3+incompatible h1:tOpm7WcpBTn4fjmVfgpQq0EfczGlG91VSDkswnjF5A8=
+github.com/logrusorgru/aurora v2.0.3+incompatible/go.mod h1:7rIyQOR62GCctdiQpZ/zOJlFyk6y+94wXzv6RNZgaR4=
 github.com/samuel/go-zookeeper v0.0.0-20201211165307-7117e9ea2414 h1:AJNDS0kP60X8wwWFvbLPwDuojxubj9pbfK7pjHw0vKg=
 github.com/samuel/go-zookeeper v0.0.0-20201211165307-7117e9ea2414/go.mod h1:gi+0XIa01GRL2eRQVjQkKGqKF3SF9vZR/HnPullcV2E=

zookeeper.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"time"
 
+	"github.com/logrusorgru/aurora"
 	"github.com/samuel/go-zookeeper/zk"
 )
 
@@ -11,16 +12,20 @@ var zkClient *zk.Conn
 
 // Connect ...
 func Connect(uri string) error {
-	c, _, err := zk.Connect([]string{uri}, time.Second*30, zk.WithLogInfo(false))
+	c, _, err := zk.Connect([]string{uri}, time.Second*30)
 	if err != nil {
 		fmt.Println("Error when connect to Zookeeper", uri, err)
 		return err
 	}
 
-	fmt.Printf("⚡️[zookeeper]: connected to %s \n", uri)
+	fmt.Println(aurora.Green("*** CONNECTED TO ZOOKEEPER: " + uri))
 
 	// Set client
 	zkClient = c
 
 	return nil
 }
+
+
+
+