akkeeper.master.MasterMain.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of akkeeper_2.12 Show documentation

akkeeper

There is a newer version: 0.3.3

/*
 * Copyright 2017-2018 Iaroslav Zeigerman
 *
 * 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 akkeeper.master

import java.io.File
import akkeeper.utils.CliArguments._
import scopt.OptionParser
import scala.util.control.NonFatal

object MasterMain extends App {

  val optParser = new OptionParser[MasterArguments]("akkeeperMaster") {
    head("akkeeperMaster", "0.2.0")

    opt[String](AppIdArg).required().action((v, c) => {
      c.copy(appId = v)
    }).text("ID of this application")

    opt[File](ConfigArg).valueName("").optional().action((v, c) => {
      c.copy(config = Some(v))
    }).text("custom configuration file")

    opt[String](PrincipalArg).valueName("principal").action((v, c) => {
      c.copy(principal = Some(v))
    })
  }

  try {
    optParser.parse(args, MasterArguments()).foreach(args => {
      new YarnMasterRunner().run(args)
      sys.exit()
    })
  } catch {
    case NonFatal(e) =>
      e.printStackTrace()
      sys.exit(1)
  }
}