From aa867235218e75cce6ff97e10e497456e51ad1c5 Mon Sep 17 00:00:00 2001
From: reshke <reshke@double.cloud>
Date: Fri, 21 Feb 2025 05:43:11 +0000
Subject: [PATCH 1/2] Yezzey: fix truncate metadata

---
 expected/yezzey-trunc.out | 12 ++++++++++++
 smgr.c                    |  1 +
 sql/yezzey-trunc.sql      |  2 ++
 yezzey.c                  |  8 ++++++++
 4 files changed, 23 insertions(+)

diff --git a/expected/yezzey-trunc.out b/expected/yezzey-trunc.out
index 30d0a73..12e4ce1 100644
--- a/expected/yezzey-trunc.out
+++ b/expected/yezzey-trunc.out
@@ -74,6 +74,12 @@ SELECT count(1) FROM trunc_regaoty;
 (1 row)
 
 TRUNCATE trunc_regaoty;
+SELECT count(1) from yezzey_dump_virtual_index('trunc_regaoty');
+ count 
+-------
+     0
+(1 row)
+
 SELECT count(1) FROM trunc_regaoty;
  count 
 -------
@@ -183,6 +189,12 @@ SELECT count(1) FROM trunc_regaocsty;
 (1 row)
 
 TRUNCATE trunc_regaocsty;
+SELECT count(1) from yezzey_dump_virtual_index('trunc_regaoty');
+ count 
+-------
+     0
+(1 row)
+
 SELECT count(1) FROM trunc_regaocsty;
  count 
 -------
diff --git a/smgr.c b/smgr.c
index f58ed6f..842b43d 100644
--- a/smgr.c
+++ b/smgr.c
@@ -155,6 +155,7 @@ void yezzey_unlink(RelFileNodeBackend rnode, ForkNumber forkNum, bool isRedo,
 #else
   if (rnode.node.spcNode == YEZZEYTABLESPACE_OID) {
     /*do nothing */
+    
     return;
   }
 
diff --git a/sql/yezzey-trunc.sql b/sql/yezzey-trunc.sql
index 550d5eb..7d842e8 100644
--- a/sql/yezzey-trunc.sql
+++ b/sql/yezzey-trunc.sql
@@ -27,6 +27,7 @@ INSERT INTO trunc_regaoty SELECT * FROM generate_series(1, 100000);
 SELECT count(1) FROM trunc_regaoty;
 
 TRUNCATE trunc_regaoty;
+SELECT count(1) from yezzey_dump_virtual_index('trunc_regaoty');
 SELECT count(1) FROM trunc_regaoty;
 
 SELECT count() FROM yezzey_offload_relation_status('trunc_regaoty');
@@ -67,6 +68,7 @@ INSERT INTO trunc_regaocsty SELECT * FROM generate_series(1, 100000);
 SELECT count(1) FROM trunc_regaocsty;
 
 TRUNCATE trunc_regaocsty;
+SELECT count(1) from yezzey_dump_virtual_index('trunc_regaoty');
 SELECT count(1) FROM trunc_regaocsty;
 
 SELECT count() FROM yezzey_offload_relation_status('trunc_regaocsty');
diff --git a/yezzey.c b/yezzey.c
index 9782a98..8e6706c 100644
--- a/yezzey.c
+++ b/yezzey.c
@@ -1320,6 +1320,12 @@ static void yezzey_define_gucs() {
                           PGC_SUSET, 0, NULL, NULL, NULL);
 }
 
+void yezzey_TrackObjDrop (Relation rel)
+{
+  if (rel->rd_node.spcNode == YEZZEYTABLESPACE_OID)
+    (void)emptyYezzeyIndex(YezzeyFindAuxIndex(RelationGetRelid(rel)), rel->rd_node.relNode);
+}
+
 void _PG_init(void) {
   /* Allocate shared memory for yezzey workers */
 
@@ -1340,4 +1346,6 @@ void _PG_init(void) {
 
   ExecutorStart_hook = yezzey_ExecuterStartHook;
   ExecutorEnd_hook = yezzey_ExecuterEndHook;
+
+  TrackDropObject_hook = yezzey_TrackObjDrop;
 }
\ No newline at end of file

From 32438c848e3241035256c283360e70c2f13346a0 Mon Sep 17 00:00:00 2001
From: reshke <reshke@double.cloud>
Date: Fri, 21 Feb 2025 08:02:48 +0000
Subject: [PATCH 2/2] fix

---
 src/virtual_tablespace.cpp | 7 ++++---
 yezzey.c                   | 2 +-
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/virtual_tablespace.cpp b/src/virtual_tablespace.cpp
index 798852d..9fc1bbf 100644
--- a/src/virtual_tablespace.cpp
+++ b/src/virtual_tablespace.cpp
@@ -68,13 +68,14 @@ void YezzeyATExecSetTableSpace(Relation aorel, Oid reloid,
    */
 
   /* drop old relation, and close new one */
-  RelationDropStorage(aorel);
+  if (desttablespace_oid == YEZZEYTABLESPACE_OID)
+    RelationDropStorage(aorel);
 
   /* update the pg_class row */
-  if (desttablespace_oid != YEZZEYTABLESPACE_OID) {
+  if (desttablespace_oid != YEZZEYTABLESPACE_OID)
     rd_rel->relfilenode = GetNewRelFileNode(desttablespace_oid, NULL,
                                             aorel->rd_rel->relpersistence);
-  }
+
   rd_rel->reltablespace = desttablespace_oid;
 
 #if IsGreenplum6
diff --git a/yezzey.c b/yezzey.c
index 8e6706c..f75e683 100644
--- a/yezzey.c
+++ b/yezzey.c
@@ -333,7 +333,7 @@ int yezzey_load_relation_internal(Oid reloid, const char *dest_path) {
   * Do not empty, just drop
   * Dropping yezzey virtual index for loaded relation allow to
   * remove extension dependecy on this object (relation)
-  * empty all track info **
+  * empty all track info
   */
  
   (void)emptyYezzeyIndex(yandexoid, origrelfilenode);