org.apache.spark.SparkBuildInfo.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spark-common-utils_2.13 Show documentation
The newest version!
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.spark

import java.util.Properties

private[spark] object SparkBuildInfo {

  val (
    spark_version: String,
    spark_branch: String,
    spark_revision: String,
    spark_build_user: String,
    spark_repo_url: String,
    spark_build_date: String,
    spark_doc_root: String) = {

    val resourceStream = Thread.currentThread().getContextClassLoader.
      getResourceAsStream("spark-version-info.properties")
    if (resourceStream == null) {
      throw new SparkException("Could not find spark-version-info.properties")
    }

    try {
      val unknownProp = ""
      val props = new Properties()
      props.load(resourceStream)
      (
        props.getProperty("version", unknownProp),
        props.getProperty("branch", unknownProp),
        props.getProperty("revision", unknownProp),
        props.getProperty("user", unknownProp),
        props.getProperty("url", unknownProp),
        props.getProperty("date", unknownProp),
        props.getProperty("docroot", unknownProp)
      )
    } catch {
      case e: Exception =>
        throw new SparkException("Error loading properties from spark-version-info.properties", e)
    } finally {
      if (resourceStream != null) {
        try {
          resourceStream.close()
        } catch {
          case e: Exception =>
            throw new SparkException("Error closing spark build info resource stream", e)
        }
      }
    }
  }
}