extern/hishtory
1
0
Fork 0
mirror of https://github.com/ddworken/hishtory.git synced 2025-04-12 03:28:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Configure custom logger when running hishtory server with sqlite

Browse Source
This commit is contained in:
David Dworken 2023-11-19 08:48:46 -08:00
parent 1c68b23790
commit b4f5f50ef7
No known key found for this signature in database
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/server/server.go
View File

@ -34,9 +34,19 @@ func isProductionEnvironment() bool {
return os.Getenv("HISHTORY_ENV") == "prod"
}
func getLoggerConfig() logger.Interface {
// The same as the default logger, except with a higher SlowThreshold
return logger.New(log.New(os.Stdout, "\r\n", log.LstdFlags), logger.Config{
SlowThreshold: 1000 * time.Millisecond,
LogLevel: logger.Info,
IgnoreRecordNotFoundError: false,
Colorful: true,
})
}
func OpenDB() (*database.DB, error) {
if isTestEnvironment() {
db, err := database.OpenSQLite("file::memory:?_journal_mode=WAL&cache=shared", &gorm.Config{})
db, err := database.OpenSQLite("file::memory:?_journal_mode=WAL&cache=shared", &gorm.Config{Logger: getLoggerConfig()})
if err != nil {
return nil, fmt.Errorf("failed to connect to the DB: %w", err)
}
@ -53,20 +63,12 @@ func OpenDB() (*database.DB, error) {
return db, nil
}
// The same as the default logger, except with a higher SlowThreshold
customLogger := logger.New(log.New(os.Stdout, "\r\n", log.LstdFlags), logger.Config{
SlowThreshold: 1000 * time.Millisecond,
LogLevel: logger.Warn,
IgnoreRecordNotFoundError: false,
Colorful: true,
})
var sqliteDb string
if os.Getenv("HISHTORY_SQLITE_DB") != "" {
sqliteDb = os.Getenv("HISHTORY_SQLITE_DB")
}
config := gorm.Config{Logger: customLogger}
config := gorm.Config{Logger: getLoggerConfig()}
fmt.Println("Connecting to DB")
var db *database.DB