在VSCODE中首次使DBCHAT工作 - 第9部分

小伙伴们有没有觉得学习文章很有意思？有意思就对了！今天就给大家带来《在VSCODE中首次使DBCHAT工作 - 第9部分》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

Hexmos创始人Shrijith Venkatrama正在构建LiveAPI，一个能快速生成高质量API文档的工具，并以此简化工程流程。本教程系列记录了他构建dbchat的历程——一个使用AI聊天探索和改进数据库的便捷工具。

之前的文章：

• 构建dbchat - 通过简单的聊天探索和改进您的数据库（第一部分）
• dbchat：在Golang中构建玩具部署（第二部分）
• dbchat第三部分 - 配置、连接和数据库转储
• 通过dbchat和Gemini与您的数据库聊天（第四部分）
• 语言服务器协议 - 构建dbchat（第五部分）
• 制作dbchat VS Code扩展 - 带LSP后端的乒乓球（第六部分）
• 启动dbchat的VS Code扩展UI（第七部分）
• 从VS Code扩展程序管理Toml配置 - dbchat第八部分
• 工作版本的截图
• dbchat的默认视图：

选择数据库后，进入聊天视图：

使用简单的提示获取Gmail用户和非Gmail用户：

查询结果以表格形式呈现。同样的逻辑也适用于“获取所有非Gmail用户”查询。

dbchat更新的更详细视图

代码处理与LSP客户端（语言服务器协议客户端）的通信：

1. 设置阶段：配置和初始化

• 加载配置： 读取配置文件（包含数据库连接字符串等设置）。
• 创建查询处理程序： 设置查询处理程序，负责理解和执行用户请求。
• 构建处理程序： 创建dbchatHandler，管理配置、查询处理程序和数据库连接。

    config, err := utils.loadconfig()
    if err != nil {
        log.printf("warning: could not load config: %v", err)
        config = &utils.config{connections: make(map[string]string)}
    }

    queryhandler, err := query.newhandler(config.llm.geminikey)
    if err != nil {
        log.printf("warning: could not create query handler: %v", err)
    }

    handler := &dbchathandler{
        config:       config,
        queryhandler: queryhandler,
    }

2. LSP通信：心跳

• LSP循环： 持续监听LSP客户端的消息。
• 读取消息： 读取客户端消息，注意内容长度标题。
• 解析消息： 将原始数据转换为结构化的JSON-RPC消息。

    for {
        // read content-length header
        // ... (code for reading header) ...

        // read message body
        // ... (code for reading message body) ...

        // parse json-rpc message
        var msg jsonrpcmessage
        if err := json.unmarshal(body, &msg); err != nil {
            // ... (handle parsing error) ...
        }
    }

3. 处理请求

• “ping”请求： 返回“pong”。
• “chat”请求： 提取用户聊天消息，传递给dbchatHandler的eval方法处理，并将结果打包到响应中发送回客户端。

    switch msg.method {
        case "ping":
            response.result = "pong"
        case "chat":
            var params struct {
                message string `json:"message"`
            }
            if err := json.unmarshal(msg.params, ¶ms); err != nil {
                // ... (handle invalid params) ...
            }

            log.printf("processing chat message: %q", params.message)
            result := handler.eval(params.message)
            log.printf("eval result: %q", result)

            response.result = map[string]interface{}{
                "message": result,
            }
        // ... (handle other methods) ...
    }

4. 错误处理和响应

• 捕获错误： 优雅地处理解析消息或数据库错误。
• 发送响应： 以正确的JSON-RPC格式发送响应消息。

if err := writeResponse(response); err != nil {
    log.Printf("Error writing response: %v", err)
}

结论： dbchat能够管理数据库连接，支持多种数据库，允许使用自然语言查询，并以表格格式返回结果。

今天带大家了解了的相关知识，希望对你有所帮助；关于文章的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~