卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34116本站已运行391

表示 SQL 表的 Go 结构体

表示 sql 表的 go 结构体

问题内容

我对 go 还很陌生,我正在尝试找到设置数据库通信的最佳方法。基本上,我记得在我以前的工作场所中,在 php 中,您可以创建一个表示 sql 表的类,当您需要将数据插入数据库时​​,您将创建该类的一个对象,其中包含所有必要的数据,调用 insert(),传递update() 的工作方式非常相似,只不过它会更新而不是插入。不幸的是,我不记得那个 php 框架的名称,但也许有人知道在 go 中实现类似功能的方法,或者它不是一个东西?

假设我有一个结构:

type patients struct {
    id              int
    name            string 
    image           string    
}

现在我想要一个函数,将患者对象作为参数并将其插入到患者 postgres 表中,自动将患者转换为 postgres 期望的内容:

func (patients *patients) insert(patient patients) {

}

然后 update() 将获取一个患者对象并基本上执行这段代码,而无需我编写它:

stmt := `update patients set
    name = $1,
    image = $2,
    where id = $3
`

_, err := db.ExecContext(ctx, stmt,
    patient.Name,
    patient.Image,
    patient.ID
)


正确答案


您正在寻找一种称为 orm(对象关系映射器)的东西。 go 中有一些,但最流行的是 gorm。这是一个有点争议的话题,但我认为如果您不熟悉 go 和/或数据库,那么使用 orm 是个好主意。这将为您节省大量时间和精力。

另一种方法是使用database/sql 包并编写您自己的sql 查询。如果您是经验丰富的 go 开发人员和/或数据库管理员,这是一个好主意。它将让您更好地控制查询并且效率更高。推荐阅读:https://www.php.cn/link/a12f69495f41bb3b637ba1b6238884d6 。此方法的推荐库包括 sqlx 和 pgx。

这是您的结构作为 gorm 模型的样子:

type patient struct {
    id              int `gorm:"primarykey"`
    name            string 
    image           string    
}

这是一个如何将患者插入数据库的示例程序:

package main

import (
    "fmt"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

type patient struct {
    id              int `gorm:"primarykey"`
    name            string 
    image           string    
}

func main() {
    dsn := "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable timezone=utc"
    db, err := gorm.open(postgres.open(dsn), &gorm.config{})
    if err != nil {
        panic("failed to connect database")
    }

    db.automigrate(&patient{})

    patient := patient{
        name: "john smith",
        image: "https://example.com/image.png",
    }

    result := db.create(&patient)
    if result.error != nil {
        panic(result.error)
    }

    fmt.println(patient)
}

如果您想使用 sqlx,您可以编写如下内容:

package main

import (
    "database/sql"
    "fmt"
    "log"

    _ "github.com/lib/pq"
)

type patient struct {
    id              int
    name            string 
    image           string    
}

func main() {
    dsn := "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable timezone=utc"
    db, err := sql.open("postgres", dsn)
    if err != nil {
        log.fatal(err)
    }
    defer db.close()

    _, err = db.exec(`
        create table if not exists patients (
            id serial primary key,
            name text,
            image text
        )
    `)
    if err != nil {
        log.fatal(err)
    }

    patient := patient{
        name: "john smith",
        image: "https://example.com/image.png",
    }

    _, err = db.exec(`
        insert into patients (name, image) values ($1, $2)
    `, patient.name, patient.image)
    if err != nil {
        log.fatal(err)
    }

    fmt.println(patient)
}

当然,使用 orm 管理数据库模式会稍微复杂一些。您可以使用迁移,但我更喜欢使用名为 goose 的工具。设置起来有点麻烦,但它非常强大且灵活。以下是如何使用它的示例:

package main

import (
    "fmt"
    "log"

    "github.com/pressly/goose"
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

type patient struct {
    id              int `gorm:"primarykey"`
    name            string 
    image           string    
}

func main() {
    dsn := "host=localhost user=postgres password=postgres dbname=postgres port=5432 sslmode=disable timezone=utc"
    db, err := gorm.open(postgres.open(dsn), &gorm.config{})
    if err != nil {
        panic("failed to connect database")
    }

    goose.setdialect("postgres")
    goose.settablename("schema_migrations")

    err = goose.run("up", db.db(), "migrations")
    if err != nil {
        log.fatal(err)
    }

    patient := patient{
        name: "john smith",
        image: "https://example.com/image.png",
    }

    result := db.create(&patient)
    if result.error != nil {
        panic(result.error)
    }

    fmt.println(patient)
}

您的迁移目录如下所示:

migrations/
    00001_create_patients.up.sql
    00001_create_patients.down.sql

您的迁移如下所示:

-- 00001_create_patients.up.sql
create table patients (
    id serial primary key,
    name text,
    image text
);
-- 00001_create_patients.down.sql
DROP TABLE patients;

我希望这有帮助!如果您有任何疑问,请告诉我。

卓越飞翔博客
上一篇: 在 Golang 中解码 asn1.RawValue 的正确方法
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏