|
|
|
@ -87,19 +87,22 @@ func (mh *MetricsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
|
|
|
|
|
func (mh *MetricsHandler) updateMetrics() {
|
|
|
|
|
// Update metricURLsTotalGauge
|
|
|
|
|
for state := db.PasteStateUndef; state <= db.PasteStateDeleted; state++ {
|
|
|
|
|
for ty := db.PasteTypeUndef; ty <= db.PasteTypeFileUpload; ty++ {
|
|
|
|
|
var count int64
|
|
|
|
|
query := mh.db.Unscoped().Model(&db.Paste{}).Where("type = ? AND state = ?", ty, state).Count(&count)
|
|
|
|
|
if err := query.Error; err != nil {
|
|
|
|
|
log.Printf("error: %v", errors.Wrap(err, "fetching pastes_total metric"))
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
labels := map[string]string{
|
|
|
|
|
"state": state.String(),
|
|
|
|
|
"type": ty.String(),
|
|
|
|
|
}
|
|
|
|
|
metricURLsTotalGauge.With(labels).Set(float64(count))
|
|
|
|
|
results := make([](struct {
|
|
|
|
|
Type db.PasteType
|
|
|
|
|
State db.PasteState
|
|
|
|
|
Count float64
|
|
|
|
|
}), 0)
|
|
|
|
|
query := mh.db.Unscoped().Model(&db.Paste{}).Select("type", "state", "COUNT(*) as count").Group("type, state").Find(&results)
|
|
|
|
|
if err := query.Error; err != nil {
|
|
|
|
|
log.Printf("error: %v", errors.Wrap(err, "fetching pastes_total metric"))
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
metricURLsTotalGauge.Reset()
|
|
|
|
|
for _, r := range results {
|
|
|
|
|
labels := map[string]string{
|
|
|
|
|
"type": r.Type.String(),
|
|
|
|
|
"state": r.State.String(),
|
|
|
|
|
}
|
|
|
|
|
metricURLsTotalGauge.With(labels).Set(r.Count)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|