Golang如何从数据库中获取可用表

感谢您的努力。实际上，我想通过ODBC适配器接口安全地连接到任何数据库（例如：PostgreSQL、MySQL、MSSQL等），并检索在指定为数据库源的源中可用的表。

在Golang中从数据库获取可用表的方法取决于具体的数据库类型。以下是针对不同数据库的通用实现方法：

1. 修复你的ODBC连接实现

首先，你的数据源传递方式需要调整：

func InitialiseDBConnection(driverType, dataSourceName string) error {
    // 直接使用传入的dataSourceName，而不是重新构建
    db, err := sql.Open(driverType, dataSourceName)
    if err != nil {
        return fmt.Errorf("failed to open database: %v", err)
    }

    // 检查数据库是否已连接
    if err = db.Ping(); err != nil {
        return fmt.Errorf("failed to ping database: %v", err)
    }

    DBClient = db
    return nil // 不要在这里关闭连接！
}

2. 通用表查询方法

// 获取所有表名
func GetAllTables() ([]string, error) {
    var tables []string
    
    // 通用SQL查询（需要根据数据库类型调整）
    query := `
        SELECT table_name 
        FROM information_schema.tables 
        WHERE table_schema = 'public' 
        AND table_type = 'BASE TABLE'
    `
    
    rows, err := DBClient.Query(query)
    if err != nil {
        // 如果通用查询失败，尝试特定数据库的查询
        return getTablesByDriver()
    }
    defer rows.Close()
    
    for rows.Next() {
        var tableName string
        if err := rows.Scan(&tableName); err != nil {
            return nil, fmt.Errorf("scan error: %v", err)
        }
        tables = append(tables, tableName)
    }
    
    return tables, nil
}

3. 根据数据库类型获取表

func getTablesByDriver() ([]string, error) {
    var tables []string
    var query string
    
    // 根据驱动类型使用不同的系统表查询
    switch driverType {
    case "postgres", "pgx":
        query = "SELECT tablename FROM pg_catalog.pg_tables WHERE schemaname != 'pg_catalog' AND schemaname != 'information_schema'"
    
    case "mysql":
        query = "SHOW TABLES"
    
    case "sqlserver", "mssql":
        query = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_TYPE = 'BASE TABLE'"
    
    case "sqlite3":
        query = "SELECT name FROM sqlite_master WHERE type='table'"
    
    case "odbc":
        // ODBC通用查询
        query = "SELECT table_name FROM information_schema.tables"
    
    default:
        return nil, fmt.Errorf("unsupported driver type: %s", driverType)
    }
    
    rows, err := DBClient.Query(query)
    if err != nil {
        return nil, fmt.Errorf("query error: %v", err)
    }
    defer rows.Close()
    
    for rows.Next() {
        var tableName string
        if err := rows.Scan(&tableName); err != nil {
            return nil, fmt.Errorf("scan error: %v", err)
        }
        tables = append(tables, tableName)
    }
    
    return tables, nil
}

4. 改进的Table结构和方法

type TableInfo struct {
    TableName    string
    TableSchema  string
    TableType    string
    CreateTime   sql.NullTime
}

// 获取详细的表信息
func GetTableDetails() ([]TableInfo, error) {
    var tables []TableInfo
    
    query := `
        SELECT 
            table_name, 
            table_schema, 
            table_type,
            create_time
        FROM information_schema.tables 
        WHERE table_schema NOT IN ('information_schema', 'pg_catalog', 'sys')
        ORDER BY table_schema, table_name
    `
    
    rows, err := DBClient.Query(query)
    if err != nil {
        return nil, fmt.Errorf("query error: %v", err)
    }
    defer rows.Close()
    
    for rows.Next() {
        var table TableInfo
        if err := rows.Scan(&table.TableName, &table.TableSchema, 
                           &table.TableType, &table.CreateTime); err != nil {
            return nil, fmt.Errorf("scan error: %v", err)
        }
        tables = append(tables, table)
    }
    
    return tables, nil
}

5. 使用示例

package main

import (
    "database/sql"
    "fmt"
    "log"
    _ "github.com/alexbrainman/odbc"
)

func main() {
    // ODBC连接字符串示例
    dsn := "Driver={SQL Server};Server=localhost;Database=testdb;Trusted_Connection=yes;"
    
    // 初始化连接
    db, err := sql.Open("odbc", dsn)
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()
    
    // 获取所有表
    tables, err := GetAllTables(db)
    if err != nil {
        log.Fatal(err)
    }
    
    fmt.Println("Available tables:")
    for _, table := range tables {
        fmt.Printf("- %s\n", table)
    }
}

// 独立函数版本
func GetAllTables(db *sql.DB) ([]string, error) {
    var tables []string
    
    // 尝试多种查询方法
    queries := []string{
        "SELECT name FROM sysobjects WHERE xtype='U'", // SQL Server
        "SELECT table_name FROM information_schema.tables",
        "SHOW TABLES", // MySQL
    }
    
    for _, query := range queries {
        rows, err := db.Query(query)
        if err == nil {
            defer rows.Close()
            for rows.Next() {
                var tableName string
                if err := rows.Scan(&tableName); err == nil {
                    tables = append(tables, tableName)
                }
            }
            if len(tables) > 0 {
                return tables, nil
            }
        }
    }
    
    return tables, fmt.Errorf("unable to retrieve tables")
}

6. 处理特定数据库的系统表

// 获取SQL Server的表
func GetSQLServerTables(db *sql.DB) ([]string, error) {
    var tables []string
    
    rows, err := db.Query(`
        SELECT s.name + '.' + t.name 
        FROM sys.tables t
        INNER JOIN sys.schemas s ON t.schema_id = s.schema_id
        ORDER BY s.name, t.name
    `)
    if err != nil {
        return nil, err
    }
    defer rows.Close()
    
    for rows.Next() {
        var tableName string
        if err := rows.Scan(&tableName); err != nil {
            return nil, err
        }
        tables = append(tables, tableName)
    }
    
    return tables, nil
}

关键点：

1. 数据源名称应该直接传递给sql.Open()，而不是在函数内部重新构建
2. 不同数据库有不同的系统表查询方式
3. 使用information_schema.tables是跨数据库的通用方法
4. 需要处理不同驱动类型的特定查询语法