|
|
package mydb
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
//"gorm.io/driver/postgres"
|
|
|
|
|
|
"gorm.io/driver/postgres"
|
|
|
|
|
|
"gorm.io/gorm"
|
|
|
"log"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
func init() {
|
|
|
fmt.Println("db init()")
|
|
|
initDatabase()
|
|
|
}
|
|
|
|
|
|
var DB *gorm.DB
|
|
|
|
|
|
func initDatabase() {
|
|
|
|
|
|
//mysql
|
|
|
//dsn := "root:Skyinno251,@tcp(192.168.2.25: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
|
|
|
//3.30: 10.19.1.219 3.29:10.19.1.154 localhost
|
|
|
//dsn := "host=localhost user=postgres password=452131wW dbname=postgres port=5432 sslmode=disable"
|
|
|
dsn := "host=192.168.2.25 user=postgres password=452131wW dbname=weather port=5432 sslmode=disable"
|
|
|
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
|
|
|
}
|