fix(db): sets max open and idle connections for postgres

rapyuta-robotics · Nov 30, 2023 · 85ffa20 · 85ffa20
1 parent b01f1f1
commit 85ffa20
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@
 - Code reorganisation, a lot of code has moved, please review the following PRs accordingly [#1444](https://github.com/juanfont/headscale/pull/1444)
 
 ### Changes
+- Set max open and idle connections for postgres
 
 ## 0.22.3 (2023-05-12)
 
@@ -19,7 +20,7 @@
 ### Changes
 
 - Add environment flags to enable pprof (profiling) [#1382](https://github.com/juanfont/headscale/pull/1382)
-  - Profiles are continously generated in our integration tests.
+- Profiles are continously generated in our integration tests.
 - Fix systemd service file location in `.deb` packages [#1391](https://github.com/juanfont/headscale/pull/1391)
 - Improvements on Noise implementation [#1379](https://github.com/juanfont/headscale/pull/1379)
 - Replace node filter logic, ensuring nodes with access can see eachother [#1381](https://github.com/juanfont/headscale/pull/1381)

diff --git a/hscontrol/db.go b/hscontrol/db.go
@@ -251,10 +251,17 @@ func (h *Headscale) openDB() (*gorm.DB, error) {
 		sqlDB.SetConnMaxIdleTime(time.Hour)
 
 	case Postgres:
-		db, err = gorm.Open(postgres.Open(h.dbString), &gorm.Config{
+		db, err := gorm.Open(postgres.Open(h.dbString), &gorm.Config{
 			DisableForeignKeyConstraintWhenMigrating: true,
 			Logger:                                   log,
 		})
+
+		sqlDB, _ := db.DB()
+		sqlDB.SetMaxOpenConns(10)
+		sqlDB.SetMaxIdleConns(10)
+		sqlDB.SetConnMaxIdleTime(time.Hour)
+
+		return db, err
 	}
 
 	if err != nil {