自定义 DataCap 插件 Kotlin 实现

qianmoQ

DataCap 是一套基于 SpringBoot 开发的全平台数据 (数据库管理工具) 功能比较完善，建议下载使用: github.com/EdurtIO/datacap 目前已经支持 40+ 多种数据源。国内首个应用 ChatGPT 到数据管理系统中项目。

DataCap 支持自定义插件，使用者可以编写自己的插件集成到系统中。该文档主要讲解如何快速集成一个插件到 DataCap 系统中。

本文使用集成基于 JDBC 协议的 StarRocks 数据存储系统来演示。

`pom.xml` 依赖

<dependency>
    <groupId>io.edurt.datacap</groupId>
    <artifactId>datacap-spi</artifactId>
    <scope>provided</scope>
</dependency>
<dependency>
    <groupId>io.edurt.datacap</groupId>
    <artifactId>datacap-common</artifactId>
</dependency>
<dependency>
    <groupId>commons-beanutils</groupId>
    <artifactId>commons-beanutils</artifactId>
</dependency>
<dependency>
    <groupId>org.slf4j</groupId>
    <artifactId>slf4j-api</artifactId>
</dependency>
<dependency>
    <groupId>org.slf4j</groupId>
    <artifactId>slf4j-simple</artifactId>
    <scope>test</scope>
</dependency>
<dependency>
    <groupId>org.testcontainers</groupId>
    <artifactId>testcontainers</artifactId>
    <scope>test</scope>
</dependency>

以上配置添加了 datacap-spi 和 datacap-common 模块，其他的是一些辅助依赖。

需要注意的是如果您是单独开启的项目需要指定各个依赖的版本号。

插件加载器

class StarRocksPluginModule : AbstractPluginModule(), PluginModule {
    override fun getName(): String {
        return "StarRocks"
    }

    override fun getType(): PluginType {
        return PluginType.JDBC
    }

    override fun get(): AbstractPluginModule {
        return this
    }

    override fun configure() {
        val module = Multibinder.newSetBinder(binder(), String::class.java)
        module.addBinding().toInstance(this.javaClass.simpleName)
        val plugin: Multibinder<Plugin> = Multibinder.newSetBinder(binder(), Plugin::class.java)
        plugin.addBinding().to(StarRocksPlugin::class.java)
    }
}

加载器需要继承 AbstractPluginModule 类，并实现 PluginModule 接口，这样系统会在启动时自动将插件加载到系统中。

需要注意的是，需要覆盖父类中的 configure() 方法，并将插件绑定到系统中。

插件执行器

class StarRocksPlugin : Plugin {
    private val log = getLogger(StarRocksPlugin::class.java)

    private var jdbcConfigure: JdbcConfigure? = null
    private var jdbcConnection: JdbcConnection? = null
    private var jdbcResponse: Response? = null

    override fun name(): String {
        return "StarRocks"
    }

    override fun description(): String {
        return "Integrate StarRocks data sources"
    }

    override fun type(): PluginType {
        return PluginType.JDBC
    }

    override fun connect(configure: Configure?) {
        try {
            log.info("Connecting to StarRocks")
            jdbcResponse = Response()
            jdbcConfigure = JdbcConfigure()
            BeanUtils.copyProperties(jdbcConfigure, configure)
            jdbcConfigure!!.jdbcDriver = "com.mysql.cj.jdbc.Driver"
            jdbcConfigure!!.jdbcType = "mysql"
            jdbcConnection = object : JdbcConnection(jdbcConfigure, jdbcResponse) {}
        } catch (ex: Exception) {
            jdbcResponse!!.isConnected = false
            jdbcResponse!!.message = ex.message
        }
    }

    override fun execute(content: String?): Response {
        if (ObjectUtils.isNotEmpty(jdbcConnection)) {
            log.info("Execute starrocks plugin logic started")
            jdbcResponse = jdbcConnection?.response
            val processor = JdbcAdapter(jdbcConnection)
            jdbcResponse = processor.handlerExecute(content)
            log.info("Execute starrocks plugin logic end")
        }
        destroy()
        return jdbcResponse!!
    }

    override fun destroy() {
        if (ObjectUtils.isNotEmpty(jdbcConnection)) {
            jdbcConnection?.destroy()
            jdbcConnection = null
        }
    }
}

执行器需要实现 Plugin 接口，该接口中提供了以下方法

name(): 插件有唯一名称，同名插件只会在第一次加载时生效
description(): 对于该插件的描述
type(): 插件类型
connect(Configure configure): 插件需要提前连接信息，比如当前插件插件，就是插件的连接阶段（系统预设 HTTP 连接方式直接使用）。
execute(String content): 具体执行操作逻辑
destroy(): 插件最后的销毁，注意销毁需要包含连接中的信息

插件转换器

插件转换器用于对当前插件执行后的结果的转化，将其转换为 DataCap 中可以使用的逻辑。主要是用于封装 Response 返回结果。

本文是基于 JDBC 的插件所以直接继承 JdbcAdapter 父类即可实现部分功能。

SPI 加载器

在 resources 源目录下添加 META-INF 和 services 目录

services 在 resources 目录中需要

创建 io.edurt.datacap.spi.PluginModule 文件，内容如下

io.edurt.datacap.plugin.jdbc.starrocks.StarRocksPluginModule

该文件的内容是我们定义好的插件加载模块。

插件的单元测试可以参考已经发布的插件进行测试

小助手

我认为使用Kotlin实现自定义DataCap插件是一个非常值得感谢的观点。首先，Kotlin是一种非常强大和灵活的编程语言，它具有许多现代化的特性，可以提高开发效率和代码质量。使用Kotlin来实现插件可以使我们能够更快地开发和扩展DataCap功能。

其次，自定义DataCap插件的实现对于那些希望根据自己的需求和业务逻辑来定制DataCap功能的开发人员来说是非常有价值的。通过自定义插件，我们可以根据实际需求添加新的功能、修改现有功能，甚至完全改写DataCap的行为。这样一来，我们可以更好地适应各种不同的应用场景和业务要求。

当然，与谁讨论这个问题是个有争议的话题。有些人可能认为DataCap本身已经具备足够的功能和扩展性，不需要进行自定义插件的开发。他们认为使用DataCap的默认功能就能够满足大多数用户的需求，而自定义插件只会增加开发和维护的复杂性。

然而，我认为与DataCap的开发团队进行讨论是最好的选择。他们可以提供有关DataCap使用和扩展的专业建议，帮助我们更好地理解和利用DataCap的功能。此外，与其他开发者和用户社区讨论也是非常有益的，可以分享经验和解决问题，共同推动DataCap插件的发展和改进。

总而言之，使用Kotlin实现自定义DataCap插件是一个值得感谢的观点。它可以提高开发效率和代码质量，同时也为我们提供了更大的自由度来定制DataCap的功能。与谁讨论这个问题可能存在一些争议，但与DataCap的开发团队和用户社区讨论是一个明智的选择。让我们激发更多的创意和想法，为DataCap插件的发展贡献力量。