genofire/hs_monolith
genofire
/
hs_monolith
Archived
1
0
Fork 0

[TASK] add connection to sql log (fix #4)

This commit is contained in:
Martin Geno 2017-04-04 09:49:46 +02:00
parent ff299c42a7
commit 84f060cb3b
No known key found for this signature in database
GPG Key ID: F0D39A37E925E941
2 changed files with 8 additions and 1 deletions

1
.gitignore vendored
View File

@ -126,4 +126,5 @@ __pycache__
# go project
config.conf
cmd/stock/config.conf

View File

@ -4,6 +4,8 @@ import (
"github.com/jinzhu/gorm"
_ "github.com/jinzhu/gorm/dialects/postgres"
_ "github.com/jinzhu/gorm/dialects/sqlite"
"github.com/genofire/hs_master-kss-monolith/lib/log"
)
var (
@ -21,19 +23,23 @@ type Config struct {
}
func Open(c Config) (err error) {
writeLog := log.Log.WithField("db", "write")
config = &c
Write, err = gorm.Open(config.Type, config.Connection)
Write.SingularTable(true)
Write.LogMode(c.Logging)
Write.SetLogger(writeLog)
Write.Callback().Create().Remove("gorm:update_time_stamp")
Write.Callback().Update().Remove("gorm:update_time_stamp")
if err != nil {
return
}
if len(config.ReadConnection) > 0 {
readLog := log.Log.WithField("db", "read")
Read, err = gorm.Open(config.Type, config.ReadConnection)
Read.SingularTable(true)
Read.LogMode(c.Logging)
Read.SetLogger(readLog)
Read.Callback().Create().Remove("gorm:update_time_stamp")
Read.Callback().Update().Remove("gorm:update_time_stamp")
} else {