io.micronaut.ktor.env.MicronautKtorEnvironmentConfig.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of micronaut-ktor Show documentation
Includes extensions to Micronaut for Kotlin
There is a newer version: 4.4.0
/*
 * Copyright 2017-2020 original authors
 *
 * 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
 *
 * https://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 io.micronaut.ktor.env

import io.ktor.server.config.*
import io.micronaut.context.env.Environment
import io.micronaut.core.type.Argument
import java.util.*

/**
 * Implementation of the ApplicationConfig interface for Micronaut Ktor.
 *
 * @author graemerocher
 * @since 1.0
 */
class MicronautKtorEnvironmentConfig(val env : Environment, private val prefix : String? = "") : ApplicationConfig {

    private val configMap: Map
            by lazy(LazyThreadSafetyMode.PUBLICATION) {
                mapOf(env)
            }

    override fun config(path: String): ApplicationConfig {
        if (env.containsProperties(path)) {
            return MicronautKtorEnvironmentConfig(env, path)
        } else {
            throw ApplicationConfigurationException("No configuration found for path: $path")
        }
    }

    override fun configList(path: String): List {
        return Collections.singletonList(config(path))
    }

    override fun property(path: String): ApplicationConfigValue {
        return propertyOrNull(path) ?: throw ApplicationConfigurationException("No configuration found for path: $path")
    }

    override fun propertyOrNull(path: String): ApplicationConfigValue? {
        val fullPath = if(prefix.isNullOrEmpty()) path else "$prefix.$path"
        return if (env.containsProperty(fullPath)) {
            KtorApplicationConfigValue(fullPath, env)
        } else {
            null
        }
    }

    override fun keys(): Set {
        return Collections.unmodifiableSet(configMap.keys)
    }

    override fun toMap(): Map {
        return Collections.unmodifiableMap(configMap)
    }

    private fun mapOf(env: Environment): Map {
        val envMap: MutableMap> = LinkedHashMap()
        for (ps in env.propertySources) {
            for (s in ps) {
                if (env.containsProperty(s)) {
                    envMap.computeIfAbsent(s) { ArrayList() }.add(ps[s].toString())
                }
            }
        }
        return envMap
    }

    @Suppress("UNCHECKED_CAST")
    class KtorApplicationConfigValue(private val prop : String, private val env: Environment) : ApplicationConfigValue {
        override fun getList(): List {
            val requiredProperty = env.getProperty(prop, Argument.of(List::class.java, String::class.java))
            return requiredProperty as List
        }

        override fun getString(): String {
            return env.getRequiredProperty(prop, String::class.java)
        }
    }
}