| ///* |
| // * 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.amaterasu.leader.common.utilities |
| // |
| //import java.io.{File, FileInputStream} |
| //import java.nio.file.{Files, Paths} |
| // |
| //import com.fasterxml.jackson.databind.ObjectMapper |
| //import com.fasterxml.jackson.dataformat.yaml.YAMLFactory |
| //import com.fasterxml.jackson.module.scala.DefaultScalaModule |
| //import org.apache.amaterasu.common.configuration.ClusterConfig |
| //import org.apache.amaterasu.common.dataobjects.{ActionData, ExecData, TaskData} |
| //import org.apache.amaterasu.common.execution.dependencise.{Dependencies, PythonDependencies} |
| //import org.apache.amaterasu.common.logging.Logging |
| //import org.apache.amaterasu.common.runtime.Environment |
| //import org.yaml.snakeyaml.Yaml |
| // |
| //import scala.collection.JavaConverters._ |
| //import scala.collection.mutable |
| //import scala.io.Source |
| // |
| // |
| //object DataLoader extends Logging { |
| // |
| // val mapper = new ObjectMapper() |
| // mapper.registerModule(DefaultScalaModule) |
| // |
| // val ymlMapper = new ObjectMapper(new YAMLFactory()) |
| // ymlMapper.registerModule(DefaultScalaModule) |
| // |
| // def getTaskDataBytes(actionData: ActionData, env: String): Array[Byte] = { |
| // mapper.writeValueAsBytes(getTaskData(actionData, env)) |
| // } |
| // |
| // def getTaskData(actionData: ActionData, env: String): TaskData = { |
| // val srcFile = actionData.getSrc |
| // var src = "" |
| // |
| // if(!srcFile.isEmpty){ |
| // src = Source.fromFile(s"repo/src/$srcFile").mkString |
| // } |
| // |
| // val envValue = Source.fromFile(s"repo/env/$env/job.yml").mkString |
| // |
| // val envData = ymlMapper.readValue(envValue, classOf[Environment]) |
| // |
| // val exports = actionData.getExports.asScala.toMap // Kotlin to Scala TODO: Remove me as fast as you can |
| // |
| // TaskData(src, envData, actionData.getGroupId, actionData.getTypeId, exports) |
| // } |
| // |
| // def getTaskDataString(actionData: ActionData, env: String): String = { |
| // mapper.writeValueAsString(getTaskData(actionData, env)) |
| // } |
| // |
| // def getExecutorDataBytes(env: String, clusterConf: ClusterConfig): Array[Byte] = { |
| // mapper.writeValueAsBytes(getExecutorData(env, clusterConf)) |
| // } |
| // |
| // def getExecutorData(env: String, clusterConf: ClusterConfig): ExecData = { |
| // |
| // // loading the job configuration |
| // val envValue = Source.fromFile(s"repo/env/$env/job.yml").mkString //TODO: change this to YAML |
| // val envData = ymlMapper.readValue(envValue, classOf[Environment]) |
| // // loading all additional configurations |
| // val files = new File(s"repo/env/$env/").listFiles().filter(_.isFile).filter(_.getName != "job.yml") |
| // val config = files.map(yamlToMap).toMap |
| // // loading the job's dependencies |
| // var depsData: Dependencies = null |
| // var pyDepsData: PythonDependencies = null |
| // if (Files.exists(Paths.get("repo/deps/jars.yml"))) { |
| // val depsValue = Source.fromFile(s"repo/deps/jars.yml").mkString |
| // depsData = ymlMapper.readValue(depsValue, classOf[Dependencies]) |
| // } |
| // if (Files.exists(Paths.get("repo/deps/python.yml"))) { |
| // val pyDepsValue = Source.fromFile(s"repo/deps/python.yml").mkString |
| // pyDepsData = ymlMapper.readValue(pyDepsValue, classOf[PythonDependencies]) |
| // } |
| // val data = mapper.writeValueAsBytes(ExecData(envData, depsData, pyDepsData, config)) |
| // ExecData(envData, depsData, pyDepsData, config) |
| // } |
| // |
| // def yamlToMap(file: File): (String, Map[String, Any]) = { |
| // |
| // val yaml = new Yaml() |
| // val conf = yaml.load(new FileInputStream(file)).asInstanceOf[java.util.Map[String, Any]].asScala.toMap |
| // |
| // (file.getName.replace(".yml", ""), conf) |
| // } |
| // |
| // def getExecutorDataString(env: String, clusterConf: ClusterConfig): String = { |
| // mapper.writeValueAsString(getExecutorData(env, clusterConf)) |
| // } |
| // |
| //} |
| // |
| //class ConfMap[String, T <: ConfMap[String, T]] extends mutable.ListMap[String, Either[String, T]] |