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

Update the healthcheck endpoint so it also writes to the DB to detect future cases of #38

Browse Source
This commit is contained in:
David Dworken 2022-11-17 23:43:25 -08:00
parent 4eda871232
commit 0bd9249df3
No known key found for this signature in database
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
backend/server/server.go
View File

@ -351,15 +351,26 @@ func healthCheckHandler(w http.ResponseWriter, r *http.Request) {
panic(fmt.Sprintf("failed to ping DB: %v", err))
}
if isProductionEnvironment(r) {
// Check that we have a reasonable looking set of devices/entries in the DB
var count int64
checkGormResult(GLOBAL_DB.Model(&shared.EncHistoryEntry{}).Count(&count))
if count < 100 {
if count < 1000 {
panic("Suspiciously few enc history entries!")
}
checkGormResult(GLOBAL_DB.Model(&shared.Device{}).Count(&count))
if count < 50 {
if count < 100 {
panic("Suspiciously few devices!")
}
// Check that we can write to the DB. This entry will get written and then eventually cleaned by the cron.
checkGormResult(GLOBAL_DB.Create(&shared.EncHistoryEntry{
EncryptedData: []byte("data"),
Nonce: []byte("nonce"),
DeviceId: "healthcheck_device_id",
UserId: "healthcheck_user_id",
Date: time.Now(),
EncryptedId: "healthcheck_enc_id",
ReadCount: 10000,
}))
}
ok := "OK"
w.Write([]byte(ok))