database

package

v0.5.13 Latest Latest Go to latest Published: Aug 13, 2026 License: Apache-2.0 Imports: 4 Imported by: 0

Details

Package database 提供数据库查询工具

支持 SQL 数据库查询操作。

View Source

const (
	// DefaultMaxRows 默认最大返回行数，防止 OOM
	DefaultMaxRows = 1000
)

This section is empty.

This section is empty.

type DatabaseTool struct {
	// contains filtered or unexported fields
}

DatabaseTool 数据库工具

func NewDatabaseTool(db *sql.DB, opts ...Option) *DatabaseTool

NewDatabaseTool 创建数据库工具

func (t *DatabaseTool) Tools() []tool.Tool

Tools 返回数据库工具集合

type Option func(*DatabaseTool)

Option 数据库工具选项

func WithMaxRows(maxRows int) Option

WithMaxRows 设置查询最大返回行数（默认 1000，防止 OOM）

func WithReadOnly(readOnly bool) Option

WithReadOnly 设置为只读模式

type QueryInput struct {
	SQL    string `json:"sql" description:"SQL 查询语句"`
	Params []any  `json:"params,omitempty" description:"查询参数"`
}

QueryInput 查询输入

type QueryOutput struct {
	Columns []string `json:"columns"`
	Rows    [][]any  `json:"rows"`
	Count   int      `json:"count"`
}

QueryOutput 查询输出

func QuickQuery(ctx context.Context, db *sql.DB, query string, args ...any) (QueryOutput, error)

QuickQuery 快速查询辅助函数