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

<code class="go">    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,
    }</code>

<code class="go">    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) ...
        }
    }</code>

<code class="go">    switch msg.method {
        case "ping":
            response.result = "pong"
        case "chat":
            var params struct {
                message string `json:"message"`
            }
            if err := json.unmarshal(msg.params, &params); 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) ...
    }</code>

<code class="go">if err := writeResponse(response); err != nil {
    log.Printf("Error writing response: %v", err)
}</code>