@@ -21,14 +21,19 @@ defmodule Csv2sql.MainServer do
2121 end
2222
2323 def handle_info ( :kickoff , worker_count ) do
24- Csv2sql.Helpers . greet ( )
24+ set_insert_schema = Application . get_env ( :csv2sql , Csv2sql.Worker ) [ :set_insert_schema ]
25+ set_insert_data = Application . get_env ( :csv2sql , Csv2sql.Worker ) [ :set_insert_data ]
2526
26- Csv2sql.Database . prepare_db ( )
27+ if set_insert_schema || set_insert_data do
28+ Csv2sql.Database . prepare_db ( )
29+ end
2730
2831 schema_file_path = Application . get_env ( :csv2sql , Csv2sql.MainServer ) [ :db_worker_count ]
2932 db_worker_count = Application . get_env ( :csv2sql , Csv2sql.MainServer ) [ :db_worker_count ]
3033
31- File . rm ( "#{ schema_file_path } /schema.sql" )
34+ if Application . get_env ( :csv2sql , Csv2sql.Worker ) [ :set_make_schema ] do
35+ File . rm ( "#{ schema_file_path } /schema.sql" )
36+ end
3237
3338 1 .. worker_count
3439 |> Enum . each ( fn _ -> Csv2sql.WorkerSupervisor . add_worker ( ) end )
@@ -40,6 +45,8 @@ defmodule Csv2sql.MainServer do
4045 end
4146
4247 def handle_cast ( :done , 1 ) do
48+ set_validate = Application . get_env ( :csv2sql , Csv2sql.MainServer ) [ :set_validate ]
49+
4350 wait_for_pending_jobs ( )
4451
4552 :timer . sleep ( 2000 )
@@ -65,12 +72,16 @@ defmodule Csv2sql.MainServer do
6572 end
6673 )
6774
68- Csv2sql.Helpers . print_msg ( "\n Validation Process Started...\n \n " , :green )
75+ if ( set_validate ) do
76+ Csv2sql.Helpers . print_msg ( "\n Validation Process Started...\n \n " , :green )
6977
70- imported_csv_directory =
71- Application . get_env ( :csv2sql , Csv2sql.MainServer ) [ :imported_csv_directory ]
78+ imported_csv_directory =
79+ Application . get_env ( :csv2sql , Csv2sql.MainServer ) [ :imported_csv_directory ]
7280
73- Csv2sql.ImportValidator . validate_import ( imported_csv_directory )
81+ Csv2sql.ImportValidator . validate_import ( imported_csv_directory )
82+ else
83+ Csv2sql.Helpers . print_msg ( "\n Validation Process Skipped...\n \n " , :green )
84+ end
7485
7586 Csv2sql.TimerServer . get_time_spend ( )
7687
0 commit comments