Setup conns according to number of cpus

netbirdio · Sep 14, 2023 · 3a112b4 · 3a112b4
1 parent 5efa612
commit 3a112b4
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/management/server/sqlite_store.go b/management/server/sqlite_store.go
@@ -2,6 +2,7 @@ package server
 
 import (
 	"path/filepath"
+	"runtime"
 	"strings"
 	"sync"
 	"time"
@@ -41,12 +42,12 @@ func NewSqliteStore(dataDir string, metrics telemetry.AppMetrics) (*SqliteStore,
 		return nil, err
 	}
 
-	//sql, err := db.DB()
-	//if err != nil {
-	//	return nil, err
-	//}
-	//conns := runtime.NumCPU()
-	//sql.SetMaxOpenConns(conns) // TODO: make it configurable
+	sql, err := db.DB()
+	if err != nil {
+		return nil, err
+	}
+	conns := runtime.NumCPU()
+	sql.SetMaxOpenConns(conns) // TODO: make it configurable
 
 	err = db.AutoMigrate(
 		&SetupKey{}, &Peer{}, &User{}, &PersonalAccessToken{}, &Group{}, &Rule{},