|
@@ -14,12 +14,12 @@
|
|
|
* See the License for the specific language governing permissions and
|
|
|
* limitations under the License.
|
|
|
*/
|
|
|
+
|
|
|
package org.apache.dolphinscheduler.common.task.spark;
|
|
|
|
|
|
import org.apache.dolphinscheduler.common.enums.ProgramType;
|
|
|
import org.apache.dolphinscheduler.common.process.ResourceInfo;
|
|
|
import org.apache.dolphinscheduler.common.task.AbstractParameters;
|
|
|
-import org.apache.dolphinscheduler.common.utils.CollectionUtils;
|
|
|
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.List;
|
|
@@ -29,203 +29,214 @@ import java.util.List;
|
|
|
*/
|
|
|
public class SparkParameters extends AbstractParameters {
|
|
|
|
|
|
- /**
|
|
|
- * major jar
|
|
|
- */
|
|
|
- private ResourceInfo mainJar;
|
|
|
-
|
|
|
- /**
|
|
|
- * major class
|
|
|
- */
|
|
|
- private String mainClass;
|
|
|
-
|
|
|
- /**
|
|
|
- * deploy mode
|
|
|
- */
|
|
|
- private String deployMode;
|
|
|
-
|
|
|
- /**
|
|
|
- * arguments
|
|
|
- */
|
|
|
- private String mainArgs;
|
|
|
-
|
|
|
- /**
|
|
|
- * driver-cores Number of cores used by the driver, only in cluster mode
|
|
|
- */
|
|
|
- private int driverCores;
|
|
|
-
|
|
|
- /**
|
|
|
- * driver-memory Memory for driver
|
|
|
- */
|
|
|
-
|
|
|
- private String driverMemory;
|
|
|
-
|
|
|
- /**
|
|
|
- * num-executors Number of executors to launch
|
|
|
- */
|
|
|
- private int numExecutors;
|
|
|
-
|
|
|
- /**
|
|
|
- * executor-cores Number of cores per executor
|
|
|
- */
|
|
|
- private int executorCores;
|
|
|
-
|
|
|
- /**
|
|
|
- * Memory per executor
|
|
|
- */
|
|
|
- private String executorMemory;
|
|
|
-
|
|
|
- /**
|
|
|
- * resource list
|
|
|
- */
|
|
|
- private List<ResourceInfo> resourceList = new ArrayList<>();
|
|
|
-
|
|
|
- /**
|
|
|
- * The YARN queue to submit to
|
|
|
- */
|
|
|
- private String queue;
|
|
|
-
|
|
|
- /**
|
|
|
- * other arguments
|
|
|
- */
|
|
|
- private String others;
|
|
|
-
|
|
|
- /**
|
|
|
- * program type
|
|
|
- * 0 JAVA,1 SCALA,2 PYTHON
|
|
|
- */
|
|
|
- private ProgramType programType;
|
|
|
-
|
|
|
- /**
|
|
|
- * spark version
|
|
|
- */
|
|
|
- private String sparkVersion;
|
|
|
-
|
|
|
- public ResourceInfo getMainJar() {
|
|
|
- return mainJar;
|
|
|
- }
|
|
|
-
|
|
|
- public void setMainJar(ResourceInfo mainJar) {
|
|
|
- this.mainJar = mainJar;
|
|
|
- }
|
|
|
-
|
|
|
- public String getMainClass() {
|
|
|
- return mainClass;
|
|
|
- }
|
|
|
-
|
|
|
- public void setMainClass(String mainClass) {
|
|
|
- this.mainClass = mainClass;
|
|
|
- }
|
|
|
-
|
|
|
- public String getDeployMode() {
|
|
|
- return deployMode;
|
|
|
- }
|
|
|
-
|
|
|
- public void setDeployMode(String deployMode) {
|
|
|
- this.deployMode = deployMode;
|
|
|
- }
|
|
|
-
|
|
|
- public String getMainArgs() {
|
|
|
- return mainArgs;
|
|
|
- }
|
|
|
-
|
|
|
- public void setMainArgs(String mainArgs) {
|
|
|
- this.mainArgs = mainArgs;
|
|
|
- }
|
|
|
-
|
|
|
- public int getDriverCores() {
|
|
|
- return driverCores;
|
|
|
- }
|
|
|
-
|
|
|
- public void setDriverCores(int driverCores) {
|
|
|
- this.driverCores = driverCores;
|
|
|
- }
|
|
|
-
|
|
|
- public String getDriverMemory() {
|
|
|
- return driverMemory;
|
|
|
- }
|
|
|
-
|
|
|
- public void setDriverMemory(String driverMemory) {
|
|
|
- this.driverMemory = driverMemory;
|
|
|
- }
|
|
|
+ /**
|
|
|
+ * main jar
|
|
|
+ */
|
|
|
+ private ResourceInfo mainJar;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * main class
|
|
|
+ */
|
|
|
+ private String mainClass;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * deploy mode
|
|
|
+ */
|
|
|
+ private String deployMode;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * arguments
|
|
|
+ */
|
|
|
+ private String mainArgs;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * driver-cores Number of cores used by the driver, only in cluster mode
|
|
|
+ */
|
|
|
+ private int driverCores;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * driver-memory Memory for driver
|
|
|
+ */
|
|
|
+
|
|
|
+ private String driverMemory;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * num-executors Number of executors to launch
|
|
|
+ */
|
|
|
+ private int numExecutors;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * executor-cores Number of cores per executor
|
|
|
+ */
|
|
|
+ private int executorCores;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Memory per executor
|
|
|
+ */
|
|
|
+ private String executorMemory;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * app name
|
|
|
+ */
|
|
|
+ private String appName;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The YARN queue to submit to
|
|
|
+ */
|
|
|
+ private String queue;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * other arguments
|
|
|
+ */
|
|
|
+ private String others;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * program type
|
|
|
+ * 0 JAVA,1 SCALA,2 PYTHON
|
|
|
+ */
|
|
|
+ private ProgramType programType;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * spark version
|
|
|
+ */
|
|
|
+ private String sparkVersion;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * resource list
|
|
|
+ */
|
|
|
+ private List<ResourceInfo> resourceList = new ArrayList<>();
|
|
|
+
|
|
|
+ public ResourceInfo getMainJar() {
|
|
|
+ return mainJar;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setMainJar(ResourceInfo mainJar) {
|
|
|
+ this.mainJar = mainJar;
|
|
|
+ }
|
|
|
+
|
|
|
+ public String getMainClass() {
|
|
|
+ return mainClass;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setMainClass(String mainClass) {
|
|
|
+ this.mainClass = mainClass;
|
|
|
+ }
|
|
|
+
|
|
|
+ public String getDeployMode() {
|
|
|
+ return deployMode;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setDeployMode(String deployMode) {
|
|
|
+ this.deployMode = deployMode;
|
|
|
+ }
|
|
|
+
|
|
|
+ public String getMainArgs() {
|
|
|
+ return mainArgs;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setMainArgs(String mainArgs) {
|
|
|
+ this.mainArgs = mainArgs;
|
|
|
+ }
|
|
|
+
|
|
|
+ public int getDriverCores() {
|
|
|
+ return driverCores;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setDriverCores(int driverCores) {
|
|
|
+ this.driverCores = driverCores;
|
|
|
+ }
|
|
|
+
|
|
|
+ public String getDriverMemory() {
|
|
|
+ return driverMemory;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setDriverMemory(String driverMemory) {
|
|
|
+ this.driverMemory = driverMemory;
|
|
|
+ }
|
|
|
|
|
|
- public int getNumExecutors() {
|
|
|
- return numExecutors;
|
|
|
- }
|
|
|
+ public int getNumExecutors() {
|
|
|
+ return numExecutors;
|
|
|
+ }
|
|
|
|
|
|
- public void setNumExecutors(int numExecutors) {
|
|
|
- this.numExecutors = numExecutors;
|
|
|
- }
|
|
|
+ public void setNumExecutors(int numExecutors) {
|
|
|
+ this.numExecutors = numExecutors;
|
|
|
+ }
|
|
|
|
|
|
- public int getExecutorCores() {
|
|
|
- return executorCores;
|
|
|
- }
|
|
|
-
|
|
|
- public void setExecutorCores(int executorCores) {
|
|
|
- this.executorCores = executorCores;
|
|
|
- }
|
|
|
+ public int getExecutorCores() {
|
|
|
+ return executorCores;
|
|
|
+ }
|
|
|
|
|
|
- public String getExecutorMemory() {
|
|
|
- return executorMemory;
|
|
|
- }
|
|
|
+ public void setExecutorCores(int executorCores) {
|
|
|
+ this.executorCores = executorCores;
|
|
|
+ }
|
|
|
|
|
|
- public void setExecutorMemory(String executorMemory) {
|
|
|
- this.executorMemory = executorMemory;
|
|
|
- }
|
|
|
+ public String getExecutorMemory() {
|
|
|
+ return executorMemory;
|
|
|
+ }
|
|
|
|
|
|
+ public void setExecutorMemory(String executorMemory) {
|
|
|
+ this.executorMemory = executorMemory;
|
|
|
+ }
|
|
|
|
|
|
- public String getQueue() {
|
|
|
- return queue;
|
|
|
- }
|
|
|
+ public String getAppName() {
|
|
|
+ return appName;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void setAppName(String appName) {
|
|
|
+ this.appName = appName;
|
|
|
+ }
|
|
|
|
|
|
- public void setQueue(String queue) {
|
|
|
- this.queue = queue;
|
|
|
- }
|
|
|
+ public String getQueue() {
|
|
|
+ return queue;
|
|
|
+ }
|
|
|
|
|
|
- public List<ResourceInfo> getResourceList() {
|
|
|
- return resourceList;
|
|
|
- }
|
|
|
+ public void setQueue(String queue) {
|
|
|
+ this.queue = queue;
|
|
|
+ }
|
|
|
|
|
|
- public void setResourceList(List<ResourceInfo> resourceList) {
|
|
|
- this.resourceList = resourceList;
|
|
|
- }
|
|
|
+ public String getOthers() {
|
|
|
+ return others;
|
|
|
+ }
|
|
|
|
|
|
- public String getOthers() {
|
|
|
- return others;
|
|
|
- }
|
|
|
+ public void setOthers(String others) {
|
|
|
+ this.others = others;
|
|
|
+ }
|
|
|
|
|
|
- public void setOthers(String others) {
|
|
|
- this.others = others;
|
|
|
- }
|
|
|
+ public List<ResourceInfo> getResourceList() {
|
|
|
+ return resourceList;
|
|
|
+ }
|
|
|
|
|
|
- public ProgramType getProgramType() {
|
|
|
- return programType;
|
|
|
- }
|
|
|
+ public void setResourceList(List<ResourceInfo> resourceList) {
|
|
|
+ this.resourceList = resourceList;
|
|
|
+ }
|
|
|
|
|
|
- public void setProgramType(ProgramType programType) {
|
|
|
- this.programType = programType;
|
|
|
- }
|
|
|
+ public ProgramType getProgramType() {
|
|
|
+ return programType;
|
|
|
+ }
|
|
|
|
|
|
- public String getSparkVersion() {
|
|
|
- return sparkVersion;
|
|
|
- }
|
|
|
+ public void setProgramType(ProgramType programType) {
|
|
|
+ this.programType = programType;
|
|
|
+ }
|
|
|
|
|
|
- public void setSparkVersion(String sparkVersion) {
|
|
|
- this.sparkVersion = sparkVersion;
|
|
|
- }
|
|
|
+ public String getSparkVersion() {
|
|
|
+ return sparkVersion;
|
|
|
+ }
|
|
|
|
|
|
- @Override
|
|
|
- public boolean checkParameters() {
|
|
|
- return mainJar != null && programType != null;
|
|
|
- }
|
|
|
+ public void setSparkVersion(String sparkVersion) {
|
|
|
+ this.sparkVersion = sparkVersion;
|
|
|
+ }
|
|
|
|
|
|
- @Override
|
|
|
- public List<ResourceInfo> getResourceFilesList() {
|
|
|
- if (mainJar != null && !resourceList.contains(mainJar)) {
|
|
|
- resourceList.add(mainJar);
|
|
|
+ @Override
|
|
|
+ public boolean checkParameters() {
|
|
|
+ return mainJar != null && programType != null;
|
|
|
}
|
|
|
- return resourceList;
|
|
|
- }
|
|
|
|
|
|
+ @Override
|
|
|
+ public List<ResourceInfo> getResourceFilesList() {
|
|
|
+ if (mainJar != null && !resourceList.contains(mainJar)) {
|
|
|
+ resourceList.add(mainJar);
|
|
|
+ }
|
|
|
+ return resourceList;
|
|
|
+ }
|
|
|
|
|
|
}
|