main class and main jar file changes

jinnthehuman · Aug 12, 2024 · a2feabb · a2feabb
1 parent db78ecf
commit a2feabb
Showing 1 changed file with 44 additions and 20 deletions.
diff --git a/src/jobs/submitJob.tsx b/src/jobs/submitJob.tsx
@@ -495,26 +495,50 @@ function SubmitJob({
     jarFileSelected: string[] | string,
     argumentSelected: string[] | string
   ) => {
-    return {
-      sparkJob: {
-        mainJarFileUri: mainClassSelected,
-        ...(propertyObject && {
-          properties: propertyObject
-        }),
-        ...(archieveFileSelected !== '' && {
-          archiveUris: archieveFileSelected
-        }),
-        ...(fileSelected !== '' && {
-          fileUris: [fileSelected]
-        }),
-        ...(jarFileSelected !== '' && {
-          jarFileUris: jarFileSelected
-        }),
-        ...(argumentSelected !== '' && {
-          args: argumentSelected
-        })
-      }
-    };
+    if(mainClassSelected.includes('.jar')){
+      return {
+        sparkJob: {
+          mainJarFileUri: mainClassSelected,
+          ...(propertyObject && {
+            properties: propertyObject
+          }),
+          ...(archieveFileSelected !== '' && {
+            archiveUris: archieveFileSelected
+          }),
+          ...(fileSelected !== '' && {
+            fileUris: [fileSelected]
+          }),
+          ...(jarFileSelected !== '' && {
+            jarFileUris: jarFileSelected
+          }),
+          ...(argumentSelected !== '' && {
+            args: argumentSelected
+          })
+        }
+      };
+    }
+    else{
+      return {
+        sparkJob: {
+          mainClass: mainClassSelected,
+          ...(propertyObject && {
+            properties: propertyObject
+          }),
+          ...(archieveFileSelected !== '' && {
+            archiveUris: archieveFileSelected
+          }),
+          ...(fileSelected !== '' && {
+            fileUris: [fileSelected]
+          }),
+          ...(jarFileSelected !== '' && {
+            jarFileUris: jarFileSelected
+          }),
+          ...(argumentSelected !== '' && {
+            args: argumentSelected
+          })
+        }
+      };
+    }
   };
 
   const createSparkRPayload = (