shark.SharkServer2.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of shark_2.10 Show documentation
shark
The newest version!
package shark

import org.apache.commons.logging.LogFactory
import org.apache.hadoop.hive.common.LogUtils
import org.apache.hadoop.hive.common.LogUtils.LogInitializationException
import org.apache.hadoop.hive.conf.HiveConf
import org.apache.hive.service.cli.thrift.ThriftCLIService
import org.apache.hive.service.server.{HiveServer2, ServerOptionsProcessor}
import org.apache.spark.SparkEnv
import shark.server.SharkCLIService

object SharkServer2 extends LogHelper {
  SharkEnv.init()
  var sparkEnv: SparkEnv = SparkEnv.get
  var LOG = LogFactory.getLog(classOf[SharkServer2])

  def main(args: Array[String]) {
    try {
      LogUtils.initHiveLog4j()
    } catch {
      case e: LogInitializationException => {
        LOG.warn(e.getMessage)
      }
    }
    val optproc = new ServerOptionsProcessor("sharkserver2") //TODO: include load RDDs

    if (!optproc.process(args)) {
      LOG.fatal("Error starting SharkServer2 with given arguments")
      System.exit(-1)
    }

    Runtime.getRuntime.addShutdownHook(
      new Thread() {
        override def run() {
          SharkEnv.stop()
        }
      }
    )
  }

  try {
    val hiveConf = new HiveConf
    SharkConfVars.initializeWithDefaults(hiveConf)
    val server = new SharkServer2
    server.init(hiveConf)
    server.start()
    logInfo("SharkServer2 started")
  } catch {
    case t: Throwable => {
      LOG.fatal("Error starting SharkServer2", t)
      System.exit(-1)
    }
  }
}

class SharkServer2 extends HiveServer2 {
  override def init(hiveConf: HiveConf) {
    this.synchronized {
      val sharkCLIService = new SharkCLIService
      Utils.setSuperField("cliService", sharkCLIService, this)
      addService(sharkCLIService)
      val sthriftCLIService = new ThriftCLIService(sharkCLIService)
      Utils.setSuperField("thriftCLIService", sthriftCLIService, this)
      addService(sthriftCLIService)
      sharkInit(hiveConf)
    }
  }
}