Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.cdap.cdap.internal.app.workflow.DefaultWorkflowConditionConfigurer Maven / Gradle / Ivy
/*
* Copyright © 2015 Cask Data, Inc.
*
* Licensed 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 io.cdap.cdap.internal.app.workflow;
import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import io.cdap.cdap.api.Predicate;
import io.cdap.cdap.api.customaction.CustomAction;
import io.cdap.cdap.api.feature.FeatureFlagsProvider;
import io.cdap.cdap.api.schedule.SchedulableProgramType;
import io.cdap.cdap.api.workflow.Condition;
import io.cdap.cdap.api.workflow.ConditionSpecification;
import io.cdap.cdap.api.workflow.WorkflowConditionConfigurer;
import io.cdap.cdap.api.workflow.WorkflowConditionNode;
import io.cdap.cdap.api.workflow.WorkflowContext;
import io.cdap.cdap.api.workflow.WorkflowForkConfigurer;
import io.cdap.cdap.api.workflow.WorkflowForkNode;
import io.cdap.cdap.api.workflow.WorkflowNode;
import io.cdap.cdap.common.id.Id;
import io.cdap.cdap.internal.app.deploy.pipeline.AppDeploymentRuntimeInfo;
import io.cdap.cdap.internal.app.runtime.artifact.PluginFinder;
import io.cdap.cdap.internal.app.runtime.plugin.PluginInstantiator;
import io.cdap.cdap.internal.app.workflow.condition.DefaultConditionConfigurer;
import io.cdap.cdap.internal.workflow.condition.DefaultConditionSpecification;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import javax.annotation.Nullable;
/**
* Default implementation of the {@link WorkflowConditionConfigurer}.
* @param the type of the parent configurer
*/
public class DefaultWorkflowConditionConfigurer
implements WorkflowConditionConfigurer, WorkflowConditionAdder, WorkflowForkJoiner {
private final Predicate predicate;
private final Condition condition;
private final T parentConfigurer;
private final List ifBranch = Lists.newArrayList();
private final List elseBranch = Lists.newArrayList();
private final Id.Namespace deployNamespace;
private final Id.Artifact artifactId;
private final PluginFinder pluginFinder;
private final PluginInstantiator pluginInstantiator;
private final AppDeploymentRuntimeInfo runtimeInfo;
private final FeatureFlagsProvider featureFlagsProvider;
private List currentBranch;
private boolean addingToIfBranch = true;
public DefaultWorkflowConditionConfigurer(Predicate predicate, T parentConfigurer,
Id.Namespace deployNamespace, Id.Artifact artifactId,
PluginFinder pluginFinder,
PluginInstantiator pluginInstantiator,
@Nullable AppDeploymentRuntimeInfo runtimeInfo,
FeatureFlagsProvider featureFlagsProvider) {
this(predicate, null, parentConfigurer, deployNamespace, artifactId, pluginFinder, pluginInstantiator,
runtimeInfo, featureFlagsProvider);
}
public DefaultWorkflowConditionConfigurer(Condition condition, T parentConfigurer, Id.Namespace deployNamespace,
Id.Artifact artifactId, PluginFinder pluginFinder,
PluginInstantiator pluginInstantiator,
@Nullable AppDeploymentRuntimeInfo runtimeInfo,
FeatureFlagsProvider featureFlagsProvider) {
this(null, condition, parentConfigurer, deployNamespace, artifactId, pluginFinder, pluginInstantiator,
runtimeInfo, featureFlagsProvider);
}
private DefaultWorkflowConditionConfigurer(@Nullable Predicate predicate,
@Nullable Condition condition, T parentConfigurer,
Id.Namespace deployNamespace, Id.Artifact artifactId,
PluginFinder pluginFinder,
PluginInstantiator pluginInstantiator,
@Nullable AppDeploymentRuntimeInfo runtimeInfo,
FeatureFlagsProvider featureFlagsProvider) {
this.condition = condition;
this.predicate = predicate;
this.parentConfigurer = parentConfigurer;
this.deployNamespace = deployNamespace;
this.artifactId = artifactId;
this.pluginFinder = pluginFinder;
this.pluginInstantiator = pluginInstantiator;
this.currentBranch = Lists.newArrayList();
this.runtimeInfo = runtimeInfo;
this.featureFlagsProvider = featureFlagsProvider;
}
@Override
public WorkflowConditionConfigurer addMapReduce(String mapReduce) {
currentBranch.add(WorkflowNodeCreator.createWorkflowActionNode(mapReduce, SchedulableProgramType.MAPREDUCE));
return this;
}
@Override
public WorkflowConditionConfigurer addSpark(String spark) {
currentBranch.add(WorkflowNodeCreator.createWorkflowActionNode(spark, SchedulableProgramType.SPARK));
return this;
}
@Override
public WorkflowConditionConfigurer addAction(CustomAction action) {
currentBranch.add(WorkflowNodeCreator.createWorkflowCustomActionNode(action, deployNamespace, artifactId,
pluginFinder, pluginInstantiator,
runtimeInfo, featureFlagsProvider));
return this;
}
@Override
public WorkflowForkConfigurer> fork() {
return new DefaultWorkflowForkConfigurer<>(this, deployNamespace, artifactId, pluginFinder,
pluginInstantiator, runtimeInfo, featureFlagsProvider);
}
@Override
@SuppressWarnings("unchecked")
public WorkflowConditionConfigurer> condition(
Predicate predicate) {
return new DefaultWorkflowConditionConfigurer<>(predicate, this, deployNamespace, artifactId, pluginFinder,
pluginInstantiator, runtimeInfo, featureFlagsProvider);
}
@Override
@SuppressWarnings("unchecked")
public WorkflowConditionConfigurer> condition(Condition condition) {
return new DefaultWorkflowConditionConfigurer<>(condition, this, deployNamespace, artifactId, pluginFinder,
pluginInstantiator, runtimeInfo, featureFlagsProvider);
}
@Override
public WorkflowConditionConfigurer otherwise() {
ifBranch.addAll(currentBranch);
addingToIfBranch = false;
currentBranch = Lists.newArrayList();
return this;
}
@Override
public T end() {
if (addingToIfBranch) {
ifBranch.addAll(currentBranch);
} else {
elseBranch.addAll(currentBranch);
}
if (predicate != null) {
parentConfigurer.addWorkflowConditionNode(predicate, ifBranch, elseBranch);
} else {
parentConfigurer.addWorkflowConditionNode(condition, ifBranch, elseBranch);
}
return parentConfigurer;
}
@Override
public void addWorkflowConditionNode(Predicate predicate, List ifBranch,
List elseBranch) {
ConditionSpecification spec = new DefaultConditionSpecification(predicate.getClass().getName(),
predicate.getClass().getSimpleName(), "",
new HashMap(),
new HashSet());
currentBranch.add(new WorkflowConditionNode(spec.getName(), spec, ifBranch, elseBranch));
}
@Override
public void addWorkflowConditionNode(Condition condition, List ifBranch,
List elseBranch) {
Preconditions.checkArgument(condition != null, "Condition is null.");
ConditionSpecification spec = DefaultConditionConfigurer.configureCondition(condition, deployNamespace,
artifactId, pluginFinder,
pluginInstantiator, runtimeInfo,
featureFlagsProvider);
currentBranch.add(new WorkflowConditionNode(spec.getName(), spec, ifBranch, elseBranch));
}
@Override
public void addWorkflowForkNode(List> branches) {
currentBranch.add(new WorkflowForkNode(null, branches));
}
}
