package/aqi
1
0
Fork 0
mirror of https://github.com/wonli/aqi.git synced 2025-10-06 09:06:56 +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_mysql.go
View File

@@ -14,6 +14,9 @@ import (
type MySQLStore struct {
configKey string
callback callback
hasCallback bool
}
func (m *MySQLStore) Config() *config.MySQL {
@@ -26,6 +29,11 @@ func (m *MySQLStore) Config() *config.MySQL {
return r
}
func (m *MySQLStore) Callback(fn callback) {
m.callback = fn
m.hasCallback = true
}
func (m *MySQLStore) Use() *gorm.DB {
r := m.Config()
if r == nil {
@@ -49,6 +57,10 @@ func (m *MySQLStore) Use() *gorm.DB {
return nil
}
if m.hasCallback {
m.callback(db)
}
sqlDB, err := db.DB()
if err != nil {
logger.SugarLog.Error("Error pinging database", zap.String("error", err.Error()))