Golang sql/database多查询rows.Next()合并结果问题求助

我正在开发一个获取用户信息的函数，需求是将两个SQL查询的结果合并为一个返回结果，需在rows.Next()与rows1.Next()中组合user_id、subject、phone及body字段。以下是我的代码，但无法正常运行，恳请提供解决方案建议：

func GetUsers() (users []Users, err error) {
    users = make([]Users, 0)
    rows, err := db1.Query("SELECT user_id, subject,phone FROM users limit 11")
    rows1, err := db1.Query("Select body from users limit 11")
    defer rows.C...
}

解决方案建议 #

嘿，我来帮你梳理下问题！首先咱们先从最合理的优化方向入手——其实你完全没必要拆分两次查询同一个表的字段，这既低效又容易引发结果集不同步的问题。

方案1：合并SQL查询（推荐）

直接一次查询获取所有需要的字段，代码逻辑更简洁，也从根源上避免了两个结果集的同步问题：

func GetUsers() (users []Users, err error) {
    users = make([]Users, 0)
    // 一次查询拿到所有目标字段
    rows, err := db1.Query("SELECT user_id, subject, phone, body FROM users LIMIT 11")
    if err != nil {
        return nil, err
    }
    defer rows.Close() // 确保结果集资源被释放

    for rows.Next() {
        var u Users
        // 一次性扫描所有字段到结构体
        if err := rows.Scan(&u.UserID, &u.Subject, &u.Phone, &u.Body); err != nil {
            return nil, err
        }
        users = append(users, u)
    }
    // 检查遍历过程中是否出现错误
    if err := rows.Err(); err != nil {
        return nil, err
    }
    return users, nil
}

方案2：如果必须保留两个查询（不推荐）

如果你因为特殊需求必须拆分两次查询，需要注意以下几个核心问题，我帮你修正了代码：

func GetUsers() (users []Users, err error) {
    users = make([]Users, 0)
    // 第一个查询：单独处理错误，避免被后续覆盖
    rows, err := db1.Query("SELECT user_id, subject, phone FROM users LIMIT 11")
    if err != nil {
        return nil, err
    }
    // 安全释放资源：判断rows非nil再关闭
    defer func() {
        if rows != nil {
            rows.Close()
        }
    }()

    // 第二个查询：同样单独处理错误
    rows1, err := db1.Query("SELECT body FROM users LIMIT 11")
    if err != nil {
        return nil, err
    }
    defer func() {
        if rows1 != nil {
            rows1.Close()
        }
    }()

    // 同步遍历两个结果集：只要有一个遍历完就停止
    for rows.Next() && rows1.Next() {
        var u Users
        // 扫描第一个结果集的字段
        if err := rows.Scan(&u.UserID, &u.Subject, &u.Phone); err != nil {
            return nil, err
        }
        // 扫描第二个结果集的字段
        if err := rows1.Scan(&u.Body); err != nil {
            return nil, err
        }
        users = append(users, u)
    }

    // 检查两个结果集的遍历是否出现错误
    if err := rows.Err(); err != nil {
        return nil, err
    }
    if err := rows1.Err(); err != nil {
        return nil, err
    }

    return users, nil
}

你原来代码的核心问题 #

1. 错误覆盖：第二个rows1, err := ...会覆盖第一个查询的错误，导致你无法发现第一个查询的异常。
2. 资源未正确释放：defer rows.C...未完成，且没有处理两个结果集的关闭逻辑，容易造成资源泄漏。
3. 结果集不同步：没有同步遍历两个结果集，可能出现一个结果集已遍历完、另一个还有数据的情况，导致数据匹配错误。

内容的提问来源于stack exchange，提问作者Medone