dataframe-duckdb.json

{
    "description": "DuckDB integration with Kotlin Dataframe",
    "properties": [
        {
            "name": "duckdb-version",
            "value": "0.10.2"
        },
        {
            "name": "duckdb-database",
            "value": ""
        },
        {
            "name": "arrow-version",
            "value": "16.0.0"
        },
        {
            "name": "duckdb-renovate-hint",
            "value": "update: package=org.duckdb:duckdb_jdbc"
        }
    ],
    "dependencies": [
        "org.duckdb:duckdb_jdbc:$duckdb-version",
        "org.apache.arrow:arrow-c-data:$arrow-version",
        "org.apache.arrow:arrow-memory-unsafe:$arrow-version"
    ],
    "imports": [
        "org.duckdb.DuckDBConnection",
        "org.duckdb.DuckDBResultSet",
        "java.sql.DriverManager",
        "java.sql.Statement",
        "org.apache.arrow.memory.RootAllocator",
        "org.apache.arrow.vector.ipc.ArrowReader"
    ],
    "init": [
        "%use dataframe(\"0.14.0-dev-3035\")",
        "fun DuckDBConnection.queryForDataFrame(query: String):AnyFrame{\r\nthis.createStatement().use {\r\nval duckDbResult = it.executeQuery(query) as DuckDBResultSet\r\nval arrowReader = duckDbResult.arrowExportStream(RootAllocator(),512) as ArrowReader\r\nreturn DataFrame.readArrow(arrowReader)\r\n}\r\n}",
        "val duckdbConnection = DriverManager.getConnection(\"jdbc:duckdb:$duckdb-database\") as DuckDBConnection"
    ]
}