使用 GORM 在 Postgres 中创建数据库

一分耕耘，一分收获！既然都打开这篇《使用 GORM 在 Postgres 中创建数据库》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

这主要侧重于为我计划编写的涉及创建数据库的测试套件提供 setup() 和teardown() 方法。

我已经弄清楚如何使用 gorm 创建数据库。但是，我不确定这是否是最好的方法。

package main

import (
    "fmt"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/postgres"
    "log"
)

func main() {
    db, err := gorm.Open("postgres", "host=127.0.0.1 port=5432 user=superuser dbname=postgres password='' sslmode=disable")
    capture(err)
    db = db.Exec("CREATE DATABASE test_db;")
    if db.Error != nil {
        fmt.Println("Unable to create DB test_db, attempting to connect assuming it exists...")
        db, err = gorm.Open("postgres", "host=127.0.0.1 port=5432 user=superuser dbname=test_db password='' sslmode=disable")
        if err != nil {
           fmt.Println("Unable to connect to test_db")
           capture(err)
        }
    }
    defer db.Close()
}

func capture(err error) {
    if err != nil {
        log.Fatalf("%s", err)
   }
}

我首先连接到默认的 postgres 数据库，然后创建我计划使用的第二个测试数据库。

这是最好的方法吗？或者有没有一种方法可以在没有预先存在的数据库的情况下连接到 postgres。

注意：我已经查找过人们使用 sql 驱动程序仅使用连接字符串 user:password@/ 连接到数据库的答案。这对我来说不起作用。（就像这里）

我还尝试了没有数据库名称的连接字符串，这导致驱动程序尝试连接到与用户同名的数据库。由于这样的数据库不存在，因此失败。

解决方案

这里是我如何使用gorm实现创建postgresql数据库，关键是仅连接到postgresql，创建数据库不需要连接到“数据库”，只需连接到数据库引擎就足够了。只是不要在连接字符串中传递数据库基数。

在main.go中

package main

import (
    "gorm.io/driver/postgres"
    "gorm.io/gorm"
    "fmt"
    "github.com/joho/godotenv"
)

func createDatabase() {
    dsn := fmt.Sprintf("host=%s port=%s user=%s password=%s sslmode=disable TimeZone=%s", Config("DB_HOST"), Config("DB_PORT"), Config("DB_USER"), Config("DB_PASSWORD"), Config("DB_TIMEZONE"))
    DB, _ := gorm.Open(postgres.Open(dsn), &gorm.Config{})

    createDatabaseCommand := fmt.Sprintf("CREATE DATABASE %s", Config("DB_NAME"))
    DB.Exec(createDatabaseCommand)
}
 
func main() {
    createDatabase()
}

现在只需运行 go get -d ./... && go run main.go

以上就是《使用 GORM 在 Postgres 中创建数据库》的详细内容，更多关于的资料请关注golang学习网公众号！