extern/hishtory
1
0
Fork 0
mirror of https://github.com/ddworken/hishtory.git synced 2025-06-24 22:11:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Optimize healthcheck handler to avoid unnecessary ping in prod + add prod-only config for connection reuse

Browse Source
This commit is contained in:
David Dworken 2022-11-26 18:33:54 -08:00
parent a98bff0db8
commit 865ce06b95
No known key found for this signature in database
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/server/server.go
View File

@ -390,14 +390,6 @@ func addDeletionRequestHandler(ctx context.Context, w http.ResponseWriter, r *ht
}
func healthCheckHandler(ctx context.Context, w http.ResponseWriter, r *http.Request) {
db, err := GLOBAL_DB.DB()
if err != nil {
panic(fmt.Sprintf("failed to get DB: %v", err))
}
err = db.Ping()
if err != nil {
panic(fmt.Sprintf("failed to ping DB: %v", err))
}
if isProductionEnvironment() {
// Check that we have a reasonable looking set of devices/entries in the DB
rows, err := GLOBAL_DB.Raw("SELECT true FROM enc_history_entries LIMIT 1 OFFSET 1000").Rows()
@ -422,6 +414,15 @@ func healthCheckHandler(ctx context.Context, w http.ResponseWriter, r *http.Requ
EncryptedId: "healthcheck_enc_id",
ReadCount: 10000,
}))
} else {
db, err := GLOBAL_DB.DB()
if err != nil {
panic(fmt.Sprintf("failed to get DB: %v", err))
}
err = db.Ping()
if err != nil {
panic(fmt.Sprintf("failed to ping DB: %v", err))
}
}
ok := "OK"
w.Write([]byte(ok))
@ -629,6 +630,11 @@ func InitDB() {
if err != nil {
panic(err)
}
if isProductionEnvironment() {
tx.SetMaxIdleConns(10)
tx.SetMaxOpenConns(100)
tx.SetConnMaxLifetime(time.Minute * 30)
}
}
func decrementVersion(version string) (string, error) {