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

Add BookWyrm stats

Browse source
This commit is contained in:
Daniel Mason 2023-03-26 08:58:36 +13:00
parent 853e7ea48f
commit c252860d87
Signed by: idanoo
GPG key ID: 387387CDBC02F132
4 changed files with 25 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
internal/gomastodonstats/consts.go
View file

@ -32,6 +32,7 @@ var (
MASTODON_IDENTIFIER: "https://mastodon.nz",
MOBILIZON_IDENTIFIER: "https://openevents.nz",
PEERTUBE_IDENTIFIER: "https://peertube.nz",
BOOKWYRM_IDENTIFIER: "https://bookworm.nz",
}
// Pixelfed
@ -59,4 +60,9 @@ var (
PEERTUBE_DB_SCHEMA string
PEERTUBE_USER_QUERY = "SELECT count(*) FROM \"user\" WHERE blocked = False;"
PEERTUBE_IDENTIFIER = "peertube"
// BookWyrm
BOOKWYRM_DB_SCHEMA string
BOOKWYRM_USER_QUERY = "SELECT count(*) FROM bookwyrm_user WHERE local = True AND is_active = True;"
BOOKWYRM_IDENTIFIER = "bookwyrm"
)

17
internal/gomastodonstats/metrics.go
View file

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