summaryrefslogtreecommitdiffstats
path: root/pnda-ztt-app/src/main/scala/com/cisco/ztt/meta/YamlReader.scala
blob: 4b72cb1526377ac4ec863198f11378d1539ed066 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*
 * Copyright (c) 2018 Cisco Systems. All rights reserved.
 *
 * 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 com.cisco.ztt.meta

import org.apache.log4j.Logger
import org.springframework.core.io.support.PathMatchingResourcePatternResolver

import com.fasterxml.jackson.databind.ObjectMapper
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory
import com.fasterxml.jackson.module.scala.DefaultScalaModule

case class Unit(
        format: String,
        input_topic: String,
        processor: String,
        output_topic: String,
        publish_src: String,
        timeseries_namespace: String,
        xr_telemetry: Array[Telemetry],
        ves_telemetry: Array[Telemetry])
case class Telemetry(path: String, keys: Option[Array[Item]], content: Array[Item])
case class Item(name: String, display_name: String, ts_name: String)

object YamlReader {

    private[this] val logger = Logger.getLogger(getClass().getName());

    val mapper: ObjectMapper = new ObjectMapper(new YAMLFactory())
    mapper.registerModule(DefaultScalaModule)

    def load(pattern: String = "classpath*:meta/*.yaml"): Metadata = {
        val patternResolver = new PathMatchingResourcePatternResolver()
        val mappingLocations = patternResolver.getResources(pattern)
        val units = mappingLocations.map(loc => {
            logger.info("Reading metadata " + loc)
            mapper.readValue(loc.getInputStream, classOf[Unit])
        });
        new Metadata(units)
    }

    def parse(yaml: String): Unit = {
        val meta: Unit = mapper.readValue(yaml, classOf[Unit])
        meta
    }

    def dump(meta: Unit) = {
        println(" input = " + meta.input_topic)
        println("output = " + meta.output_topic)
        meta.xr_telemetry.map(m => {
            println("  path = " + m.path)
            println("keys:")
            m.keys.getOrElse(Array[Item]()).map(item => {
                println("    " + item.name + " -> " + item.display_name)
            });
            println("content:")
            m.content.map(item => {
                println("    " + item.name + " -> " + item.display_name)
            });
        });
    }
}