From 0447b3d29205b4c5397d891be01796211093e42e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Leszczy=C5=84ski?= <2000michal@wp.pl>
Date: Thu, 3 Oct 2024 12:25:32 +0200
Subject: [PATCH] feat(restore): log host shard count

---
 pkg/service/restore/tables_worker.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkg/service/restore/tables_worker.go b/pkg/service/restore/tables_worker.go
index a2fc1f887..16a2c7305 100644
--- a/pkg/service/restore/tables_worker.go
+++ b/pkg/service/restore/tables_worker.go
@@ -184,6 +184,9 @@ func (w *tablesWorker) stageRestoreData(ctx context.Context) error {
 	if err != nil {
 		return errors.Wrap(err, "get hosts shard count")
 	}
+	for h, sh := range hostToShard {
+		w.logger.Info(ctx, "Host shard count", "host", h, "shards", sh)
+	}
 
 	bd := newBatchDispatcher(workload, w.target.BatchSize, hostToShard, w.target.locationHosts)