Cast int to duration

scroll-tech · Mar 25, 2024 · 4238c9a · 4238c9a
1 parent 816c403
commit 4238c9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/common/database/db.go b/common/database/db.go
@@ -71,13 +71,13 @@ func InitDB(config *Config) (*gorm.DB, error) {
 	}
 
 	if (config.MaxLifetime != 0) {
-		sqlDB.SetConnMaxLifetime(time.Second * config.MaxLifetime)
+		sqlDB.SetConnMaxLifetime(time.Second * time.Duration(config.MaxLifetime))
 	} else {
 		sqlDB.SetConnMaxLifetime(time.Minute * 10)
 	}
 
 	if (config.MaxIdleTime != 0) {
-		sqlDB.SetConnMaxIdleTime(time.Second * config.MaxIdleTime)
+		sqlDB.SetConnMaxIdleTime(time.Second * time.Duration(config.MaxIdleTime))
 	} else {
 		sqlDB.SetConnMaxIdleTime(time.Minute * 5)
 	}