mist.api.FnEntryPoint.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mist-lib_2.12 Show documentation

mist-lib

The newest version!

package mist.api

import mist.api.data.{JsData, JsMap, Json4sConversion}
import org.apache.spark.SparkContext
import org.slf4j.LoggerFactory

import scala.util._

trait FnEntryPoint {

  def execute(ctx: FnContext): Try[JsData]

  def main(args: Array[String]): Unit = {
    val logger = LoggerFactory.getLogger(this.getClass)
    val sc = SparkContext.getOrCreate()

    val inputData = args.headOption match {
      case Some(s) => Json4sConversion.parseRoot(s) match {
        case Success(js) => js
        case Failure(e) =>
          logger.error("Couldn't parse input data as json map", e)
          sys.exit(1)
      }
      case None => JsMap.empty
    }

    val fnContext = FnContext(sc, inputData)

    execute(fnContext) match {
      case Success(output) =>
        logger.info("Function completed successfully")
        val formatted = Json4sConversion.formattedString(output)
        logger.info("Output data:\n" + formatted)
      case Failure(err) =>
        logger.error("Function completed with error", err)
    }
  }
}