42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
type Config struct {
|
||
Host string `toml:"host"` //数据库名称
|
||
Port int `toml:"port"` //数据库名称
|
||
Debug bool `toml:"debug"` //调试开关（会在日志打印SQL)
|
||
Dbname string `toml:"dbname"` //数据库名称
|
||
Username string `toml:"username"` //数据库用户名
|
||
Password string `toml:"password"` //数据库连接密码
|
||
ExtraParameters string `toml:"extra_parameters"` //数据库连接扩展参数
|
||
}
|
||
|
||
type Model struct {
|
||
DB *gorm.DB
|
||
DbTxStatus DbTxStatus
|
||
Redis redis.Conn
|
||
RedisPool *redis.Pool
|
||
}
|
||
|
||
func NewModel() *Model {
|
||
return NewModelDefault()
|
||
}
|
||
|
||
type DbTxStatus int
|
||
|
||
const (
|
||
DBTX_STATUS_TX_NONE DbTxStatus = 0
|
||
DBTX_STATUS_TX_DOING DbTxStatus = 1
|
||
DBTX_STATUS_TX_SUCCESS DbTxStatus = 2
|
||
DBTX_STATUS_TX_FAILED DbTxStatus = 3
|
||
)
|
||
|
||
func (m *Model) Close() {
|
||
if m.Redis != nil {
|
||
m.Redis.Close()
|
||
}
|
||
if m.DB == nil {
|
||
m.DbTxStatus = DBTX_STATUS_TX_NONE
|
||
} else if m.DbTxStatus == DBTX_STATUS_TX_DOING {
|
||
m.DB.Rollback()
|
||
m.DbTxStatus = DBTX_STATUS_TX_FAILED
|
||
}
|
||
}
|