From d5991b0b830fa0ed864c234dbc1a2a104c0b0d9a Mon Sep 17 00:00:00 2001
From: robot-clickhouse <robot-clickhouse@users.noreply.github.com>
Date: Fri, 9 Jun 2023 12:16:54 +0000
Subject: [PATCH] Backport #50585 to 23.4: Fix Log family table return wrong
 rows count after truncate

---
 src/Storages/StorageLog.cpp                   |  2 ++
 src/Storages/StorageStripeLog.cpp             |  2 ++
 .../02771_log_faminy_truncate_count.reference |  2 ++
 .../02771_log_faminy_truncate_count.sql       | 26 +++++++++++++++++++
 4 files changed, 32 insertions(+)
 create mode 100644 tests/queries/0_stateless/02771_log_faminy_truncate_count.reference
 create mode 100644 tests/queries/0_stateless/02771_log_faminy_truncate_count.sql

diff --git a/src/Storages/StorageLog.cpp b/src/Storages/StorageLog.cpp
index 8264d67aaba2..583c96c40d44 100644
--- a/src/Storages/StorageLog.cpp
+++ b/src/Storages/StorageLog.cpp
@@ -770,6 +770,8 @@ void StorageLog::truncate(const ASTPtr &, const StorageMetadataPtr &, ContextPtr
 
     marks_loaded = true;
     num_marks_saved = 0;
+    total_rows = 0;
+    total_bytes = 0;
     getContext()->dropMMappedFileCache();
 }
 
diff --git a/src/Storages/StorageStripeLog.cpp b/src/Storages/StorageStripeLog.cpp
index d54725b8b397..ebff9da2b37b 100644
--- a/src/Storages/StorageStripeLog.cpp
+++ b/src/Storages/StorageStripeLog.cpp
@@ -422,6 +422,8 @@ void StorageStripeLog::truncate(const ASTPtr &, const StorageMetadataPtr &, Cont
 
     indices_loaded = true;
     num_indices_saved = 0;
+    total_rows = 0;
+    total_bytes = 0;
     getContext()->dropMMappedFileCache();
 }
 
diff --git a/tests/queries/0_stateless/02771_log_faminy_truncate_count.reference b/tests/queries/0_stateless/02771_log_faminy_truncate_count.reference
new file mode 100644
index 000000000000..aa47d0d46d47
--- /dev/null
+++ b/tests/queries/0_stateless/02771_log_faminy_truncate_count.reference
@@ -0,0 +1,2 @@
+0
+0
diff --git a/tests/queries/0_stateless/02771_log_faminy_truncate_count.sql b/tests/queries/0_stateless/02771_log_faminy_truncate_count.sql
new file mode 100644
index 000000000000..3fb22837f5bb
--- /dev/null
+++ b/tests/queries/0_stateless/02771_log_faminy_truncate_count.sql
@@ -0,0 +1,26 @@
+DROP TABLE IF EXISTS test_log;
+CREATE TABLE test_log
+(
+    `crypto_name` String,
+    `trade_date` Date
+)
+ENGINE = Log;
+
+INSERT INTO test_log (crypto_name, trade_date) VALUES ('abc', '2021-01-01'), ('def', '2022-02-02');
+
+TRUNCATE TABLE test_log;
+SELECT count() FROM  test_log;
+
+DROP TABLE IF EXISTS test_log;
+CREATE TABLE test_log
+(
+    `crypto_name` String,
+    `trade_date` Date
+)
+ENGINE = StripeLog;
+
+INSERT INTO test_log (crypto_name, trade_date) VALUES ('abc', '2021-01-01'), ('def', '2022-02-02');
+
+TRUNCATE TABLE test_log;
+SELECT count() FROM  test_log;
+DROP TABLE test_log;