使用Golang SDK查询任意RDS表

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《使用Golang SDK查询任意RDS表》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

我正在编写一个 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()

}

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

解决方案

下面的代码对我有用 -

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

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

    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 个选择：

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

以上就是《使用Golang SDK查询任意RDS表》的详细内容，更多关于的资料请关注golang学习网公众号！