[add] Added option to run with password (-a)

Author: filipecosta90

go.mod

@@ -7,5 +7,7 @@ require (
 	github.com/RedisGraph/redisgraph-go v1.0.1-0.20210122150500-aa0feaa960ce
 	github.com/golangci/golangci-lint v1.35.2 // indirect
 	github.com/gomodule/redigo v2.0.0+incompatible
+	github.com/mitchellh/gox v1.0.1 // indirect
+	github.com/tcnksm/ghr v0.13.0 // indirect
 	golang.org/x/time v0.0.0-20201208040808-7e3f01d25324
 )

redisgraph-bechmark-go.go

@@ -102,7 +102,7 @@ func main() {
 	host := flag.String("h", "127.0.0.1", "Server hostname.")
 	port := flag.Int("p", 6379, "Server port.")
 	rps := flag.Int64("rps", 0, "Max rps. If 0 no limit is applied and the DB is stressed up to maximum.")
-	//password := flag.String("a", "", "Password for Redis Auth.")
+	password := flag.String("a", "", "Password for Redis Auth.")
 	clients := flag.Uint64("c", 50, "number of clients.")
 	numberRequests := flag.Uint64("n", 1000000, "Total number of requests")
 	debug := flag.Int("debug", 0, "Client debug level.")
@@ -148,7 +148,7 @@ func main() {
 		wg.Add(1)
 		cmd := make([]string, len(args))
 		copy(cmd, args)
-		rgs[client_id], conns[client_id] = getStandaloneConn(*graphKey, "tcp", connectionStr)
+		rgs[client_id], conns[client_id] = getStandaloneConn(*graphKey, "tcp", connectionStr, *password)
 		go ingestionRoutine(&rgs[client_id], true, query, samplesPerClient, *loop, *debug, &wg, useRateLimiter, rateLimiter)
 	}
 

standalone_conn.go

@@ -6,11 +6,16 @@ import (
 	"log"
 )
 
-func getStandaloneConn(graphName, network, addr string) (rg.Graph, redis.Conn) {
-	conn, err := redis.Dial(network, addr)
+func getStandaloneConn(graphName, network, addr string, password string) (graph rg.Graph, conn redis.Conn) {
+	var err error = nil
+	if password != "" {
+		conn, err = redis.Dial(network, addr, redis.DialPassword(password))
+	} else {
+		conn, err = redis.Dial(network, addr)
+	}
 	if err != nil {
 		log.Fatalf("Error preparing for benchmark, while creating new connection. error = %v", err)
 	}
-	graph := rg.GraphNew(graphName, conn)
+	graph = rg.GraphNew(graphName, conn)
 	return graph, conn
 }