From b404e4adf15c897321ef62b9e986af1ae7e946dc Mon Sep 17 00:00:00 2001
From: Xiang Fu <xiangfu.1024@gmail.com>
Date: Wed, 9 Feb 2022 23:01:04 -0800
Subject: [PATCH] Adding main method for LaunchDataIngestionJobCommand for
 spark-submit command (#8168)

---
 .../tools/admin/command/LaunchDataIngestionJobCommand.java  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pinot-tools/src/main/java/org/apache/pinot/tools/admin/command/LaunchDataIngestionJobCommand.java b/pinot-tools/src/main/java/org/apache/pinot/tools/admin/command/LaunchDataIngestionJobCommand.java
index 77991e5beda0..fc6f1c12028c 100644
--- a/pinot-tools/src/main/java/org/apache/pinot/tools/admin/command/LaunchDataIngestionJobCommand.java
+++ b/pinot-tools/src/main/java/org/apache/pinot/tools/admin/command/LaunchDataIngestionJobCommand.java
@@ -25,6 +25,7 @@
 import org.apache.pinot.spi.ingestion.batch.IngestionJobLauncher;
 import org.apache.pinot.spi.ingestion.batch.spec.SegmentGenerationJobSpec;
 import org.apache.pinot.spi.ingestion.batch.spec.TlsSpec;
+import org.apache.pinot.spi.plugin.PluginManager;
 import org.apache.pinot.spi.utils.GroovyTemplateUtils;
 import org.apache.pinot.tools.Command;
 import org.slf4j.Logger;
@@ -146,4 +147,9 @@ public String toString() {
   public String description() {
     return "Launch a data ingestion job.";
   }
+
+  public static void main(String[] args) {
+    PluginManager.get().init();
+    new CommandLine(new LaunchDataIngestionJobCommand()).execute(args);
+  }
 }