Skip to content

Commit 5e97455

Browse files
committed
feat: add default param for mysql
1 parent 03d8c9d commit 5e97455

1 file changed

Lines changed: 17 additions & 0 deletions

File tree

init_sqlagent.go

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,26 @@ func readDBConfig(cfgFile string) (*dsncfg.Database, error) {
5353
return dbCfg, nil
5454
}
5555

56+
func setDefaultDBParameters(cfg *dsncfg.Database) {
57+
if cfg.Type == dsncfg.MySql {
58+
defaultParams := map[string]string{
59+
"parseTime": "true",
60+
"charset": "utf8mb4,utf8",
61+
"autocommit": "true",
62+
"loc": "Asia%2FShanghai",
63+
}
64+
for k, v := range defaultParams {
65+
if _, ok := cfg.Parameters[k]; !ok {
66+
cfg.Parameters[k] = v
67+
}
68+
}
69+
}
70+
}
71+
5672
// initSqlAgent init module SqlAgent only once.
5773
func initSqlAgent(cfg *dsncfg.Database) (err error) {
5874
initOnce.Do(func() {
75+
setDefaultDBParameters(cfg)
5976
defaultAgent, err = NewSqlAgent(cfg)
6077
})
6178
return

0 commit comments

Comments
 (0)