Add option to specify flink parallelism

Author: physikerwelt

mathosphere-core/src/main/java/com/formulasearchengine/mathosphere/mlp/FlinkMlpRelationFinder.java

@@ -50,6 +50,10 @@ public static void run(FlinkMlpCommandConfig config) throws Exception {
         .writeAsText(config.getOutputDir(), WriteMode.OVERWRITE);
     //int cores = Runtime.getRuntime().availableProcessors();
     //env.setParallelism(1); // rounds down
+    final int parallelism = config.getParallelism();
+    if (parallelism > 0) {
+      env.setParallelism(parallelism);
+    }
     env.execute("Relation Finder");
   }
 

mathosphere-core/src/main/java/com/formulasearchengine/mathosphere/mlp/cli/FlinkMlpCommandConfig.java

@@ -15,6 +15,8 @@ public class FlinkMlpCommandConfig extends BaseConfig implements Serializable {
   @Parameter(names = {"-out", "--outputDir"}, description = "path to output directory")
   protected String outputdir;
 
+  @Parameter(names = {"--treads"}, description = "how many parallel threads should be used")
+  protected int parallelism = 0;
 
   public FlinkMlpCommandConfig() {
   }
@@ -60,4 +62,7 @@ public String getOutputDir() {
     return outputdir;
   }
 
+  public int getParallelism() {
+    return parallelism;
+  }
 }