deepseek辅助编写的支持gmp高精度运算duckdb客户端利用DeepSeek编写能在DuckDB中读PostgreSQL表的表函数使用DeepSeek编写DuckDB支持中文字符集编码的read_csv表函数让DeepSeek参照libxls的xls2csv例子实现的DuckDB

read_xls表函数DeepSeek辅助编写的将DuckDB查询结果写多种格式文件的通用类

但使用不同功能要调用不同的程序很麻烦而且有的程序没有输入SQL语句的界面每次改变语句都要重新编译不实用所以将它们整合到一个带有交互式界面的程序。

整合的步骤其实很机械在主程序中添加相应函数的注册代码以及用某种方式将实现该函数的cpp程序加入编译后者既可以用#include

源代码.h的方式再将源代码.cpp加入编译命令行前一种更简单不过后一种更规范。

包含头文件和源代码

//将查询结果写入xls电子表格文件和docx文档文件支持中文

#include

*db_-instance;//注册read_pg自定义函数duckdb::ExtensionUtil::RegisterFunction(db_instance,

PGTableFunction::GetFunction());//注册read_csv_e自定义函数duckdb::ExtensionUtil::RegisterFunction(db_instance,

duckdb::ReadCSVEFunction::GetFunction());//注册read_xls自定义函数ExtensionUtil::RegisterFunction(db_instance,

ReadXLSFunction::GetFunction());//注册copy

to自定义函数,

支持xls和docx格式duckdb::RegisterMultiFormatCopyFunction(db_instance);因为有同名冲突修改了两处

1.在copy_files2.cpp中重新定义unsigned64_t后再包含xlslib.h再取消

#define

unsigned64_t2.在read_xls2.cpp中将结构体XLSGlobalState改名为ReadXLSGlobalState并同步修改两处对它的使用。

struct

{duckdb::unique_ptrReadXLSGlobalState

xls_state;

};unique_ptrGlobalTableFunctionState

ReadXLSFunction::InitGlobal(ClientContext

context,

make_uniqXLSFunctionGlobalState();auto

bind_data

)*input.bind_data;result-xls_state

make_uniqReadXLSGlobalState();用如下命令行编译通过注意xlslib是用的是修改后的支持中文的版本

export

LIBRARY_PATH/par:/usr/local/lib:/par/duck/build/src

export

LD_LIBRARY_PATH/par:/usr/local/lib:/par/duck/build/srcg

csv_reader2.cpp

minidocx-0.6.0/src/minidocx.cpp

minidocx-0.6.0/3rdparty/pugixml-1.13/*.cpp

minidocx-0.6.0/3rdparty/zip-0.2.1/*.c

ducksql2.cpp

minidocx-0.6.0/3rdparty/pugixml-1.13

minidocx-0.6.0/3rdparty/zip-0.2.1

xlslib/src

-O3这里将minidocx-0.6.0的代码直接链接到二进制文件中而没有使用动态链接库如果使用后者可以自行修改。

用官方duckdb

┌───────────────┬──────────────┬─────────────┬───────────────┬──────────┬───────────────┬─────────────┬───┬──────────────────┬──────────────────┬──────────┬──────────────┬───────────┬───────────┬────────────┐

database_name

├───────────────┼──────────────┼─────────────┼───────────────┼──────────┼───────────────┼─────────────┼───┼──────────────────┼──────────────────┼──────────┼──────────────┼───────────┼───────────┼────────────┤

system

├───────────────┴──────────────┴─────────────┴───────────────┴──────────┴───────────────┴─────────────┴───┴──────────────────┴──────────────────┴──────────┴──────────────┴───────────┴───────────┴────────────┤

rows

└──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘

D然后用程序打开同一个数据库将现在的函数列表与原有的备份比较找出新增的

./ducksql2

┌───────────────┬───────────────┐

function_name

├───────────────┼───────────────┤

read_csv_e

└───────────────┴───────────────┘可见新增的copy

to函数不在其中但其实它们存在当加载官方excel插件时报了如下错误

duckdb

/root/.duckdb/extensions/v1.3.0/linux_amd64/excel.duckdb_extension

threw

xlsx而把copy_files2.cpp中涉及xlsx格式的内容完全删除以后重新编译执行以上错误消失证明我们原有的copy_files2.cpp确实处理了xlsx格式虽然它实际上就是把xls文件改个后缀名但duckdb还是不允许。

加载excel插件后的新增函数列表如下现在我们的程序和它和平共处了。

duckdb

┌───────────────┬───────────────┐

function_name

├───────────────┼───────────────┤

read_pg

├───────────────┴───────────────┤

rows

└───────────────────────────────┘各个函数的用法在引用的文章中都有就不重复了。

另外将程序输出结果的部分做了如下修改使得explain语句能像官方CLI那样输出而不是输出explain_value表的结果。

std::string

lower;lower.resize(query.size());

query.end(),

┌───────────────┬───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┐

explain_key

├───────────────┼───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┤

physical_plan

┌───────────────────────────┐\n│

PROJECTION

└───────────────┴───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────┘修改后

duckdb

└───────────────────────────┘其他语句现在的输出看起来不错比如copy

duckdb

mpz_mul(111111111111111111111111,222222222222222222222222))

mpz_mul.docx;