From 1631bab658985347115a522a35dc159e987bf028 Mon Sep 17 00:00:00 2001
From: Tomasz Grabiec <tgrabiec@scylladb.com>
Date: Wed, 3 Apr 2024 00:06:53 +0200
Subject: [PATCH] compaction: Avoid deprecated sharder::shard_of()

---
 compaction/compaction.cc | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/compaction/compaction.cc b/compaction/compaction.cc
index 8f16a24d29..7b6e014b6b 100644
--- a/compaction/compaction.cc
+++ b/compaction/compaction.cc
@@ -1705,7 +1705,12 @@ public:
     }
 
     compaction_writer create_compaction_writer(const dht::decorated_key& dk) override {
-        auto shard = _sharder->shard_of(dk.token());
+        auto shards = _sharder->shard_for_writes(dk.token());
+        if (shards.size() != 1) {
+            // Resharding is not supposed to run on tablets, so this case does not have to be supported.
+            on_internal_error(clogger, fmt::format("Got {} shards for token {} in table {}.{}", shards.size(), dk.token(), _schema->ks_name(), _schema->cf_name()));
+        }
+        auto shard = shards[0];
         auto sst = _sstable_creator(shard);
         setup_new_sstable(sst);