com.dimajix.flowman.spec.connection.ConnectionReferenceSpec.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of flowman-spec Show documentation

A Spark based ETL tool

There is a newer version: 1.2.0-synapse3.3-spark3.3-hadoop3.3

/*
 * Copyright 2021 Kaya Kupferschmidt
 *
 * 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.dimajix.flowman.spec.connection

import com.fasterxml.jackson.core.JsonParser
import com.fasterxml.jackson.core.JsonProcessingException
import com.fasterxml.jackson.core.JsonToken
import com.fasterxml.jackson.databind.DeserializationContext
import com.fasterxml.jackson.databind.JsonMappingException
import com.fasterxml.jackson.databind.annotation.JsonDeserialize
import com.fasterxml.jackson.databind.deser.std.StdDeserializer

import com.dimajix.flowman.execution.Context
import com.dimajix.flowman.model.Connection
import com.dimajix.flowman.model.ConnectionIdentifier
import com.dimajix.flowman.model.ConnectionReference
import com.dimajix.flowman.model.Prototype
import com.dimajix.flowman.model.Reference
import com.dimajix.flowman.spec.Spec


@JsonDeserialize(using=classOf[ConnectionReferenceDeserializer])
abstract class ConnectionReferenceSpec extends Spec[Reference[Connection]]
final case class IdentifierConnectionReferenceSpec(connection:String) extends ConnectionReferenceSpec {
    override def instantiate(context: Context): Reference[Connection] = {
        val id = ConnectionIdentifier.parse(context.evaluate(connection))
        ConnectionReference(context,id)
    }
}
final case class ValueConnectionReferenceSpec(connection:Prototype[Connection]) extends ConnectionReferenceSpec {
    override def instantiate(context: Context): Reference[Connection] = {
        ConnectionReference(context, connection)
    }
}


private class ConnectionReferenceDeserializer(vc:Class[_]) extends StdDeserializer[ConnectionReferenceSpec](vc) {
    import java.io.IOException

    def this() = this(null)

    @throws[IOException]
    @throws[JsonProcessingException]
    def deserialize(jp: JsonParser, ctxt: DeserializationContext): ConnectionReferenceSpec = {
        jp.getCurrentToken match {
            case JsonToken.VALUE_STRING => {
                IdentifierConnectionReferenceSpec(jp.getText)
            }
            case JsonToken.START_OBJECT => {
                val spec = jp.readValueAs(classOf[ConnectionSpec])
                ValueConnectionReferenceSpec(spec)
            }
            case _ => throw JsonMappingException.from(jp, "Wrong type for ConnectionReference")
        }
    }
}