LordMZTE/mautrix-whatsapp
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Make database log a sublogger of Bridge, like all other modules. (#260)

Browse source
This commit is contained in:
Alexandre Macabies 2021-01-25 20:01:54 +01:00 committed by GitHub
parent 500cb0cd7c
commit 95f6487912
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
database/database.go
View file

@ -38,7 +38,7 @@ type Database struct {
Message *MessageQuery
}
func New(dbType string, uri string) (*Database, error) {
func New(dbType string, uri string, baseLog log.Logger) (*Database, error) {
conn, err := sql.Open(dbType, uri)
if err != nil {
return nil, err
@ -50,7 +50,7 @@ func New(dbType string, uri string) (*Database, error) {
db := &Database{
DB: conn,
log: log.Sub("Database"),
log: baseLog.Sub("Database"),
dialect: dbType,
}
db.User = &UserQuery{

6
main.go
View file

@ -96,7 +96,7 @@ func (bridge *Bridge) GenerateRegistration() {
}
func (bridge *Bridge) MigrateDatabase() {
oldDB, err := database.New(flag.Arg(0), flag.Arg(1))
oldDB, err := database.New(flag.Arg(0), flag.Arg(1), bridge.Log)
if err != nil {
fmt.Println("Failed to open old database:", err)
os.Exit(30)
@ -107,7 +107,7 @@ func (bridge *Bridge) MigrateDatabase() {
os.Exit(31)
}
newDB, err := database.New(bridge.Config.AppService.Database.Type, bridge.Config.AppService.Database.URI)
newDB, err := database.New(bridge.Config.AppService.Database.Type, bridge.Config.AppService.Database.URI, bridge.Log)
if err != nil {
fmt.Println("Failed to open new database:", err)
os.Exit(32)
@ -223,7 +223,7 @@ func (bridge *Bridge) Init() {
bridge.AS.Log = log.Sub("Matrix")
bridge.Log.Debugln("Initializing database connection")
bridge.DB, err = database.New(bridge.Config.AppService.Database.Type, bridge.Config.AppService.Database.URI)
bridge.DB, err = database.New(bridge.Config.AppService.Database.Type, bridge.Config.AppService.Database.URI, bridge.Log)
if err != nil {
bridge.Log.Fatalln("Failed to initialize database connection:", err)
os.Exit(14)