change CreateLock

zdw2018 · Aug 13, 2019 · be778fe · be778fe
1 parent ab9076c
commit be778fe
Showing 1 changed file with 19 additions and 9 deletions.
diff --git a/src/Surging.Core/Surging.Core.Consul/ConsulServiceRouteManager.cs b/src/Surging.Core/Surging.Core.Consul/ConsulServiceRouteManager.cs
@@ -182,16 +182,26 @@ private async Task<List<IDistributedLock>> CreateLock()
             var clients = await _consulClientProvider.GetClients();
             foreach (var client in clients)
             {
-                var distributedLock = await client.AcquireLock(new LockOptions($"lock_{_configInfo.RoutePath}")
+                var key = $"lock_{_configInfo.RoutePath}";
+                var writeResult = await client.KV.Get(key);
+                if (writeResult.Response != null)
                 {
-                    SessionTTL = TimeSpan.FromSeconds(_configInfo.LockDelay),
-                    LockTryOnce = true,
-                    LockWaitTime = TimeSpan.FromSeconds(_configInfo.LockDelay)
-                }, _configInfo.LockDelay == 0 ?
-                    default :
-                     new CancellationTokenSource(TimeSpan.FromSeconds(_configInfo.LockDelay)).Token);
-
-                result.Add(distributedLock);
+                    var distributedLock = await client.AcquireLock(key);
+                    result.Add(distributedLock);
+                }
+                else
+                {
+                    var distributedLock = await client.AcquireLock(new LockOptions($"lock_{_configInfo.RoutePath}")
+                    {
+                        SessionTTL = TimeSpan.FromSeconds(_configInfo.LockDelay),
+                        LockTryOnce = true,
+                        LockWaitTime = TimeSpan.FromSeconds(_configInfo.LockDelay)
+                    }, _configInfo.LockDelay == 0 ?
+                        default :
+                         new CancellationTokenSource(TimeSpan.FromSeconds(_configInfo.LockDelay)).Token);
+                    result.Add(distributedLock);
+                }
+
             }
             return result;
         }