快捷導航

Go語言實現(xiàn)管理多個數(shù)據(jù)庫連接

更新時間：2023年10月23日 11:15:19 作者：洛天楓

在軟件開發(fā)過程中,使用?MySQL、PostgreSQL?或其他數(shù)據(jù)庫是很常見的,由于配置和要求不同,管理這些連接可能具有挑戰(zhàn)性,下面就來和大家聊聊如何在Go中管理多個數(shù)據(jù)庫連接吧

簡介

作為軟件工程師，經(jīng)常會遇到應(yīng)用程序需要與多個數(shù)據(jù)庫協(xié)同工作的情況，而每個數(shù)據(jù)庫都有其獨特的要求和配置。在遵守最佳編碼實踐的同時，有效地處理這些不同的數(shù)據(jù)庫連接對于構(gòu)建健壯且易于維護的應(yīng)用程序至關(guān)重要。

挑戰(zhàn)：多個數(shù)據(jù)庫連接

在軟件開發(fā)過程中，使用 MySQL、PostgreSQL 或其他數(shù)據(jù)庫是很常見的。由于配置和要求不同，管理這些連接可能具有挑戰(zhàn)性。為了有效應(yīng)對這一挑戰(zhàn)，我們將提供一個分步指南和一個實用示例。

步驟 1：數(shù)據(jù)庫配置

首先定義一個 DBConfig 結(jié)構(gòu)，用于保存每個數(shù)據(jù)庫的配置詳細信息。這些配置包括數(shù)據(jù)庫類型、用戶憑證、主機、端口等關(guān)鍵參數(shù)。

// DBConfig 表示數(shù)據(jù)庫的配置。
type DBConfig struct {
 IdentificationName string // IdentificationName 用于獲取特定的數(shù)據(jù)庫連接。
 DB                 string   
 User               string   
 Password           string   `json:"_"` 
 Host               string   
 Port               string  
 Type               string   // Type of the database ("mysql", "postgres", "mssql", etc.).
 SSLMode            string   
 TimeZone           string   
 dialector          gorm.Dialector 
}

// Connect 根據(jù)提供的配置建立數(shù)據(jù)庫連接。
func (config *DBConfig) Connect() (DBConnection, error) {
 db, err := gorm.Open(config.dialector, &gorm.Config{})
 return db, err
}

步驟 2：數(shù)據(jù)庫連接接口

type DBConnection *gorm.DB

type DatabaseConnection interface {
    Connect() (DBConnection, error)
}

步驟 3：實施數(shù)據(jù)庫連接

實現(xiàn)兩種類型的數(shù)據(jù)庫連接：MySQL 和 PostgreSQL。每種連接類型都有自己的 Connect 方法，可根據(jù)提供的配置配置數(shù)據(jù)庫連接參數(shù)，并返回一個 GORM DB 實例。

// MySQLConnection 實現(xiàn)了 MySQL 的 DatabaseConnection。
type MySQLConnection struct {
 Config *DBConfig
}

// Connect 連接到 MySQL 數(shù)據(jù)庫，并返回一個 GORM DB 實例。
func (m *MySQLConnection) Connect() (DBConnection, error) {
 dsn := "%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=%s"
 m.Config.dialector = mysql.Open(fmt.Sprintf(dsn, m.Config.User, m.Config.Password, m.Config.Host, m.Config.Port, m.Config.DB, m.Config.TimeZone))
 db, err := m.Config.Connect()
 return db, err
}

// PostgresConnection 實現(xiàn)了 PostgreSQL 的 DatabaseConnection。
type PostgresConnection struct {
 Config *DBConfig
}

// Connect 連接 PostgreSQL 數(shù)據(jù)庫并返回 GORM DB 實例。
func (p *PostgresConnection) Connect() (DBConnection, error) {
 dsn := "host=%s user=%s password=%s dbname=%s port=%s sslmode=%s TimeZone=%s"
 p.Config.dialector = postgres.Open(fmt.Sprintf(dsn, p.Config.Host, p.Config.User, p.Config.Password, p.Config.DB, p.Config.Port, p.Config.SSLMode, p.Config.TimeZone))
 db, err := p.Config.Connect()
 return db, err
}

步驟 4：創(chuàng)建和管理數(shù)據(jù)庫連接

創(chuàng)建一個 NewConnection 函數(shù)，根據(jù)給定配置生成一個新的數(shù)據(jù)庫連接。該函數(shù)還能自動執(zhí)行數(shù)據(jù)庫遷移，以確保數(shù)據(jù)庫表結(jié)構(gòu)是最新的。

// NewConnection 根據(jù)給定的配置創(chuàng)建新的數(shù)據(jù)庫連接。
func (config *DBConfig) NewConnection() (DBConnection, error) {
 var dbConnection DatabaseConnection
 switch config.Type {
 case "mysql":
  dbConnection = &MySQLConnection{Config: config}
 case "postgres":
  dbConnection = &PostgresConnection{Config: config}
 default:
  return nil, fmt.Errorf("Unsupported database type: %s", config.Type)
 }

 // 創(chuàng)建新連接
 con, err := dbConnection.Connect()
 if err != nil {
  return nil, err
 }

 // 自動遷移表結(jié)構(gòu)
 err = con.Statement.AutoMigrate(&dto.User{})
 if err != nil {
  return nil, err
 }

 return con, nil
}

步驟 5：使用數(shù)據(jù)庫連接

在 main 函數(shù)中，初始化并使用數(shù)據(jù)庫連接。通過標識名訪問特定連接，并執(zhí)行必要的數(shù)據(jù)庫操作。

func init() {
 // 在程序啟動時初始化數(shù)據(jù)庫連接。
 configs.InitDBConnections()
}

func main() {
 users := []dto.User{
  {
   UserName: "user1",
   Password: "test1",
  },
  {
   UserName: "user2",
   Password: "test2",
  },
 }

 // 使用指定的連接名稱創(chuàng)建新的用戶資源庫
 repo := repo.NewUserRepo("TEST_POSTGRES_CON")

 err := repo.Save(users...)
 if err != nil {
  return
 }

 users, err = repo.FindAll()
 if err != nil {
  return
 }

 for _, user := range users {
  fmt.Printf("%+v\n", user)
 }

 // 設(shè)置偵聽操作系統(tǒng)信號的通道（例如 Ctrl+C）
 c := make(chan os.Signal, 1)
 signal.Notify(c, os.Interrupt, syscall.SIGTERM)
 // 等待信號（如 Ctrl+C），優(yōu)雅地退出程序
 <-c

 // 程序終止時關(guān)閉數(shù)據(jù)庫連接。
 defer configs.CloseDBConnections()
}