postgresql - sqlx 将 postgres 数组扫描到结构中

Question

我正在尝试在 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})
}

score 3 · Accepted Answer

你可以定义你的类型：

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) {
  // ...
}

有关实施的更多信息，请参见此处

score 0 · Accepted Answer

首先，无论您是否使用 Postgres 数组，都无法使用 sqlx 执行此操作。

其次，您的 SQL 查询只是聚合帖子 ID，而不是帖子的内容，因此无法获取您想要的数据（使用 Go 或其他方式）。

因此，您可以执行以下操作：

使用匿名嵌入式结构，捕获 SQL 查询中的所有 Post 内容，然后合并重复的线程。

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 ThreadDb 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"`
    Post
}

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 := []ThreadDb{}
    err := db.Select(&threads, `
    SELECT thread.id,thread.name,thread.profile_id,post.id,post.comment,post.profile_id,post.thread_id
    FROM thread
    INNER JOIN post ON thread.id = post.thread_id
    GROUP BY post.id;
  `)

    thread_map := make(map[string]Thread)

    for i, thread := range threads {
        if _, ok := thread_map[thread.Id]; ok {
            thread_map[thread.Id].Posts = append(thread_map[thread.Id].Posts, thread.Post)
        } else {
            thread_map[thread.Id] = Thread{thread.Id, thread.Name, thread.ProfileId, []Post{thread.Post}}
        }
    }

    var threadSlice []string

    for k := range thread_map {
        threadSlice = append(threadSlice, k)
    }

    if err != nil {
        log.Fatal(err)
    }
    c.JSON(http.StatusOK, gin.H{"data": threadSlice})
}

使用GROUP_CONCAT或类似。除非您计划每个线程最多发布大约 100 个帖子，否则我不会推荐。

postgresql - sqlx 将 postgres 数组扫描到结构中

2 回答 2

Related

Reference