logan
a9510235bb
驱动接入（全部本机真实例验收）： - Elasticsearch 原生 SQL API、InfluxDB 原生 HTTP+Flux（官方 JDBC 付费许可，实测弃用） - ZooKeeper/MongoDB JVM agent（连接时自动下载官方驱动 jar，sha256 钉版本到 ~/.opsxdb/drivers/） - 达梦 JDBC agent 桥就绪（待厂商 jar+实例）；TiDB 真实例验收 - 协议族代码审查放行 PolarDB/TDSQL/OceanBase/Kingbase/GaussDB/Oracle - 驱动能力模型：按驱动开关对象树分组/右键菜单/行编辑，NoSQL 树拍平（对象直挂库节点） 驱动管理： - 驱动管理弹框（状态/预装/离线包导入/JDK 检测）+ /api/drivers/* 三接口 - opsxdb-driver 驱动商店仓库（清单/fetch.sh/pack.sh 离线包/import.sh） - 连接时懒加载：DMG 零 Java 产物，首连自动下载（实测 ZK 九件套 1.35s），离线包自动回退 UI/交互： - 设置中心（⚙）：常规/执行/结果/驱动/快捷键/关于，左导航+右面板（对标 opsxshell） - 实例级展开箭头（行首 SVG chevron，替代字体回退失真的 Unicode 三角） - 树内嵌套渲染修复：对象树挂在激活连接下；工具栏窄窗口不断行（按钮 nowrap+整颗换行） - 侧栏连接行精简（图标+连接名）、SQLite SSH 守卫、Oracle 死参数清理 质量： - Go 协议层/离线回退单测、E2E 门控测试（ES/ZK/Influx/Mongo/TiDB/JDBC 桥） - 驱动注册表单一事实来源 connection-types.json，README/BACKLOG/AGENTS 同步
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
package ai
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
)
|
||
|
||
// stubProvider 离线占位实现:未配置真实模型时的兜底,内容为字符串模板。
|
||
type stubProvider struct{}
|
||
|
||
func (stubProvider) Name() string { return "stub" }
|
||
|
||
func (stubProvider) Complete(_ context.Context, req ChatReq) (ChatRes, error) {
|
||
return chatStub(req), nil
|
||
}
|
||
|
||
func chatStub(req ChatReq) ChatRes {
|
||
driver := req.Driver
|
||
if driver == "" {
|
||
driver = "mysql"
|
||
}
|
||
db := req.Database
|
||
if db == "" {
|
||
db = "your_db"
|
||
}
|
||
sql := strings.TrimSpace(req.SQL)
|
||
prompt := strings.TrimSpace(req.Prompt)
|
||
|
||
switch strings.ToLower(req.Action) {
|
||
case "generate", "ask":
|
||
q := prompt
|
||
if q == "" {
|
||
q = "示例查询"
|
||
}
|
||
gen := fmt.Sprintf("-- 根据描述生成（%s / %s）\n-- 需求: %s\nSELECT *\nFROM `%s`.some_table\nWHERE 1=1\nLIMIT 100;", driver, db, q, db)
|
||
return ChatRes{
|
||
Content: fmt.Sprintf("已根据「%s」生成 SQL（stub，可替换为真实模型）。\n\n```sql\n%s\n```", q, gen),
|
||
SQL: gen,
|
||
}
|
||
case "explain":
|
||
if sql == "" {
|
||
sql = "SELECT 1"
|
||
}
|
||
return ChatRes{Content: fmt.Sprintf("## 解释 SQL\n\n```sql\n%s\n```\n\n- 这是一条在 **%s** 上执行的查询。\n- 建议检查 WHERE 条件是否走索引，避免 `SELECT *` 拉取过大结果集。\n- （stub 回复）", sql, driver)}
|
||
case "optimize":
|
||
opt := sql
|
||
if opt == "" {
|
||
opt = "SELECT id, name FROM users WHERE status = 1 LIMIT 100;"
|
||
} else if strings.Contains(strings.ToUpper(opt), "SELECT *") {
|
||
opt = strings.Replace(opt, "*", "id /* TODO: 显式列 */", 1)
|
||
}
|
||
opt = strings.TrimSpace(opt)
|
||
if !strings.Contains(strings.ToUpper(opt), "LIMIT") {
|
||
opt = opt + "\nLIMIT 100;"
|
||
}
|
||
return ChatRes{
|
||
Content: fmt.Sprintf("## 优化建议\n\n1. 避免不必要的全表扫描\n2. 显式列出列名\n3. 加合理 LIMIT\n\n```sql\n%s\n```", opt),
|
||
SQL: opt,
|
||
}
|
||
case "fix":
|
||
fixed := sql
|
||
if fixed == "" {
|
||
fixed = "SELECT 1 AS ok;"
|
||
}
|
||
msg := req.Error
|
||
if msg == "" {
|
||
msg = "未知错误"
|
||
}
|
||
return ChatRes{
|
||
Content: fmt.Sprintf("## 修复错误\n\n原错误：`%s`\n\n尝试修正后的 SQL：\n\n```sql\n%s\n```\n\n（stub：请结合真实报错继续调整）", msg, fixed),
|
||
SQL: fixed,
|
||
}
|
||
case "dialect":
|
||
out := sql
|
||
if out == "" {
|
||
out = "SELECT * FROM dual"
|
||
}
|
||
out = strings.ReplaceAll(out, "`", "\"")
|
||
return ChatRes{
|
||
Content: fmt.Sprintf("## 方言转换 → PostgreSQL 风格（示例）\n\n```sql\n%s\n```", out),
|
||
SQL: out,
|
||
}
|
||
case "sample":
|
||
sample := fmt.Sprintf("SELECT id, name, created_at\nFROM `%s`.users\nORDER BY id DESC\nLIMIT 20;", db)
|
||
return ChatRes{
|
||
Content: fmt.Sprintf("## 样例 SQL（%s）\n\n```sql\n%s\n```", driver, sample),
|
||
SQL: sample,
|
||
}
|
||
default:
|
||
return ChatRes{Content: "未知 AI 动作。可用：ask / generate / explain / optimize / fix / dialect / sample"}
|
||
}
|
||
}
|