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 同步
115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
package db
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
func driverKey(d string) string {
|
||
return strings.ToLower(strings.TrimSpace(d))
|
||
}
|
||
|
||
func isMySQLFamily(d string) bool {
|
||
switch driverKey(d) {
|
||
case "mysql", "mariadb", "tidb", "oceanbase", "polardb", "tdsql":
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func isPostgresFamily(d string) bool {
|
||
switch driverKey(d) {
|
||
case "postgresql", "postgres", "gaussdb", "opengauss", "kingbase":
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
func isMSSQL(d string) bool {
|
||
switch driverKey(d) {
|
||
case "sqlserver", "mssql":
|
||
return true
|
||
default:
|
||
return false
|
||
}
|
||
}
|
||
|
||
// Open 按驱动打开 *sql.DB
|
||
func Open(c ConnReq) (*sql.DB, error) {
|
||
var (
|
||
db *sql.DB
|
||
err error
|
||
)
|
||
switch {
|
||
case isMySQLFamily(c.Driver):
|
||
db, err = openMySQL(c)
|
||
case isPostgresFamily(c.Driver):
|
||
db, err = openPostgres(c)
|
||
case driverKey(c.Driver) == "sqlserver" || driverKey(c.Driver) == "mssql":
|
||
db, err = openMSSQL(c)
|
||
case driverKey(c.Driver) == "oracle":
|
||
db, err = openOracle(c)
|
||
case driverKey(c.Driver) == "sqlite":
|
||
db, err = openSQLite(c)
|
||
default:
|
||
return nil, fmt.Errorf("驱动「%s」尚未实现（当前支持: mysql/mariadb/tidb、postgresql、sqlserver、oracle、sqlite、redis）", c.Driver)
|
||
}
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
db.SetConnMaxLifetime(2 * time.Minute)
|
||
db.SetConnMaxIdleTime(5 * time.Minute) // 防空闲连接堆积
|
||
db.SetMaxOpenConns(5)
|
||
db.SetMaxIdleConns(2)
|
||
return db, nil
|
||
}
|
||
|
||
func TestConnection(c ConnReq) error {
|
||
if isAgentDriver(c.Driver) {
|
||
// JDBC agent 桥的 TCP 由 JVM 侧发起，SSH 隧道（本地端口转发）对 agent 不生效，明确拒绝
|
||
if c.SSHEnabled {
|
||
return fmt.Errorf("%s 走 JDBC agent，暂不支持 SSH 隧道", c.Driver)
|
||
}
|
||
if err := agentPing(c); err != nil {
|
||
return fmt.Errorf("数据库: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
dial, tun, err := ApplySSHTunnel(c)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if tun != nil {
|
||
defer tun.Close()
|
||
}
|
||
if err := testDial(dial); err != nil {
|
||
return fmt.Errorf("数据库: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func testDial(c ConnReq) error {
|
||
if isRedis(c.Driver) {
|
||
return testRedis(c)
|
||
}
|
||
if isES(c.Driver) {
|
||
return esPing(c)
|
||
}
|
||
if isInflux(c.Driver) {
|
||
return ifxPing(c)
|
||
}
|
||
if isAgentDriver(c.Driver) {
|
||
return agentPing(c)
|
||
}
|
||
db, err := Open(c)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer db.Close()
|
||
return db.Ping()
|
||
}
|