From 225bb9f7b7677eed64310a5deef8253afeacd176 Mon Sep 17 00:00:00 2001
From: Jasder <2053003901@@qq.com>
Date: Tue, 24 Nov 2020 17:00:42 +0800
Subject: [PATCH] Update table structure's rake task

---
 lib/tasks/sync_table_structure.rake | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/lib/tasks/sync_table_structure.rake b/lib/tasks/sync_table_structure.rake
index e1a775c5f..026d52140 100644
--- a/lib/tasks/sync_table_structure.rake
+++ b/lib/tasks/sync_table_structure.rake
@@ -4,7 +4,17 @@ namespace :sync_table_structure do
   task import_csv: :environment do
     puts "init table structure......."
 
-    system "mysql -uroot -poracle10g -h127.0.0.1 forge_development < #{Rails.root}/db/structure.sql"
+    database_config = Rails.configuration.database_configuration
+
+    database = database_config[Rails.env]["database"]
+    database_username = database_config[Rails.env]["username"]
+    database_password = database_config[Rails.env]["password"]
+    database_host = database_config[Rails.env]["host"]
+    database_port = database_config[Rails.env]["port"] || 3306
+
+    puts "bash: mysql -u#{database_username} -p#{database_password} -P#{database_port} -h#{database_host} #{database}"
+
+    system "mysql -u#{database_username} -p#{database_password} -P#{database_port} -h#{database_host} #{database} < #{Rails.root}/db/structure.sql"
 
     puts "init success"
   end