Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,10 @@

import java.io.Serializable;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
Expand All @@ -28,6 +31,9 @@
* k8s Task ExecutionContext
*/
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
public class K8sTaskExecutionContext implements Serializable {

private String configYaml;
Expand All @@ -36,23 +42,11 @@ public class K8sTaskExecutionContext implements Serializable {

private String connectionParams;

public K8sTaskExecutionContext() {
}

@JsonCreator(mode = JsonCreator.Mode.PROPERTIES)
public K8sTaskExecutionContext(
@JsonProperty("configYaml") String configYaml,
@JsonProperty("namespace") String namespace) {
this.configYaml = configYaml;
this.namespace = namespace;
}

@Override
public String toString() {
return "K8sTaskExecutionContext{"
+ "namespace=" + namespace
+ ", configYaml='" + configYaml + '\''
+ ", connectionParams='" + connectionParams + '\''
+ '}';
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,9 @@
package org.apache.dolphinscheduler.plugin.task.api.parameters;

import org.apache.dolphinscheduler.common.utils.JSONUtils;
import org.apache.dolphinscheduler.plugin.task.api.K8sTaskExecutionContext;
import org.apache.dolphinscheduler.plugin.task.api.enums.Direct;
import org.apache.dolphinscheduler.plugin.task.api.enums.ResourceType;
import org.apache.dolphinscheduler.plugin.task.api.model.Property;
import org.apache.dolphinscheduler.plugin.task.api.model.ResourceInfo;
import org.apache.dolphinscheduler.plugin.task.api.parameters.resource.DataSourceParameters;
import org.apache.dolphinscheduler.plugin.task.api.parameters.resource.ResourceParametersHelper;
import org.apache.dolphinscheduler.plugin.task.api.utils.VarPoolUtils;

Expand Down Expand Up @@ -72,16 +69,6 @@ public Map<String, Property> getLocalParametersMap() {
return localParametersMaps;
}

public K8sTaskExecutionContext generateK8sTaskExecutionContext(ResourceParametersHelper parametersHelper,
int datasource) {
DataSourceParameters dataSourceParameters =
(DataSourceParameters) parametersHelper.getResourceParameters(ResourceType.DATASOURCE, datasource);
K8sTaskExecutionContext k8sTaskExecutionContext = new K8sTaskExecutionContext();
k8sTaskExecutionContext.setConnectionParams(
Objects.nonNull(dataSourceParameters) ? dataSourceParameters.getConnectionParams() : null);
return k8sTaskExecutionContext;
}

/**
* get input local parameters map if the param direct is IN
*
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,12 @@

package org.apache.dolphinscheduler.plugin.task.api.parameters;

import org.apache.dolphinscheduler.plugin.task.api.K8sTaskExecutionContext;
import org.apache.dolphinscheduler.plugin.task.api.enums.ResourceType;
import org.apache.dolphinscheduler.plugin.task.api.model.Label;
import org.apache.dolphinscheduler.plugin.task.api.model.NodeSelectorExpression;
import org.apache.dolphinscheduler.plugin.task.api.model.ResourceInfo;
import org.apache.dolphinscheduler.plugin.task.api.parameters.resource.DataSourceParameters;
import org.apache.dolphinscheduler.plugin.task.api.parameters.resource.ResourceParametersHelper;

import org.apache.commons.lang3.StringUtils;
Expand All @@ -31,9 +33,6 @@
import lombok.Data;
import lombok.extern.slf4j.Slf4j;

/**
* k8s task parameters
*/
@Data
@Slf4j
public class K8sTaskParameters extends AbstractParameters {
Expand Down Expand Up @@ -67,4 +66,21 @@ public ResourceParametersHelper getResources() {
resources.put(ResourceType.DATASOURCE, datasource);
return resources;
}

public K8sTaskExecutionContext generateK8sTaskExecutionContext(
ResourceParametersHelper parametersHelper,
int datasource) {
DataSourceParameters dataSourceParameters =
(DataSourceParameters) parametersHelper
.getResourceParameters(ResourceType.DATASOURCE, datasource);

String connectionParams = null;
if (dataSourceParameters != null) {
connectionParams = dataSourceParameters.getConnectionParams();
}

return K8sTaskExecutionContext.builder()
.connectionParams(connectionParams)
.build();
}
}
Loading