登录
首页 >  Golang >  Go问答

从两个不同的结构中提取信息

来源:stackoverflow

时间:2024-02-22 14:09:25 170浏览 收藏

最近发现不少小伙伴都对Golang很感兴趣,所以今天继续给大家介绍Golang相关的知识,本文《从两个不同的结构中提取信息》主要内容涉及到等等知识点,希望能帮到你!当然如果阅读本文时存在不同想法,可以在评论中表达,但是请勿使用过激的措辞~

问题内容

有了这个 user 和 post 结构,当用户创建新帖子时,我尝试将 user 的名称包含在 post 结构中。

type user struct {
    id             int
    name           string
    created        time.time
}

type post struct {
    id int
    posttitle string
    postdesc string
    created time.time
}

如何在这两个结构之间创建一些连接,例如帖子的作者?

目标是尝试使用以下代码从 user 结构中获取帖子作者的姓名:

post, err := app.Models.Posts.GetPost(id)

getpost() 只是运行 select 查询并扫描行


正确答案


这种方法没有任何 orm。
这是一个可以返回多行的简单查询。您必须扫描整个结果集并将每一列映射到结构的字段上。

请记住始终检查错误。

您可以在下面找到解决方案:

package main

import (
    "database/sql"
    "fmt"
    "time"

    _ "github.com/lib/pq"
)

type post struct {
    id        int
    posttitle string
    postdesc  string
    created   time.time
    userid    int
    user      user
}

type user struct {
    id      int
    name    string
    created time.time
}

func main() {
    conn, err := sql.open("postgres", "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable")
    if err != nil {
        panic(err)
    }
    defer conn.close()

    // get max id
    var id int
    conn.queryrow(`select max(id) from posts`).scan(&id)

    // insert
    sqlinsertstmt := `insert into posts (id, post_title, post_desc, created, user_id) values ($1,$2,$3,$4,$5)`
    if _, err = conn.exec(sqlinsertstmt, id+1, "tdd", "introduction to test-driven-development", time.now(), 1); err != nil {
        panic(err)
    }

    // read
    rows, err := conn.query(`select posts.id, post_title, post_desc, posts.created, users.id, users.name, users.created from posts inner join users on posts.user_id=users.id`)
    if err != nil {
        panic(err)
    }

    var posts []post
    for rows.next() {
        var post post
        if err = rows.scan(&post.id, &post.posttitle, &post.postdesc, &post.created, &post.user.id, &post.user.name, &post.user.created); err != nil {
            panic(err)
        }
        posts = append(posts, post)
    }

    if err = rows.err(); err != nil {
        panic(err)
    }

    for _, v := range posts {
        fmt.printf("author name: %q\n", v.user.name)
    }
}

请告诉我这是否有帮助。

编辑

我还在 postgres 中提供了一个 insert 示例。为了实现它,我们必须使用 db.exec() 函数,并提供参数。

请注意如何构建查询,因为您可能会遇到 sql 注入漏洞。

最后,在现实场景中,您不应该在 posts 表中查找 max id,而应该自动生成。

尝试一下此解决方案,也许它可以解决您的问题问题。

首先,我以这种方式定义结构:

// "post" belongs to "user", "userid" is the foreign key
type post struct {
    gorm.model
    id        int
    posttitle string
    postdesc  string
    created   time.time
    userid    int
    user      user
}

type user struct {
    id      int
    name    string
    created time.time
}

这样,您就可以说 post 属于 user 并在 post 结构体中访问用户的信息。
要查询记录,您必须使用 preload("user") 以确保从单独的表中立即加载用户记录。

最后,您将能够访问嵌入结构中的数据(使用 dot 表示法)。
下面,您可以找到一个完整的工作示例(使用 docker 实现):

package main

import (
    "fmt"
    "time"

    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

// "Post" belongs to "User", "UserID" is the foreign key
type Post struct {
    gorm.Model
    ID        int
    PostTitle string
    PostDesc  string
    Created   time.Time
    UserID    int
    User      User
}

type User struct {
    ID      int
    Name    string
    Created time.Time
}

func main() {
    dsn := "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable"
    db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(err)
    }

    db.AutoMigrate(&Post{})

    newPost := &Post{ID: 1, PostTitle: "Golang", PostDesc: "Introduction to Golang", Created: time.Now(), UserID: 1, User: User{ID: 1, Name: "John Doe", Created: time.Now()}}
    db.Create(newPost)

    var post Post
    db.Preload("User").Find(&post, 1)
    fmt.Printf("author name: %q\n", post.User.Name)
}

如果我回答了您的问题,请告诉我!

今天关于《从两个不同的结构中提取信息》的内容就介绍到这里了,是不是学起来一目了然!想要了解更多关于的内容请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>