|
|
package DAL
|
|
|
|
|
|
import (
|
|
|
"database/sql"
|
|
|
"flx/Common"
|
|
|
"fmt"
|
|
|
"net/url"
|
|
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
_ "kingbase.com/gokb"
|
|
|
)
|
|
|
|
|
|
var cdb *sql.DB
|
|
|
|
|
|
// /connect the database
|
|
|
func DBConnect() (*sql.DB, error) {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("DBConnect:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
cfig := Common.LoadConfig()
|
|
|
|
|
|
// connStr := "host=" + cfig.Kingbase.IP + " user=" + cfig.Kingbase.Username + " password=" + cfig.Kingbase.Password + " dbname=" + cfig.Kingbase.Name + " sslmode=disable"
|
|
|
password := url.QueryEscape(cfig.Kingbase.Password)
|
|
|
connStr := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local&tls=skip-verify",
|
|
|
cfig.Kingbase.Username,
|
|
|
password,
|
|
|
cfig.Kingbase.IP,
|
|
|
3306, // 确保这是数值类型（如 3306）
|
|
|
cfig.Kingbase.Name,
|
|
|
)
|
|
|
|
|
|
db, err := sql.Open("mysql", connStr)
|
|
|
if err != nil {
|
|
|
fmt.Println(err.Error())
|
|
|
return nil, err
|
|
|
}
|
|
|
err = db.Ping()
|
|
|
if err != nil {
|
|
|
fmt.Println(err.Error())
|
|
|
return nil, err
|
|
|
}
|
|
|
return db, nil
|
|
|
}
|
|
|
|
|
|
func ConntectDB() *sql.DB {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("ConntectDB:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
if cdb == nil {
|
|
|
cdb, _ = DBConnect()
|
|
|
}
|
|
|
return cdb
|
|
|
}
|
|
|
|
|
|
// close the database connect
|
|
|
func CloseDB() bool {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("CloseDB:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
if cdb != nil {
|
|
|
err := cdb.Close()
|
|
|
if err != nil {
|
|
|
fmt.Println(err.Error())
|
|
|
return false
|
|
|
}
|
|
|
return true
|
|
|
}
|
|
|
return false
|
|
|
}
|
|
|
|
|
|
func Text() {
|
|
|
|
|
|
defer func() {
|
|
|
if r := recover(); r != nil {
|
|
|
log.Error("Text:", r)
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// connStr := "host=114.64.231.197 user=SYSTEM password=system dbname=dbcongress sslmode=disable"
|
|
|
|
|
|
connStr := "host=127.0.0.1 user=SYSTEM password=system dbname=dbcongress sslmode=disable"
|
|
|
db, err := sql.Open("kingbase", connStr)
|
|
|
if err != nil {
|
|
|
fmt.Println(err.Error())
|
|
|
}
|
|
|
err = db.Ping()
|
|
|
if err != nil {
|
|
|
fmt.Println(err.Error())
|
|
|
}
|
|
|
rows, err := db.Query("SELECT \"St_ID\",\"St_Name\" FROM dbcongress.t_staffinfo")
|
|
|
if err != nil {
|
|
|
fmt.Println(err.Error())
|
|
|
}
|
|
|
defer rows.Close()
|
|
|
|
|
|
var St_ID, St_Name string
|
|
|
for rows.Next() {
|
|
|
err := rows.Scan(&St_ID, &St_Name)
|
|
|
if err != nil {
|
|
|
break
|
|
|
}
|
|
|
fmt.Println(St_ID, St_Name)
|
|
|
}
|
|
|
|
|
|
db.Close()
|
|
|
}
|
