diff --git a/README.md b/README.md index d5b500d..bf3f6b4 100644 --- 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 index 4598936..784497b 100644 --- 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)