fix connection problem of postgres accessory

Author: ramsayleung

config/deploy.yml

@@ -40,6 +40,7 @@ env:
   clear:
     # The hostname is constructed as [service_name]-[accessory_name].
     DB_HOST: telegram_spam_sniper_bot-postgres
+    DB_PORT: 5432
     POSTGRES_USER: rails_user
     POSTGRES_DB: telegram_spam_sniper_bot_production
     # Run the Solid Queue Supervisor inside the web server's Puma process to do jobs.
@@ -101,7 +102,7 @@ accessories:
   postgres:
     image: postgres:15
     host: 192.168.1.188
-    port: "127.0.0.1:5432:5432"
+    port: "5432:5432"
     env:
       clear:
         POSTGRES_USER: rails_user
@@ -145,6 +146,7 @@ accessories:
       - "/tmp/migration_data:/rails/db/data"
       # Mount the SAME storage volume as the main app to access the database
       - "telegram_spam_sniper_bot_storage:/rails/storage"
+
   json_importer:
     host: 192.168.1.188
     image: samrayleung/telegram_spam_sniper_bot
@@ -175,6 +177,7 @@ accessories:
       - "/tmp/migration_data:/rails/db/data"
       # Mount the SAME storage volume as the main app to access the database
       - "telegram_spam_sniper_bot_storage:/rails/storage"
+
   backfiller:
     host: 192.168.1.188
     image: samrayleung/telegram_spam_sniper_bot

lib/tasks/pg_migration.rake

@@ -27,49 +27,43 @@ namespace :pg_migration do
     end
 
     data = JSON.parse(File.read(filepath))
-    now = Time.current # Use the same timestamp for all imported records
 
-    # Import in dependency order
-    if data["trained_messages"]
-      puts "Importing #{data["trained_messages"].length} TrainedMessage records..."
+    ActiveRecord::Base.transaction do
+      # Import in dependency order
 
-      # Prepare the data for bulk insert
-      attributes = data["trained_messages"].map do |record|
-        record.delete("id")
-        # Add timestamps if they don't exist in your JSON data
-        record["created_at"] ||= now
-        record["updated_at"] ||= now
-        record
+      if data["trained_messages"]
+        puts "Importinng TrainedMessage records.."
+        data["trained_messages"].each do |record|
+          record.delete("id")
+          TrainedMessage.insert!(record)
+          print "."
+        end
+        puts "\nImported #{data["trained_messages"].length} TrainedMessage records"
       end
 
-      # Perform the bulk insert
-      TrainedMessage.insert_all!(attributes)
-    end
-
-    # Repeat for BannedUser
-    if data["banned_users"]
-      puts "Importing #{data["banned_users"].length} BannedUser records..."
-      attributes = data["banned_users"].map do |record|
-        record.delete("id")
-        record["created_at"] ||= now
-        record["updated_at"] ||= now
-        record
+      if data["banned_users"]
+        puts "Importing BannedUser records..."
+        data["banned_users"].each do |record|
+          record.delete("id")
+          BannedUser.insert!(record)
+          print "."
+        end
+        puts "\nImported #{data["banned_users"].length} BannedUser records"
       end
-      BannedUser.insert_all!(attributes)
-    end
 
-    # Repeat for GroupClassifierState
-    if data["group_classifier_states"]
-      puts "Importing #{data["group_classifier_states"].length} GroupClassifierStates records..."
-      attributes = data["group_classifier_states"].map do |record|
-        record.delete("id")
-        record["created_at"] ||= now
-        record["updated_at"] ||= now
-        record
+      if data["group_classifier_states"]
+        puts "Importing GroupClassifierStates records..."
+        data["group_classifier_states"].each do |record|
+          # Find the existing record by its unique group_id or prepare a new one
+          state = GroupClassifierState.find_or_initialize_by(group_id: record["group_id"])
+
+          # Update its attributes with the data from the JSON file
+          state.update!(record.except("id"))
+          print "."
+        end
+        puts "\nImported #{data["group_classifier_states"].length} GroupClassifierStates records"
       end
-      GroupClassifierState.insert_all!(attributes)
     end
-
     puts "Migration complete!"
   end
 end