如何使用 Golang SDK 查询 RDS 的任意表答案

【问题标题】：How to query any table of RDS using Golang SDK如何使用 Golang SDK 查询 RDS 的任意表
【发布时间】：2020-04-23 21:14:03
【问题描述】：

我正在编写一个 AWS lambda 来使用 Golang SDK 从 RDS(SQL Server) 查询 10 个不同的表。到目前为止我学到的是我们必须为表创建一个类似的结构来查询它。但是由于我要查询 10 个表，所以我不想为每个表创建结构，即使表架构也可能有一天会改变。

最近，我想为每个表创建一个 CSV 文件作为查询数据的备份并将其上传到 S3。那么是否可以直接将CSV文件导入到lambda中，以便我可以直接将其上传到S3？你可以在下面看到我当前的代码

func executeQuery(dbconnection *sql.DB) {

    println("\n\n----------Executing Query ----------")
    query := "select TOP 5 City,State,Country from IMBookingApp.dbo.Address"

    rows, err := dbconnection.Query(query)
    if err != nil {
        fmt.Println("Error:")
        log.Fatal(err)
    }

    println("rows", rows)

    defer rows.Close()
    count := 0
    for rows.Next() {
        var City, State, Country string

        rows.Columns

        err := rows.Scan(&City, &State, &Country)

        if err != nil {
            fmt.Println("Error reading rows: " + err.Error())
        }

        fmt.Printf("City: %s, State: %s, Country: %s\n", City, State, Country)
        count++
    }
}

此代码仅适用于Address表，不适用于其他表

我也用 GORM 试过了

package main

import (
    "fmt"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mssql"
)

type Currency struct {
    CurrencyId  int    `gorm:"column:CurrencyId;"`
    Code        string `gorm:"column:Code;"`
    Description string `gorm:"column:Description;"`
}

func main() {
    db, err := gorm.Open("mssql", "sqlserver://***")
    db.SingularTable(true)

    gorm.DefaultTableNameHandler = func(dbVeiculosGorm *gorm.DB, defaultTableName string) string {
        return "IMBookingApp.dbo.Currency"
    }

    fmt.Println("HasTable-Currency:", db.HasTable("ClientUser"))
    var currency Currency
    db.Debug().Find(&currency)

    fmt.Println("Currency:", currency)
    fmt.Println("Error", err)

    defer db.Close()

}

通过这两种方法，我找不到任何方法使代码对多个表通用。如果有人可以给我一些建议或者您可以指出一些资源，我将不胜感激。

【问题讨论】：

标签： sql-server amazon-web-services go amazon-rds go-gorm

【解决方案1】：

我没有测试此代码，但它应该让您了解如何将 Rows 提取到字符串数组中。


    defer rows.Close()
    columns, err := rows.Columns()
    if err != nil {
        panic(err)
    }
    for rows.Next() {
        receiver := make([]*string, len(columns))

        err := rows.Scan(receiver)
        if err != nil {
            fmt.Println("Error reading rows: " + err.Error())
        }
    }

GO 在内部将许多类型转换为字符串 - https://github.com/golang/go/blob/master/src/database/sql/convert.go#L219

如果无法转换数据，您有 2 个选项：

简单 - 更新您的 SQL 查询以返回字符串或字符串兼容数据
复杂。使用interface{} 的切片代替*string 的切片，并使用基于rows.ColumnTypes() 的正确类型的默认值填充它。稍后您必须将实际值转换为字符串以保存到 csv 中。

【讨论】：

感谢您的回复。我收到代码错误 - Error reading rows: sql: expected 3 destination arguments in Scan, not 1 你有什么想法吗？
我仍然遇到同样的错误。你可以在这里看到我的代码 - play.golang.org/p/ELpDd9lwNDf。可能是因为我使用的是 SQL Server 而不是 MySQL？
@RahulSatal 尝试 err := rows.Scan(receiver) - 没有 & 接收者
我尝试从err := rows.Scan(receiver) 中删除&。但是，我仍然遇到同样的错误。我也尝试过使用 interface{} 切片而不是 *string 切片。即使它没有工作。

【解决方案2】：

下面的代码对我有用 -

conn, _ := getConnection()   // Get database connection

rows, err := conn.Query(query)
if err != nil {
    fmt.Println("Error:")
    log.Fatal(err)
}

defer rows.Close()

columns, err := rows.Columns()

if err != nil {
    panic(err)
}

for rows.Next() {
    receiver := make([]string, len(columns))

    is := make([]interface{}, len(receiver))
    for i := range is {
        is[i] = &receiver[i]

        // each is[i] will be of type interface{} - compatible with Scan()
        // using the underlying concrete `*string` values from `receiver`
    }

    err := rows.Scan(is...)

    if err != nil {
        fmt.Println("Error reading rows: " + err.Error())
    }
    fmt.Println("receiver", receiver)

参考：-sql: expected 3 destination arguments in Scan, not 1 in Golang

【讨论】：