Add charset option

pom.xml

@@ -111,7 +111,7 @@
             <plugin>
                 <groupId>org.codehaus.mojo</groupId>
                 <artifactId>findbugs-maven-plugin</artifactId>
-                <version>3.0.3</version>
+                <version>3.0.4</version>
                 <configuration>
                     <skip>${air.check.skip-findbugs}</skip>
                     <jvmArgs>-Xmx${air.build.jvmsize}</jvmArgs>

src/main/java/com/teradata/tpcds/Options.java

@@ -30,6 +30,7 @@ public class Options
     public static final boolean DEFAULT_NO_SEXISM = false;
     public static final int DEFAULT_PARALLELISM = 1;
     public static final boolean DEFAULT_OVERWRITE = false;
+    public static final String DEFAULT_CHARSET = "ISO_8859_1";
 
     @Option(name = {"--scale", "-s"}, title = "scale", description = "Volume of data to generate in GB (Default: 1)")
     public double scale = DEFAULT_SCALE;
@@ -43,6 +44,9 @@ public class Options
     @Option(name = {"--table", "-t"}, title = "table", description = "Build only the specified table.  If this property is not specified, all tables will be generated")
     public String table = DEFAULT_TABLE;
 
+    @Option(name = {"--charset"}, title = "charset", description = "Charset used for output files. (Default: ISO_8859_1)")
+    public String charset = DEFAULT_CHARSET;
+
     @Option(name = {"--null"}, title = "null", description = "String representation for null values (Default: the empty string)")
     public String nullString = DEFAULT_NULL_STRING;
 
@@ -77,7 +81,8 @@ public Session toSession()
                 doNotTerminate,
                 noSexism,
                 parallelism,
-                overwrite);
+                overwrite,
+                charset);
     }
 
     private static Optional<Table> toTableOptional(String table)

src/main/java/com/teradata/tpcds/Session.java

@@ -16,6 +16,7 @@
 
 import java.util.Optional;
 
+import static com.teradata.tpcds.Options.DEFAULT_CHARSET;
 import static com.teradata.tpcds.Options.DEFAULT_DIRECTORY;
 import static com.teradata.tpcds.Options.DEFAULT_DO_NOT_TERMINATE;
 import static com.teradata.tpcds.Options.DEFAULT_NO_SEXISM;
@@ -39,13 +40,14 @@ public class Session
     private final int parallelism;
     private final int chunkNumber;
     private final boolean overwrite;
+    private final String charset;
 
-    public Session(double scale, String targetDirectory, String suffix, Optional<Table> table, String nullString, char separator, boolean doNotTerminate, boolean noSexism, int parallelism, boolean overwrite)
+    public Session(double scale, String targetDirectory, String suffix, Optional<Table> table, String nullString, char separator, boolean doNotTerminate, boolean noSexism, int parallelism, boolean overwrite, String charset)
     {
-        this(scale, targetDirectory, suffix, table, nullString, separator, doNotTerminate, noSexism, parallelism, 1, overwrite);
+        this(scale, targetDirectory, suffix, table, nullString, separator, doNotTerminate, noSexism, parallelism, 1, overwrite, charset);
     }
 
-    public Session(double scale, String targetDirectory, String suffix, Optional<Table> table, String nullString, char separator, boolean doNotTerminate, boolean noSexism, int parallelism, int chunkNumber, boolean overwrite)
+    public Session(double scale, String targetDirectory, String suffix, Optional<Table> table, String nullString, char separator, boolean doNotTerminate, boolean noSexism, int parallelism, int chunkNumber, boolean overwrite, String charset)
     {
         this.scaling = new Scaling(scale);
         this.targetDirectory = targetDirectory;
@@ -58,6 +60,7 @@ public Session(double scale, String targetDirectory, String suffix, Optional<Tab
         this.parallelism = parallelism;
         this.chunkNumber = chunkNumber;
         this.overwrite = overwrite;
+        this.charset = charset;
     }
 
     public static Session getDefaultSession()
@@ -78,7 +81,8 @@ public Session withTable(Table table)
                 this.noSexism,
                 this.parallelism,
                 this.chunkNumber,
-                this.overwrite
+                this.overwrite,
+                this.charset
         );
     }
 
@@ -95,7 +99,8 @@ public Session withScale(double scale)
                 this.noSexism,
                 this.parallelism,
                 this.chunkNumber,
-                this.overwrite
+                this.overwrite,
+                this.charset
         );
     }
 
@@ -112,7 +117,8 @@ public Session withParallelism(int parallelism)
                 this.noSexism,
                 parallelism,
                 this.chunkNumber,
-                this.overwrite
+                this.overwrite,
+                this.charset
         );
     }
 
@@ -129,7 +135,8 @@ public Session withChunkNumber(int chunkNumber)
                 this.noSexism,
                 this.parallelism,
                 chunkNumber,
-                this.overwrite
+                this.overwrite,
+                this.charset
         );
     }
 
@@ -146,7 +153,8 @@ public Session withNoSexism(boolean noSexism)
                 noSexism,
                 this.parallelism,
                 this.chunkNumber,
-                this.overwrite
+                this.overwrite,
+                this.charset
         );
     }
 
@@ -208,6 +216,11 @@ public int getChunkNumber()
         return chunkNumber;
     }
 
+    public String getCharset()
+    {
+        return charset;
+    }
+
     public boolean shouldOverwrite()
     {
         return overwrite;
@@ -246,6 +259,9 @@ public String getCommandLineArguments()
         if (overwrite != DEFAULT_OVERWRITE) {
             output.append("--overwrite ");
         }
+        if (!charset.equals(DEFAULT_CHARSET)) {
+            output.append("--charset ").append(charset).append(" ");
+        }
 
         // remove trailing space
         if (output.length() > 0) {

src/main/java/com/teradata/tpcds/TableGenerator.java

@@ -19,7 +19,7 @@
 import java.io.IOException;
 import java.io.OutputStreamWriter;
 import java.io.Writer;
-import java.nio.charset.StandardCharsets;
+import java.nio.charset.Charset;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -77,7 +77,7 @@ private OutputStreamWriter addFileWriterForTable(Table table)
             }
         }
 
-        return new OutputStreamWriter(new FileOutputStream(path, true), StandardCharsets.ISO_8859_1);
+        return new OutputStreamWriter(new FileOutputStream(path, true), Charset.forName(session.getCharset()));
     }
 
     private String getPath(Table table)