InplaceFlowBuilder.java

package com.zillabyte.motherbrain.flow.operations.multilang.builder;

import net.sf.json.JSONObject;

import org.apache.log4j.Logger;

import com.zillabyte.motherbrain.container.ContainerWrapper;
import com.zillabyte.motherbrain.container.local.InplaceContainer;
import com.zillabyte.motherbrain.flow.Flow;
import com.zillabyte.motherbrain.flow.FlowCompilationException;
import com.zillabyte.motherbrain.flow.config.FlowConfig;
import com.zillabyte.motherbrain.flow.operations.OperationLogger;
import com.zillabyte.motherbrain.utils.Utils;

public class InplaceFlowBuilder  implements FlowFetcher {

  private static final long serialVersionUID = 5321655302235740392L;
  private static final Logger _log = Utils.getLogger(InplaceFlowBuilder.class);
  
  private MultilangFlowCompiler _flowCompiler;
  private OperationLogger _logger;
  private ContainerWrapper _container;
  private FlowConfig _flowConfig;
  
  
  /***
   * 
   * @param authToken
   */
  public InplaceFlowBuilder(FlowConfig flowConfig, ContainerWrapper destContainer, OperationLogger logger) {
    
    // Init 
    _flowCompiler = new MultilangFlowCompiler(this, flowConfig, destContainer, logger);
    _flowConfig = flowConfig;
    _logger = logger;
    _container = destContainer;
    
    // Sanity
    if (_container.getDelegate() instanceof InplaceContainer == false) {
      throw new IllegalStateException("InPlaceFlowBuilder will only work with InPlaceContainers!");
    }
  }
  
  
  
  
  /**
   * 
   * @param id
   * @param flowLogger
   * @param destDir
   * @throws InterruptedException
   * @throws LXCException
   * @throws FlowCompilationException 
   */
  @Override
  public Flow buildFlow(String flowName, JSONObject overrideConfig) throws FlowCompilationException {
    try {
      
      // Actually build the flow
      return _flowCompiler.compileFlow(flowName, overrideConfig);
        
    } catch(Exception e) {
      throw new FlowCompilationException(e);
    }
  }

  
  
  public Flow buildFlow(String flowName) throws FlowCompilationException {
    return buildFlow(flowName, new JSONObject());
  }

  
}