go_mqtt/mydb/initialize.go

package mydb

import (
	"fmt"
	//"gorm.io/driver/mysql"
	//"gorm.io/driver/postgres"
	//"gorm.io/driver/sqlserver"
	"github.com/glebarez/sqlite"
	"gorm.io/gorm"
	"log"
	"time"
)

func init() {
	fmt.Println("sqlite init()")
	initDatabase()
}

var DB *gorm.DB

func initDatabase() {

	//mysql
	//dsn := "root:Skyinno251,@tcp(192.168.2.18:3306)/gorm?charset=utf8mb4&parseTime=True&loc=Local"
	//dsn := "root:Skyinno251,@tcp(47.242.184.139:3306)/appserver?charset=utf8mb4&parseTime=True&loc=Local"
	//dsn := "root:Skyinno251,@tcp(localhost:3306)/appserver?charset=utf8mb4&parseTime=True&loc=Local"
	//dsn := "root:Skyinno251,@tcp(192.168.3.9:3306)/gorm?charset=utf8mb4&parseTime=True&loc=Local"
	//dsn := "root:Skyinno251,@tcp(localhost:3306)/go_mqtt?charset=utf8mb4&parseTime=True&loc=Local"
	//
	//db, err := gorm.Open(mysql.New(mysql.Config{
	//	DSN:                       dsn,   // DSN data source name
	//	DefaultStringSize:         255,   // string 类型字段的默认长度
	//	DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
	//	DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
	//	DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
	//	SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置
	//}), &gorm.Config{})
	//db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})

	//sqlite
	db, err := gorm.Open(sqlite.Open("/media/cat/sd/home/cat/go_mqtt/go_mqtt.db"), &gorm.Config{})
	//db, err := gorm.Open(sqlite.Open("go_mqtt.db"), &gorm.Config{})

	//postgres
	//dsn := "host=192.168.1.22 user=postgres password=452131wW dbname=postgres port=5432 sslmode=disable TimeZone=Asia/Shanghai"
	//db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})

	//sqlserver
	//dsn := "sqlserver://sa:452131wW@192.168.3.29:1433?database=go_mqtt"
	//db, err := gorm.Open(sqlserver.Open(dsn), &gorm.Config{})

	if err != nil {
		fmt.Println("failed to connect database")
	}

	// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
	sqlDB, errDB := db.DB()
	if errDB != nil {
		return
	}

	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(5)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(20)
	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Minute * 5) // 连接最大生命周期

	err = sqlDB.Ping() // 检查连接是否成功建立。如果失败，则返回错误。
	if err != nil {
		// 处理连接失败的情况。例如，打印错误信息并退出程序。
		log.Println("数据库连接失败")
		fmt.Println("Connection failed:", err)
		return
	}
	DB = db
}