You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1022 B
Go
47 lines
1022 B
Go
package dbsql
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var DSN = ""
|
|
|
|
var DSN_local = ""
|
|
|
|
// 初始化数据库
|
|
func GetConn(dsn string) (*gorm.DB, error) {
|
|
// D_airecommen:NbI40G9iTK@tcp(test-nimodatatag.cluster-cxxsbkavtbvo.ap-southeast-1.rds.amazonaws.com:3306)/labeldb?charset=utf8&parseTime=true
|
|
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
}
|
|
return db, nil
|
|
}
|
|
|
|
func Close(db *gorm.DB) {
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
return
|
|
}
|
|
sqlDB.Close()
|
|
}
|
|
|
|
func SetPageQuery(db *gorm.DB, pageIndex int, pageSize int) *gorm.DB {
|
|
return db.Limit(pageSize).Offset((pageIndex - 1) * pageSize)
|
|
}
|
|
|
|
func SetDSN(pwd string) {
|
|
DSN_local = "root:" + pwd + "@tcp(qa.flatincbr.work:3306)/qa?charset=utf8&parseTime=true"
|
|
}
|
|
|
|
func DSN_qaoms() string {
|
|
str1 := strings.Split(DSN, ")/")[0] + ")/"
|
|
str2 := "?" + strings.Split(DSN, "?")[1]
|
|
return str1 + "qaoms" + str2
|
|
}
|