Эх сурвалжийг харах

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

Alexandre Macabies 4 жил өмнө
parent
commit
95f6487912
2 өөрчлөгдсөн 5 нэмэгдсэн , 5 устгасан
  1. 2 2
      database/database.go
  2. 3 3
      main.go

+ 2 - 2
database/database.go

@@ -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{

+ 3 - 3
main.go

@@ -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)