io.edurt.datacap.plugin.scylladb.ScyllaDBPlugin.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of datacap-plugin-scylladb Show documentation

DataCap - ScyllaDB

There is a newer version: 2024.3.11

Show newest version

package io.edurt.datacap.plugin.scylladb

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings
import io.edurt.datacap.spi.Plugin
import io.edurt.datacap.spi.model.Configure
import io.edurt.datacap.spi.model.Response
import org.apache.commons.beanutils.BeanUtils.copyProperties
import org.apache.commons.lang3.ObjectUtils.isNotEmpty
import org.slf4j.LoggerFactory.getLogger

@SuppressFBWarnings(value = ["EI_EXPOSE_REP"])
class ScyllaDBPlugin : Plugin
{
    private val log = getLogger(this.javaClass)

    private var configure: Configure? = null
    private var connection: ScyllaDBConnection? = null
    private var response: Response? = null

    override fun validator(): String
    {
        return "SELECT release_version AS version FROM system.local"
    }

    override fun connect(configure: Configure?)
    {
        try
        {
            log.info("Connecting to ScyllaDB")
            response = Response()
            this.configure = Configure()
            copyProperties(this.configure, configure)
            connection = ScyllaDBConnection(this.configure !!, response !!)
        }
        catch (ex: Exception)
        {
            response !!.isConnected = false
            response !!.message = ex.message
        }
    }

    override fun execute(content: String?): Response
    {
        if (isNotEmpty(connection))
        {
            log.info("ScyllaDB connection established")
            response = connection?.response
            val processor = ScyllaDBAdapter(connection)
            response = processor.handlerExecute(content)
            log.info("ScyllaDB execution completed")
        }
        destroy()
        return response !!
    }

    override fun destroy()
    {
        if (isNotEmpty(connection))
        {
            log.info("ScyllaDB driver destroyed")
            connection?.destroy()
            configure = null
        }
    }
}