package com.asakusafw.spark.compiler

import java.io.File

import scala.collection.mutable
import scala.collection.JavaConversions._

import com.asakusafw.lang.compiler.api.CompilerOptions
import com.asakusafw.lang.compiler.api.JobflowProcessor.{ Context => JPContext }
import com.asakusafw.lang.compiler.api.testing.MockJobflowProcessorContext
import com.asakusafw.spark.compiler.subplan.{
  BranchKeysClassBuilder,
  BroadcastIdsClassBuilder
}

trait CompilerContext {

  def newContext(
    flowId: String,
    outputDir: File): SparkClientCompiler.Context = {
    newContext(
      flowId,
      new MockJobflowProcessorContext(
        new CompilerOptions("buildid", "", Map.empty[String, String]),
        Thread.currentThread.getContextClassLoader,
        outputDir))
  }

  def newContext(
    flowId: String,
    jpContext: JPContext): SparkClientCompiler.Context = {
    SparkClientCompiler.Context(
      flowId = flowId,
      jpContext = jpContext,
      externalInputs = mutable.Map.empty,
      branchKeys = new BranchKeysClassBuilder(flowId),
      broadcastIds = new BroadcastIdsClassBuilder(flowId))
  }
}

/*
 * Copyright 2011-2015 Asakusa Framework Team.
 *
 * 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 com.asakusafw.spark.compiler

import java.io.File

import scala.collection.mutable
import scala.collection.JavaConversions._

import com.asakusafw.lang.compiler.api.CompilerOptions
import com.asakusafw.lang.compiler.api.JobflowProcessor.{ Context => JPContext }
import com.asakusafw.lang.compiler.api.testing.MockJobflowProcessorContext
import com.asakusafw.spark.compiler.subplan.{
  BranchKeysClassBuilder,
  BroadcastIdsClassBuilder
}

trait CompilerContext {

  def newContext(
    flowId: String,
    outputDir: File): SparkClientCompiler.Context = {
    newContext(
      flowId,
      new MockJobflowProcessorContext(
        new CompilerOptions("buildid", "", Map.empty[String, String]),
        Thread.currentThread.getContextClassLoader,
        outputDir))
  }

  def newContext(
    flowId: String,
    jpContext: JPContext): SparkClientCompiler.Context = {
    SparkClientCompiler.Context(
      flowId = flowId,
      jpContext = jpContext,
      externalInputs = mutable.Map.empty,
      branchKeys = new BranchKeysClassBuilder(flowId),
      broadcastIds = new BroadcastIdsClassBuilder(flowId))
  }
}

Add license header for a new file.
Add license header for a new file.
