73 lines
1.1 KiB
Go
73 lines
1.1 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"git.trj.tw/golang/mtgbot/modules/config"
|
|
"github.com/jmoiron/sqlx"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
var x *sqlx.DB
|
|
|
|
// NewDB -
|
|
func NewDB() *sqlx.DB {
|
|
var err error
|
|
conf := config.GetConf()
|
|
userPassStr := conf.Database.User
|
|
if len(conf.Database.Pass) > 0 {
|
|
userPassStr += ":" + conf.Database.Pass
|
|
}
|
|
|
|
connStr := fmt.Sprintf("postgres://%s@%s/%s?sslmode=disable", userPassStr, conf.Database.Host, conf.Database.DB)
|
|
|
|
x, err = sqlx.Connect("postgres", connStr)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
x.SetMaxIdleConns(10)
|
|
x.SetMaxOpenConns(100)
|
|
err = x.Ping()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return x
|
|
}
|
|
|
|
// String - get string ptr
|
|
func String(s string) *string {
|
|
return &s
|
|
}
|
|
|
|
// Int - get int ptr
|
|
func Int(i int) *int {
|
|
return &i
|
|
}
|
|
|
|
// OrderStr -
|
|
type OrderStr string
|
|
|
|
func (o OrderStr) String() string {
|
|
return string(o)
|
|
}
|
|
|
|
// OpStr -
|
|
type OpStr string
|
|
|
|
func (s OpStr) String() string {
|
|
return string(s)
|
|
}
|
|
|
|
// Op var
|
|
const (
|
|
SearchOpEq OpStr = "="
|
|
SearchOpNeq = "!="
|
|
SearchOpLt = "<"
|
|
SearchOpLte = "<="
|
|
SearchOpGt = ">"
|
|
SearchOpGte = ">="
|
|
)
|