新增设置连接实例语法糖

keepchen · Jul 16, 2024 · 8794959 · 8794959
1 parent 620c3c3
commit 8794959
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -27,6 +27,12 @@ corgi.SetRedisProviderCluster(...)
 
 //or
 corgi.SetRedisProviderFailOver(...)
+
+//or
+corgi.SetRedisProviderClient(...)
+
+//or
+corgi.SetRedisProviderClusterClient(...)
 ```  
 #### Lock
 ```go

diff --git a/redislock.go b/redislock.go
@@ -51,6 +51,20 @@ func SetRedisProviderFailOver(opt *redisLib.FailoverOptions) {
 	})
 }
 
+// SetRedisProviderClient 设置redis连接实例(单实例)
+func SetRedisProviderClient(client *redisLib.Client) {
+	doOnce.Do(func() {
+		lockDriver.client = client
+	})
+}
+
+// SetRedisProviderClusterClient 设置redis连接实例(cluster集群)
+func SetRedisProviderClusterClient(client *redisLib.ClusterClient) {
+	doOnce.Do(func() {
+		lockDriver.clusterClient = client
+	})
+}
+
 func initClient(opt *redisLib.Options) {
 	rdb := redisLib.NewClient(opt)