From 8e788725d2f9bc68b4b822a8d6716725334dd5a1 Mon Sep 17 00:00:00 2001
From: rollno748 <rollno748@gmail.com>
Date: Thu, 4 Jan 2024 22:24:30 +0000
Subject: [PATCH] Added handling for empty line - Issue-16

---
 pom.xml                                                  | 2 +-
 .../com/di/jmeter/config/ExtendedCsvDataSetConfig.java   | 6 +++++-
 .../jmeter/config/gui/ExtendedCsvDataSetConfigGui.java   | 2 +-
 .../java/com/di/jmeter/utils/FileServerExtended.java     | 9 +++++----
 4 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/pom.xml b/pom.xml
index a5c042e..49521df 100644
--- a/pom.xml
+++ b/pom.xml
@@ -4,7 +4,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.di.jmeter.ecsv</groupId>
     <artifactId>di-extended-csv</artifactId>
-    <version>2.0</version>
+    <version>2.1</version>
 <!--Properties-->
     <properties>
         <jmeter-version>5.1.1</jmeter-version>
diff --git a/src/main/java/com/di/jmeter/config/ExtendedCsvDataSetConfig.java b/src/main/java/com/di/jmeter/config/ExtendedCsvDataSetConfig.java
index 295f2be..a450c15 100644
--- a/src/main/java/com/di/jmeter/config/ExtendedCsvDataSetConfig.java
+++ b/src/main/java/com/di/jmeter/config/ExtendedCsvDataSetConfig.java
@@ -105,6 +105,10 @@ public void iterationStart(LoopIterationEvent iterationEvent) {
                     for (int a = 0; a < variables.length && a < lineValues.length; a++) {
                         jMeterVariables.put(variables[a], lineValues[a]);
                     }
+                }else {
+                    for (String variable : variables) {
+                        jMeterVariables.put(variable, null);
+                    }
                 }
                 break;
             case "once":
@@ -329,4 +333,4 @@ public void setBlockSize(String blockSize) {
     public String printAllProperties() {
         return String.format("Filename: %s\n,FileEncoding: %s\n VariableName: %s\n IgnoreFirstLine: %s\n Delimiter: %s\n IsQuotedData: %s\n SelectRow: %s\n UpdateValue: %s\n OOValue: %s\n AutoAllocate: %s\n Allocate: %s\n BlockSize: %s\n",getFilename(),getFileEncoding(),getVariableNames(),isIgnoreFirstLine(),getDelimiter(),isQuotedData(),getSelectRow(),getUpdateValue(),getOoValue(),isAllocate(),isAutoAllocate(),getBlockSize());
     }
-}
\ No newline at end of file
+}
diff --git a/src/main/java/com/di/jmeter/config/gui/ExtendedCsvDataSetConfigGui.java b/src/main/java/com/di/jmeter/config/gui/ExtendedCsvDataSetConfigGui.java
index e8b59fd..eac0915 100644
--- a/src/main/java/com/di/jmeter/config/gui/ExtendedCsvDataSetConfigGui.java
+++ b/src/main/java/com/di/jmeter/config/gui/ExtendedCsvDataSetConfigGui.java
@@ -282,7 +282,7 @@ private void init() {
             try {
                 File file = new File(filenameField.getText());
                 Desktop desktop = Desktop.getDesktop();
-                if(filenameField.getText().equals("") || filenameField.getText().isEmpty()){
+                if(filenameField.getText().isEmpty() || filenameField.getText().isEmpty()){
                     throw new FileNotFoundException();
                 }
                 if(!file.exists()){
diff --git a/src/main/java/com/di/jmeter/utils/FileServerExtended.java b/src/main/java/com/di/jmeter/utils/FileServerExtended.java
index 673d45e..3f5ddc6 100644
--- a/src/main/java/com/di/jmeter/utils/FileServerExtended.java
+++ b/src/main/java/com/di/jmeter/utils/FileServerExtended.java
@@ -63,9 +63,9 @@ public class FileServerExtended {
 
     private File base;
     private static int rowCount;
-    private static ThreadLocal<Integer> endPos =  new ThreadLocal<>();
-    private static ThreadLocal<Integer> startPos =  new ThreadLocal<>();
-    private static ThreadLocal<Integer> readPos =  new ThreadLocal<>();
+    private static final ThreadLocal<Integer> endPos =  new ThreadLocal<>();
+    private static final ThreadLocal<Integer> startPos =  new ThreadLocal<>();
+    private static final ThreadLocal<Integer> readPos =  new ThreadLocal<>();
     private final Map<String, FileEntry> files = new HashMap<>();
     private static final FileServerExtended server = new FileServerExtended();
 
@@ -591,7 +591,8 @@ public void calculateRowCount(String filename, boolean ignoreFirstLine) {
         try (BufferedReader br = new BufferedReader(new FileReader(String.valueOf(fileEntry.file.toPath())))) {
             count = (int) br.lines().count();
         } catch (IOException e) {
-            e.printStackTrace();
+            log.error(e.toString());
+
         }
         this.setRowCount(ignoreFirstLine ? count-1 : count );
     }