idanoo/go-mastodon-stats
1
0
Fork 0
mirror of https://github.com/idanoo/go-mastodon-stats synced 2025-07-02 22:32:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add calckey

Browse source
This commit is contained in:
Daniel Mason 2023-05-01 21:40:06 +12:00
parent c252860d87
commit 6fdda6c360
Signed by: idanoo
GPG key ID: 387387CDBC02F132
3 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
cmd/gomastodonstats/main.go
View file

@ -78,6 +78,7 @@ func main() {
gms.MOBILIZON_DB_SCHEMA = os.Getenv("MOBILIZON_DB_SCHEMA")
gms.PEERTUBE_DB_SCHEMA = os.Getenv("PEERTUBE_DB_SCHEMA")
gms.BOOKWYRM_DB_SCHEMA = os.Getenv("BOOKWYRM_DB_SCHEMA")
gms.CALCKEY_DB_SCHEMA = os.Getenv("CALCKEY_DB_SCHEMA")
gms.Run()
}

5
internal/gomastodonstats/consts.go
View file

@ -65,4 +65,9 @@ var (
BOOKWYRM_DB_SCHEMA string
BOOKWYRM_USER_QUERY = "SELECT count(*) FROM bookwyrm_user WHERE local = True AND is_active = True;"
BOOKWYRM_IDENTIFIER = "bookwyrm"
// Calckey
CALCKEY_DB_SCHEMA string
CALCKEY_USER_QUERY = `SELECT count(*) FROM "user" WHERE "isDeleted" = False and "host" IS NULL;`
CALCKEY_IDENTIFIER = "calckey"
)

17
internal/gomastodonstats/metrics.go
View file

@ -155,6 +155,23 @@ func getUserCounts() ([]metric, error) {
}
}
if CALCKEY_DB_SCHEMA != "" {
userCount, err := runIntQuery(CALCKEY_DB_SCHEMA, CALCKEY_USER_QUERY)
if err != nil {
log.Println(err)
} else {
m := metric{
Service: CALCKEY_IDENTIFIER,
MetricName: METRICNAME_USERCOUNT,
MetricValue: userCount,
PreviousDayMetricValue: getLastMetric(CALCKEY_IDENTIFIER),
PreviousWeekMetricValue: getLastWeekMetric(CALCKEY_IDENTIFIER),
}
log.Printf("%s user count: %d", CALCKEY_IDENTIFIER, userCount)
metrics = append(metrics, m)
}
}
return metrics, nil
}