From 6c70628616a8ccf696b7a70eb5f321fb0597d32b Mon Sep 17 00:00:00 2001
From: StewKI <andrijastev4@gmail.com>
Date: Sun, 17 May 2026 21:21:43 +0200
Subject: [PATCH] generate script fix

---
 generator/generate.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/generator/generate.cs b/generator/generate.cs
index 468cac2..5e3b196 100644
--- a/generator/generate.cs
+++ b/generator/generate.cs
@@ -19,6 +19,7 @@ var rng = new Random(SEED);
 await using var conn = new MySqlConnection(DSN);
 await conn.OpenAsync();
 Console.WriteLine("Connected.");
+await new MySqlCommand("SET foreign_key_checks=0, unique_checks=0", conn).ExecuteNonQueryAsync();
 
 async Task Exec(string sql)
 {
@@ -480,9 +481,8 @@ while (bookingsDone < BOOKING_COUNT)
         bookingRows.Add($"({guestId},{hotelId},{D(checkin)},{D(checkout)},{S(status)},{DT(created)})");
     }
 
-    // Insert bookings and get the first inserted ID
-    long firstId = await ExecScalar("SELECT AUTO_INCREMENT FROM information_schema.tables WHERE table_schema='hotel_reservations' AND table_name='booking'");
     await Exec($"INSERT INTO booking (guest_id, hotel_id, date_from, date_to, status, created_at) VALUES {string.Join(',', bookingRows)}");
+    long firstId = await ExecScalar("SELECT LAST_INSERT_ID()");
 
     // Re-derive checkin/nights from the same rng sequence is impossible after the fact,
     // so re-parse from inserted rows to build room_bookings