Go - Run a sql file on PostgreSQL using pgx

发布时间 2023-12-10 22:20:00作者: ZhangZhihuiAAA

 

package main

import (
    "context"
    "log"
    "os"

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

const DB_SOURCE = "postgresql://root:aaa@localhost:5432/zimple_bank?sslmode=disable"
var connPool *pgxpool.Pool
var ctx = context.Background()

func init() {
    var err error
    connPool, err = pgxpool.New(ctx, DB_SOURCE)
    if err != nil {
        log.Fatal("cannot connect to db:", err)
    }
}

func main() {
    data, _ := os.ReadFile("db/schema/0000_init_schema.up.sql")

    connPool.Exec(ctx, string(data))
}