org.streampipes.manager.template.PipelineTemplateInvocationHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of streampipes-pipeline-management Show documentation
There is a newer version: 0.65.0
Show newest version
/*
 * Copyright 2018 FZI Forschungszentrum Informatik
 *
 * 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 org.streampipes.manager.template;

import org.streampipes.manager.operations.Operations;
import org.streampipes.model.base.InvocableStreamPipesEntity;
import org.streampipes.model.client.pipeline.Pipeline;
import org.streampipes.model.client.pipeline.PipelineOperationStatus;
import org.streampipes.model.staticproperty.StaticProperty;
import org.streampipes.model.template.PipelineTemplateDescription;
import org.streampipes.model.template.PipelineTemplateInvocation;
import org.streampipes.storage.management.StorageDispatcher;

import java.util.ArrayList;
import java.util.List;

public class PipelineTemplateInvocationHandler {

  private PipelineTemplateInvocation pipelineTemplateInvocation;
  private String username;

  public PipelineTemplateInvocationHandler(String username, PipelineTemplateInvocation pipelineTemplateInvocation) {
    this.username = username;
    this.pipelineTemplateInvocation = pipelineTemplateInvocation;
  }


  public PipelineOperationStatus handlePipelineInvocation() {
    Pipeline pipeline = new PipelineGenerator(pipelineTemplateInvocation.getDataSetId(), getTemplateById(pipelineTemplateInvocation.getPipelineTemplateId()), pipelineTemplateInvocation.getKviName()).makePipeline();
    pipeline.setCreatedByUser(username);
    pipeline.setCreatedAt(System.currentTimeMillis());
    replaceStaticProperties(pipeline);
    Operations.storePipeline(pipeline);
    Pipeline storedPipeline = StorageDispatcher.INSTANCE.getNoSqlStore().getPipelineStorageAPI().getPipeline(pipeline.getPipelineId());
    return Operations.startPipeline(storedPipeline);
  }

  private void replaceStaticProperties(Pipeline pipeline) {
    pipeline.getSepas().forEach(this::replace);
    pipeline.getActions().forEach(this::replace);
  }

  private void replace(InvocableStreamPipesEntity pe) {
    List newProperties = new ArrayList<>();
    pe.getStaticProperties().forEach(sp -> {
      if (existsInCustomizedElements(pe.getDOM(), sp)) {
        newProperties.add(getCustomizedElement(pe.getDOM(), pe.getDOM() + sp.getInternalName()));
      } else {
        newProperties.add(sp);
      }
    });
    pe.setStaticProperties(newProperties);
  }



  private StaticProperty getCustomizedElement(String dom, String internalName) {
    StaticProperty staticProperty = pipelineTemplateInvocation
            .getStaticProperties()
            .stream()
            .filter(sp -> sp.getInternalName().equals(internalName)).findFirst().get();

    staticProperty.setInternalName(staticProperty.getInternalName().replace(dom, ""));
    return staticProperty;
  }

  private boolean existsInCustomizedElements(String dom, StaticProperty staticProperty) {
    return pipelineTemplateInvocation
            .getStaticProperties()
            .stream()
            .anyMatch(sp -> sp.getInternalName().equals(dom +staticProperty.getInternalName()));
  }
  

  private PipelineTemplateDescription getTemplateById(String pipelineTemplateId) {
    return new PipelineTemplateGenerator().makeExampleTemplates().stream().filter(template -> template.getPipelineTemplateId().equals(pipelineTemplateId)).findFirst().get();
  }
}