From 85f553e723ee7a15c1700544dfd1ac63d79d9e47 Mon Sep 17 00:00:00 2001
From: Benny Halevy <bhalevy@scylladb.com>
Date: Thu, 24 Aug 2023 11:08:35 +0300
Subject: [PATCH] gossiper: lock_endpoint_update_semaphore: get_units with
 _abort_source

Locking the _endpoint_update_semaphore should be abortable with the
gossiper _abort_source.  No further processing should
be done once abort is requested.

Signed-off-by: Benny Halevy <bhalevy@scylladb.com>
---
 gms/gossiper.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gms/gossiper.cc b/gms/gossiper.cc
index 20198ca65b..da2e766f1d 100644
--- a/gms/gossiper.cc
+++ b/gms/gossiper.cc
@@ -818,7 +818,7 @@ future<semaphore_units<>> gossiper::lock_endpoint_update_semaphore() {
     if (this_shard_id() != 0) {
         on_internal_error(logger, "must be called on shard 0");
     }
-    return get_units(_endpoint_update_semaphore, 1);
+    return get_units(_endpoint_update_semaphore, 1, _abort_source);
 }
 
 future<> gossiper::mutate_live_and_unreachable_endpoints(std::function<void(gossiper&)> func) {