|
|
package GPUtil
|
|
|
|
|
|
import (
|
|
|
"dsDataex/Utils/CommonUtil"
|
|
|
"dsDataex/Utils/ConfigUtil"
|
|
|
"fmt"
|
|
|
_ "github.com/lib/pq"
|
|
|
"github.com/xormplus/core"
|
|
|
"github.com/xormplus/xorm"
|
|
|
"github.com/xormplus/xorm/log"
|
|
|
"os"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
var Engine *xorm.Engine
|
|
|
var ServerVersion string
|
|
|
|
|
|
/**
|
|
|
* @Author zhangjun
|
|
|
* @Description
|
|
|
* @Date 2021-02-06 10:00
|
|
|
* @Param
|
|
|
* @return
|
|
|
**/
|
|
|
func init() {
|
|
|
host := ConfigUtil.GreenPlumIp
|
|
|
port := ConfigUtil.GreenPlumPort
|
|
|
user := ConfigUtil.GreenPlumUser
|
|
|
dbname := ConfigUtil.GreenPlumDataBase
|
|
|
password := ConfigUtil.GreenPlumPwd
|
|
|
//mysql
|
|
|
master := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname)
|
|
|
|
|
|
var err error
|
|
|
Engine, err = xorm.NewEngine("postgres", master)
|
|
|
if err != nil {
|
|
|
fmt.Println(err)
|
|
|
}
|
|
|
ServerVersion = Engine.DriverName()
|
|
|
|
|
|
//设置数据库连接池
|
|
|
Engine.SetMaxOpenConns(100)
|
|
|
Engine.SetMaxIdleConns(10)
|
|
|
Engine.SetConnMaxLifetime(time.Second * 30)
|
|
|
|
|
|
Engine.Ping()
|
|
|
|
|
|
go func() {
|
|
|
for {
|
|
|
time.Sleep(time.Duration(600) * time.Second)
|
|
|
Engine.Ping()
|
|
|
fmt.Println("\n[xorm] [info] " + CommonUtil.GetCurrentTime() + " PING DATABASE greenplum")
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
//设置数据时区
|
|
|
var location *time.Location
|
|
|
location, err = time.LoadLocation("Asia/Shanghai")
|
|
|
Engine.TZLocation = location
|
|
|
//与 struct的映射方式，这里采用蛇型方法，默认是蛇形
|
|
|
Engine.SetTableMapper(core.SnakeMapper{})
|
|
|
|
|
|
//显示+记录SQL日志
|
|
|
f, _ := os.Create("./Logs/sql.log")
|
|
|
Engine.SetLogger(log.NewSimpleLogger(f))
|
|
|
|
|
|
Engine.ShowSQL(true) // 则会在控制台打印出生成的SQL语句
|
|
|
Engine.Logger().SetLevel(log.LOG_DEBUG) //则会在控制台打印info及以上的信息
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* @Author zhangjun
|
|
|
* @Description
|
|
|
* @Date 2021-02-06 10:00
|
|
|
* @Param
|
|
|
* @return
|
|
|
**/
|
|
|
func SqlQuery(sql string, param []string) (bool, []map[string]interface{}) {
|
|
|
|
|
|
result, err := Engine.SQL(sql, param).QueryInterface()
|
|
|
|
|
|
if err != nil {
|
|
|
return false, result
|
|
|
} else {
|
|
|
return true, result
|
|
|
}
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* @Author zhangjun
|
|
|
* @Description
|
|
|
* @Date 2021-02-06 11:11
|
|
|
* @Param
|
|
|
* @return
|
|
|
**/
|
|
|
func SqlQueryJson(sql string, param []string) (bool, string) {
|
|
|
|
|
|
result, err := Engine.SQL(sql, param).Query().Json()
|
|
|
|
|
|
if err != nil {
|
|
|
return false, err.Error()
|
|
|
} else {
|
|
|
return true, result
|
|
|
}
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* @Author zhangjun
|
|
|
* @Description
|
|
|
* @Date 2021-02-06 11:16
|
|
|
* @Param
|
|
|
* @return
|
|
|
**/
|
|
|
func SqlQueryXml(sql string, param []string) (bool, string) {
|
|
|
|
|
|
result, err := Engine.SQL(sql, param).Query().Xml()
|
|
|
|
|
|
if err != nil {
|
|
|
return false, err.Error()
|
|
|
} else {
|
|
|
return true, result
|
|
|
}
|
|
|
}
|
