Handle special PostgreSQL directories safely

start.sh

@@ -7,7 +7,21 @@ APP_DATA_DIR=/data/sub2api
 APP_PID=""
 REDIS_PID=""
 
-PG_KEEP_DIRS="
+PG_MARKER_DIRS="
+pg_commit_ts
+pg_dynshmem
+pg_multixact/members
+pg_multixact/offsets
+pg_notify
+pg_serial
+pg_snapshots
+pg_stat
+pg_stat_tmp
+pg_subtrans
+pg_twophase
+"
+
+PG_RECREATE_DIRS="
 pg_commit_ts
 pg_dynshmem
 pg_logical
@@ -31,20 +45,27 @@ pg_wal/summaries
 "
 
 ensure_pg_dirs() {
-  for rel in $PG_KEEP_DIRS; do
+  for rel in $PG_RECREATE_DIRS; do
     dir="$PGDATA/$rel"
-    marker="$dir/.hf-keep"
     if [ ! -d "$dir" ]; then
       mkdir -p "$dir"
       chown postgres:postgres "$dir"
       chmod 700 "$dir"
       echo "Restored PostgreSQL directory: $rel"
     fi
+  done
+
+  for rel in $PG_MARKER_DIRS; do
+    marker="$PGDATA/$rel/.hf-keep"
     if [ ! -f "$marker" ]; then
       printf '%s\n' "HF Spaces persistent directory marker" > "$marker"
       chown postgres:postgres "$marker"
     fi
   done
+
+  for rel in pg_logical pg_logical/mappings pg_logical/snapshots pg_multixact pg_replslot pg_tblspc pg_wal pg_wal/archive_status pg_wal/summaries; do
+    rm -f "$PGDATA/$rel/.hf-keep"
+  done
 }
 
 start_postgres() {
@@ -132,7 +153,7 @@ while kill -0 "$APP_PID" 2>/dev/null; do
     fi
     echo "PostgreSQL recovered"
   fi
-  sleep 5
+  sleep 2
 done
 
 wait "$APP_PID"