Create setting to declare if insertion file has a header

vieirandre · Jul 6, 2020 · 1fd9fb0 · 1fd9fb0
1 parent 7c7bb9e
commit 1fd9fb0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/Config.cs b/Config.cs
@@ -14,13 +14,14 @@ public static class Config
         public static int TargetPort => int.TryParse(Configuration.GetSection("TargetPort").Value, out int targetPort) ? targetPort : Cassandra.ProtocolOptions.DefaultPort;
         public static string TargetKeyspace => Configuration.GetSection("TargetKeyspace").Value;
         public static string TargetTable => Configuration.GetSection("TargetTable").Value;
-        public static string InsertionFilePath => Configuration.GetSection("InsertionFilePath").Value;
         public static string ExtractionFilePath => Configuration.GetSection("ExtractionFilePath").Value;
+        public static string InsertionFilePath => Configuration.GetSection("InsertionFilePath").Value;
+        public static bool InsertionFileHasHeader => string.Compare(Configuration.GetSection("InsertionFileHasHeader").Value, "true", true) == 0;
         public static int InsertionBatch => int.TryParse(Configuration.GetSection("InsertionBatch").Value, out int insertionBatch) ? insertionBatch : 100000;
 
         private static readonly IConfigurationRoot Configuration = new ConfigurationBuilder()
-                                                                        .SetBasePath(Directory.GetCurrentDirectory())
-                                                                        .AddJsonFile("appsettings.json", optional: false, reloadOnChange: true)
-                                                                        .Build();
+                                                                       .SetBasePath(Directory.GetCurrentDirectory())
+                                                                       .AddJsonFile("appsettings.json", optional: false, reloadOnChange: true)
+                                                                       .Build();
     }
 }
diff --git a/Models/Insertion.cs b/Models/Insertion.cs
@@ -106,7 +106,7 @@ private static dynamic[] PrepareRowForInsertion(IList<CColumn> columns, List<str
         private static void ConfigureCsvReader(CsvReader csvReader)
         {
             csvReader.Configuration.Delimiter = ",";
-            csvReader.Configuration.HasHeaderRecord = true;
+            csvReader.Configuration.HasHeaderRecord = Config.InsertionFileHasHeader;
             csvReader.Configuration.MissingFieldFound = null;
         }
     }

diff --git a/appsettings.json b/appsettings.json
@@ -10,5 +10,6 @@
   "TargetTable": "",
   "ExtractionFilePath": "data/extract.csv",
   "InsertionFilePath": "data/insert.csv",
+  "InsertionFileHasHeader": "",
   "InsertionBatch": "100000"
 }