FranklinWH-Toolbox/service/database/pgsql.go

package database

import (
	"log"

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

var PgSQL *gorm.DB

func InitPostgreSQL(dsn string) (*gorm.DB, error) {
	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	})
	if err != nil {
		log.Printf("failed to connect to PostgreSQL: %v", err)
		return nil, err
	}
	PgSQL = db
	// 自动迁移
	// db.AutoMigrate(&models.User{}, …)
	return db, nil
}