package/aqi
1
0
Fork 0
mirror of https://github.com/wonli/aqi.git synced 2025-10-05 08:36:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

add callback

Browse Source
This commit is contained in:
ideaa
2024-07-08 08:58:39 +08:00
parent 52d80f632f
commit 2c31cabc5b
4 changed files with 40 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
store/use_sqlserver.go
View File

@@ -13,6 +13,9 @@ import (
type SqlServerStore struct {
configKey string
callback callback
hasCallback bool
}
func (m *SqlServerStore) Config() *config.SqlServer {
@@ -25,6 +28,11 @@ func (m *SqlServerStore) Config() *config.SqlServer {
return r
}
func (m *SqlServerStore) Callback(fn callback) {
m.callback = fn
m.hasCallback = true
}
func (m *SqlServerStore) Use() *gorm.DB {
r := m.Config()
if r == nil {
@@ -44,6 +52,10 @@ func (m *SqlServerStore) Use() *gorm.DB {
return nil
}
if m.hasCallback {
m.callback(db)
}
sqlDB, err := db.DB()
if err != nil {
logger.SugarLog.Errorf("%s (ping)", err.Error())