Merge pull request #4 from Pantani/pantani/add-db-index

add database index parameter
Pantani · Sep 16, 2020 · 4dd6e6c · 4dd6e6c
2 parents 2b241a5 + 20638f2
commit 4dd6e6c
Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -6,7 +6,7 @@ Initialize the database:
 ```go
 import "github.com/Pantani/redis"
 
-cache := redis.New("localhost:6379‎", "password")
+cache := redis.New("localhost:6379", "password", 0)
 if err != nil {
     panic("Cannot initialize the redis storage")
 }

diff --git a/redis.go b/redis.go
@@ -13,11 +13,11 @@ type Redis struct {
 }
 
 // New returns a new database connection and an error if occurs.
-func New(host, password string) (*Redis, error) {
+func New(host, password string, dbIndex int) (*Redis, error) {
 	rdb := redis.NewClient(&redis.Options{
 		Addr:     host,
 		Password: password,
-		DB:       0,
+		DB:       dbIndex,
 	})
 	_, err := rdb.Ping(context.Background()).Result()
 	if err != nil {