使用 sqlx 将 PostgreSQL 数组映射到结构体

在使用 SQLx 将 PostgreSQL 数组映射到 Go 结构时，可能会遇到扫描错误。为了解决此问题，可以定义一个自定义类型来处理数组数据。通过实现 `sql.Scanner` 和 `driver.Valuer` 接口，自定义类型可以控制数据的扫描和存储过程，从而实现数组到结构的无缝映射。

我正在尝试在 go 中创建一个基本的评论 api。我似乎无法弄清楚如何将 postgresql 数组扫描到结构中的结构数组中。我想我可能可以将 thread.posts 类型设置为 jsonb 但这似乎不太优雅，因为我认为我必须对其进行解组。

sql：列索引 3 上的扫描错误，名称“posts”：不支持的扫描， 将 driver.value 类型 []uint8 存储到类型 *[]models.post 中

var threadSchema = `
CREATE TABLE IF NOT EXISTS thread (
  id         SERIAL PRIMARY KEY,
  name       VARCHAR(100) NOT NULL,
  profile_id INTEGER REFERENCES profile (id)
)`

var postSchema = `
CREATE TABLE IF NOT EXISTS post (
  id         SERIAL PRIMARY KEY,
  comment    TEXT,
  profile_id INTEGER REFERENCES profile (id),
  thread_id  INTEGER REFERENCES thread (id)
)`

type Post struct {
    Id        int    `db:"id" json:"id"`
    Comment   string `db:"comment" json:"comment" binding:"required" form:"comment"`
    ProfileId int   `db:"profile_id" json:"profile_id" binding:"required" form:"profile_id"`
    ThreadId  int    `db:"thread_id" json:"thread_id" binding:"required" form:"thread_id"`
}

type Thread struct {
    Id        int    `db:"id" json:"id"`
    Name      string `db:"name" json:"name" binding:"required" form:"name"`
    ProfileId int    `db:"profile_id" json:"profile_id" binding:"required" form:"profile_id"`
    Posts     []Post `db:"posts" json:"posts" form:"posts"`
}

func GetThreads(db *sqlx.DB, c *gin.Context) {
    threads := []Thread{}
    err := db.Select(&threads, `
    SELECT thread.id,thread.name,thread.profile_id,array_agg(post.id) AS posts
    FROM thread
    INNER JOIN post ON thread.id = post.thread_id
    GROUP BY thread.id;
  `)
    if err != nil {
        log.Fatal(err)
    }
    c.JSON(http.StatusOK, gin.H{"data": threads})
}

解决方案

您可以定义您的类型：

type Posts []Post

// Scan implements the sql.Scanner interface.
func (a *Posts) Scan(src interface{}) error {
  // ...
}

// Value implements the driver.Valuer interface.
func (a Posts) Value() (driver.Value, error) {
  // ...
}

有关实施的更多信息，请参阅例如 here

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~