net.nemerosa.ontrack.extension.influxdb.InfluxDBController.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ontrack-extension-influxdb Show documentation

Ontrack module: ontrack-extension-influxdb

There is a newer version: 4.4.5

Show newest version

package net.nemerosa.ontrack.extension.influxdb

import net.nemerosa.ontrack.model.security.ApplicationManagement
import net.nemerosa.ontrack.model.security.SecurityService
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty
import org.springframework.http.HttpStatus
import org.springframework.web.bind.annotation.*

/**
 * REST controller accessible from the UI.
 */
@ConditionalOnProperty(prefix = INFLUXDB_EXTENSION_PROPERTIES_PREFIX,
    name = ["enabled"],
    havingValue = "true",
    matchIfMissing = false)
@RestController
@RequestMapping("/extension/influxdb")
class InfluxDBController(
    private val influxDBExtensionProperties: InfluxDBExtensionProperties,
    private val influxDBConnection: InfluxDBConnection,
    private val securityService: SecurityService
) {

    /**
     * Gets the status of the current connection
     */
    @GetMapping("")
    fun getStatus(): InfluxDBStatus {
        securityService.checkGlobalFunction(ApplicationManagement::class.java)
        return InfluxDBStatus(
            url = influxDBExtensionProperties.uri,
            db = influxDBExtensionProperties.db,
            status = influxDBConnection.isValid
        )
    }

    /**
     * Resets the connection
     */
    @PostMapping("")
    @ResponseStatus(HttpStatus.NO_CONTENT)
    fun reset() {
        securityService.checkGlobalFunction(ApplicationManagement::class.java)
        influxDBConnection.reset()
    }

    class InfluxDBStatus(
        val url: String,
        val db: String,
        val status: Boolean
    )

}