mirror of
https://github.com/injoyai/tdx.git
synced 2025-11-26 21:25:35 +08:00
增加Manage对mysql的支持
This commit is contained in:
19
codes.go
19
codes.go
@@ -34,12 +34,6 @@ func NewCodesMysql(c *Client, dsn string) (*Codes, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
db.SetMapper(core.SameMapper{})
|
db.SetMapper(core.SameMapper{})
|
||||||
if err := db.Sync2(new(CodeModel)); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
if err := db.Sync2(new(UpdateModel)); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return NewCodes(c, db)
|
return NewCodes(c, db)
|
||||||
}
|
}
|
||||||
@@ -62,6 +56,12 @@ func NewCodesSqlite(c *Client, filenames ...string) (*Codes, error) {
|
|||||||
}
|
}
|
||||||
db.SetMapper(core.SameMapper{})
|
db.SetMapper(core.SameMapper{})
|
||||||
db.DB().SetMaxOpenConns(1)
|
db.DB().SetMaxOpenConns(1)
|
||||||
|
|
||||||
|
return NewCodes(c, db)
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewCodes(c *Client, db *xorm.Engine) (*Codes, error) {
|
||||||
|
|
||||||
if err := db.Sync2(new(CodeModel)); err != nil {
|
if err := db.Sync2(new(CodeModel)); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -69,11 +69,6 @@ func NewCodesSqlite(c *Client, filenames ...string) (*Codes, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return NewCodes(c, db)
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewCodes(c *Client, db *xorm.Engine) (*Codes, error) {
|
|
||||||
|
|
||||||
update := new(UpdateModel)
|
update := new(UpdateModel)
|
||||||
{ //查询或者插入一条数据
|
{ //查询或者插入一条数据
|
||||||
has, err := db.Get(update)
|
has, err := db.Get(update)
|
||||||
@@ -366,7 +361,7 @@ type UpdateModel struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (*UpdateModel) TableName() string {
|
func (*UpdateModel) TableName() string {
|
||||||
return "update"
|
return "codes_update"
|
||||||
}
|
}
|
||||||
|
|
||||||
type CodeModel struct {
|
type CodeModel struct {
|
||||||
|
|||||||
10
workday.go
10
workday.go
@@ -24,9 +24,6 @@ func NewWorkdayMysql(c *Client, dsn string) (*Workday, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
db.SetMapper(core.SameMapper{})
|
db.SetMapper(core.SameMapper{})
|
||||||
if err := db.Sync2(new(WorkdayModel)); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return NewWorkday(c, db)
|
return NewWorkday(c, db)
|
||||||
}
|
}
|
||||||
@@ -47,14 +44,15 @@ func NewWorkdaySqlite(c *Client, filenames ...string) (*Workday, error) {
|
|||||||
}
|
}
|
||||||
db.SetMapper(core.SameMapper{})
|
db.SetMapper(core.SameMapper{})
|
||||||
db.DB().SetMaxOpenConns(1)
|
db.DB().SetMaxOpenConns(1)
|
||||||
if err := db.Sync2(new(WorkdayModel)); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return NewWorkday(c, db)
|
return NewWorkday(c, db)
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewWorkday(c *Client, db *xorm.Engine) (*Workday, error) {
|
func NewWorkday(c *Client, db *xorm.Engine) (*Workday, error) {
|
||||||
|
if err := db.Sync2(new(WorkdayModel)); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
w := &Workday{
|
w := &Workday{
|
||||||
Client: c,
|
Client: c,
|
||||||
db: db,
|
db: db,
|
||||||
|
|||||||
Reference in New Issue
Block a user