在关系型数据库中进行数据映射到嵌套切片结构

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《在关系型数据库中进行数据映射到嵌套切片结构》，聊聊，我们一起来看看吧！

从 sql 数据库查询数据并将其映射到包含另一种结构类型切片的结构的好方法是什么？

就我而言，我使用 postgres sql、pgx 来查询数据，并使用 scany 将行结果映射到对象中。

我有一个帖子表和评论表。关系是一个帖子可以有很多评论。

这些是我代表数据库表的模型

package models

type post struct {
    id int64
    // more fields

    comments []comment
}

type comment struct {
    id     int64
    postid int64
    // more fields

    post post
}

这是我的 postrepository 对象

package repository

import (
    "context"
    "fmt"

    "github.com/georgysavva/scany/pgxscan"
    "github.com/jackc/pgx/v4/pgxpool"
    "kosev.com/databases/models"
)

type PostRepository struct {
    dbPool *pgxpool.Pool
}

func NewPostRepository(dbPool *pgxpool.Pool) *PostRepository {
    return &PostRepository{
        dbPool: dbPool,
    }
}

func (pr *PostRepository) GetAll() []models.Post {
    query := `SELECT * FROM posts
              LEFT OUTER JOIN "comments"
              ON "comments".post_id  = posts.id;`

    var posts []models.Post

    rows, _ := pr.dbPool.Query(context.Background(), query)

    err := pgxscan.ScanAll(&posts, rows)

    if err != nil {
        fmt.Println("Error: ", err.Error())
    }

    return posts
}

我想要实现的是获取所有 posts 并在每个帖子中为该 post 提供相应的 comments 切片

现在我知道我可以

• 查询所有帖子
• 循环浏览帖子并构建一个查询，获取每个帖子的评论
• 将适当的评论添加到相应帖子的片段中

我想知道是否有更好的解决方案，该解决方案只需要我执行查询和一些足够智能的 scan() 方法，可以在帖子中添加评论。

正确答案

我尝试使用postgres的json容量，对于简单的数据库来说可能没问题，但我不确定更复杂的查询的性能

1. docker-compose

version: '3'
services:

  db:
    image: postgres:10.16
    depends_on:
      - adminer  
    environment:
      postgres_host_auth_method: trust
      postgres_user: user
      postgres_password: password
      postgres_db: test
    volumes:
      - ./volume:/var/lib/mysql
      - ./sql/:/docker-entrypoint-initdb.d/
    ports:
      - 5432:5432

  adminer:
    image: adminer:latest
    environment:
      adminer_default_server: localhost
    ports:
      - 8080:8080

1. 文件夹sql来初始化数据库

create table if not exists posts (
  id serial primary key,
  title varchar(20)
);

create table if not exists comments (
  id serial primary key,
  post_id integer,
  txt text
);

insert into posts (title)
values ('first'),('second');

insert into comments (post_id, txt)
values (1,'cool'), (1,'notcool'), (2,'third');

1. 在管理员 (localhost:8080) 上您可以尝试此查询

select p.id, p.title, json_agg(c) as comments
    from posts p left join comments c on c.post_id = p.id
    group by p.id;

您将获得 json 格式的嵌套注释

1   first   [{"id":1,"post_id":1,"txt":"cool"}, {"id":2,"post_id":1,"txt":"notcool"}]
2   second  [{"id":3,"post_id":2,"txt":"third"}]

现在可以扫描它的嵌套结构字段

package main

import (
    "context"
    "fmt"
    "os"

    "github.com/jackc/pgx/v5"
)

type Post struct {
    Id       int64
    Title    string
    Comments []Comment
}

type Comment struct {
    Id  int64
    Txt string
}

func main() {
    urlExample := "postgres://user:password@localhost:5432/test"
    conn, err := pgx.Connect(context.Background(), urlExample)
    if err != nil {
        fmt.Fprintf(os.Stderr, "Unable to connect to database: %v\n", err)
        os.Exit(1)
    }
    defer conn.Close(context.Background())

    rows, err := conn.Query(context.Background(),
        `SELECT p.id, p.title,  
    json_agg(c) as comments
        FROM posts p
        LEFT JOIN comments c ON c.post_id = p.id
        GROUP BY p.id;`)

    defer rows.Close()
    for rows.Next() {
        var post Post
        err := rows.Scan(&post.Id, &post.Title, &post.Comments)
        if err != nil {
            panic(fmt.Sprintf("%v", err))
        }
        fmt.Println(post)
    }
}

参考号：example page for query postgres postgres building json example

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。