use falkordb schema in url

FalkorDB · Apr 9, 2024 · e3842d7 · e3842d7
1 parent d5c0c08
commit e3842d7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/falkordb.go b/falkordb.go
@@ -3,6 +3,7 @@ package falkordb
 import (
 	"context"
 	"errors"
+	"strings"
 
 	"github.com/redis/go-redis/v9"
 )
@@ -57,6 +58,12 @@ func FalkorDBNew(options *ConnectionOption) (*FalkorDB, error) {
 
 // Creates a new FalkorDB instance from a URL.
 func FromURL(url string) (*FalkorDB, error) {
+	if strings.HasPrefix(url, "falkor://") {
+		url = "redis://" + url[len("falkor://"):]
+	} else if strings.HasPrefix(url, "falkors://") {
+		url = "rediss://" + url[len("falkors://"):]
+	}
+
 	options, err := redis.ParseURL(url)
 	if err != nil {
 		return nil, err