Assert
2 years ago
committed by
GitHub
9 changed files with 156 additions and 20 deletions
@ -0,0 +1,44 @@ |
|||||||
|
/* |
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||||
|
* contributor license agreements. See the NOTICE file distributed with |
||||||
|
* this work for additional information regarding copyright ownership. |
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||||
|
* (the "License"); you may not use this file except in compliance with |
||||||
|
* the License. You may obtain a copy of the License at |
||||||
|
* |
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* |
||||||
|
* Unless required by applicable law or agreed to in writing, software |
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
* See the License for the specific language governing permissions and |
||||||
|
* limitations under the License. |
||||||
|
*/ |
||||||
|
|
||||||
|
package org.apache.dolphinscheduler.plugin.task.seatunnel.self; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.plugin.task.seatunnel.DeployModeEnum; |
||||||
|
import org.apache.dolphinscheduler.plugin.task.seatunnel.SeatunnelParameters; |
||||||
|
|
||||||
|
public class SeatunnelEngineParameters extends SeatunnelParameters { |
||||||
|
|
||||||
|
private DeployModeEnum deployMode; |
||||||
|
|
||||||
|
private String others; |
||||||
|
|
||||||
|
public DeployModeEnum getDeployMode() { |
||||||
|
return deployMode; |
||||||
|
} |
||||||
|
|
||||||
|
public void setDeployMode(DeployModeEnum deployMode) { |
||||||
|
this.deployMode = deployMode; |
||||||
|
} |
||||||
|
|
||||||
|
public String getOthers() { |
||||||
|
return others; |
||||||
|
} |
||||||
|
|
||||||
|
public void setOthers(String others) { |
||||||
|
this.others = others; |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,58 @@ |
|||||||
|
/* |
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||||
|
* contributor license agreements. See the NOTICE file distributed with |
||||||
|
* this work for additional information regarding copyright ownership. |
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0 |
||||||
|
* (the "License"); you may not use this file except in compliance with |
||||||
|
* the License. You may obtain a copy of the License at |
||||||
|
* |
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* |
||||||
|
* Unless required by applicable law or agreed to in writing, software |
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
* See the License for the specific language governing permissions and |
||||||
|
* limitations under the License. |
||||||
|
*/ |
||||||
|
|
||||||
|
package org.apache.dolphinscheduler.plugin.task.seatunnel.self; |
||||||
|
|
||||||
|
import org.apache.dolphinscheduler.common.utils.JSONUtils; |
||||||
|
import org.apache.dolphinscheduler.plugin.task.api.TaskExecutionContext; |
||||||
|
import org.apache.dolphinscheduler.plugin.task.seatunnel.Constants; |
||||||
|
import org.apache.dolphinscheduler.plugin.task.seatunnel.SeatunnelTask; |
||||||
|
|
||||||
|
import org.apache.commons.lang3.StringUtils; |
||||||
|
|
||||||
|
import java.util.List; |
||||||
|
import java.util.Objects; |
||||||
|
|
||||||
|
public class SeatunnelEngineTask extends SeatunnelTask { |
||||||
|
|
||||||
|
private SeatunnelEngineParameters seatunnelParameters; |
||||||
|
public SeatunnelEngineTask(TaskExecutionContext taskExecutionContext) { |
||||||
|
super(taskExecutionContext); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void init() { |
||||||
|
seatunnelParameters = |
||||||
|
JSONUtils.parseObject(taskExecutionContext.getTaskParams(), SeatunnelEngineParameters.class); |
||||||
|
setSeatunnelParameters(seatunnelParameters); |
||||||
|
super.init(); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public List<String> buildOptions() throws Exception { |
||||||
|
List<String> args = super.buildOptions(); |
||||||
|
if (!Objects.isNull(seatunnelParameters.getDeployMode())) { |
||||||
|
args.add(Constants.DEPLOY_MODE_OPTIONS); |
||||||
|
args.add(seatunnelParameters.getDeployMode().getCommand()); |
||||||
|
} |
||||||
|
if (StringUtils.isNotBlank(seatunnelParameters.getOthers())) { |
||||||
|
args.add(seatunnelParameters.getOthers()); |
||||||
|
} |
||||||
|
return args; |
||||||
|
} |
||||||
|
|
||||||
|
} |
Loading…
Reference in new issue