Golang sqlx多对多关联查询结果转数组报错解决方案咨询

问题原因 #

PostgreSQL的array_agg(ROW(...))返回的是数据库特有的行数组类型，Go标准库和sqlx默认无法直接将其映射到[]UnitImages这类结构体切片，驱动会把它当成原始字节数组（[]uint8）返回，最终导致扫描失败。

方法一：自定义扫描器（类型安全，推荐生产环境） #

给结构体切片实现sql.Scanner接口，手动解析PostgreSQL行数组的字节内容：

1. 为[]UnitImages定义别名类型，方便实现接口：

type UnitImagesSlice []UnitImages

2. 实现Scan方法解析行数组格式：

import (
    "database/sql/driver"
    "fmt"
    "strings"
    "strconv"
)

func (u *UnitImagesSlice) Scan(value interface{}) error {
    // 转换数据库返回的原始字节为字符串
    bytes, ok := value.([]uint8)
    if !ok {
        return fmt.Errorf("expected []uint8, got %T", value)
    }
    str := string(bytes)

    // 处理空数组情况
    str = strings.Trim(str, "{}")
    if str == "" {
        *u = []UnitImages{}
        return nil
    }

    // 拆分每一行数据
    rows := strings.Split(str, "),(")
    for i := range rows {
        rows[i] = strings.Trim(rows[i], "()")
    }

    // 解析每一行到UnitImages结构体
    var images []UnitImages
    for _, row := range rows {
        fields := strings.Split(row, ",")
        if len(fields) != 3 {
            return fmt.Errorf("invalid row format: %s", row)
        }

        // 注意：字段顺序要和SQL中ROW()的顺序完全匹配
        id := parseInt(fields[0])
        image := &fields[1]
        isMain := parseBool(fields[2])

        images = append(images, UnitImages{
            ID:     id,
            Image:  image,
            IsMain: isMain,
        })
    }

    *u = images
    return nil
}

// 辅助函数：字符串转*int
func parseInt(s string) *int {
    num, err := strconv.Atoi(s)
    if err != nil {
        return nil
    }
    return &num
}

// 辅助函数：字符串转*bool（适配PostgreSQL的t/f格式）
func parseBool(s string) *bool {
    b := s == "t"
    return &b
}

3. 更新Unit结构体的字段类型：

type Unit struct {
    ID                     *int             `json:"id" db:"id"`
    Name                   *string          `json:"name" db:"name"`
    ... 更多字段
    UnitImages             UnitImagesSlice  `json:"unit_images" db:"unit_images"`
}

完成后sqlx会自动调用自定义的Scan方法完成解析。

方法二：SQL转JSON格式（实现简单，快速上手） #

修改SQL语句，用json_agg直接生成JSON数组，避免原生行数组的解析问题：

SELECT 
    un.*, 
    json_agg(ROW(ui.id, ui.image, ui.is_main)) AS unit_images
FROM unit un
INNER JOIN unitimage ui ON ui.unit_id = un.id
GROUP BY un.id;

调整Unit结构体的字段标签，让sqlx识别JSON类型：

type Unit struct {
    ID                     *int         `json:"id" db:"id"`
    Name                   *string      `json:"name" db:"name"`
    ... 更多字段
    UnitImages             []UnitImages `json:"unit_images" db:"unit_images" sql:"type:json"`
}

sqlx会自动将JSON字符串解析为结构体切片，无需额外代码。

注意事项 #

• 确保SQL中ROW()的字段顺序和UnitImages结构体的字段顺序完全一致，否则会出现字段值错位。
• 如果允许Unit没有关联的UnitImage，将INNER JOIN改为LEFT JOIN，同时处理unit_images为NULL的情况（自定义扫描器中需判断value是否为nil）。

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