mirror of
https://codeberg.org/forgejo/forgejo
synced 2024-08-28 18:41:33 +02:00
Part of #27065 This reduces the usage of `db.DefaultContext`. I think I've got enough files for the first PR. When this is merged, I will continue working on this. Considering how many files this PR affect, I hope it won't take to long to merge, so I don't end up in the merge conflict hell. --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
389 lines
8.6 KiB
Go
Executable file
389 lines
8.6 KiB
Go
Executable file
// Copyright 2018 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package metrics
|
|
|
|
import (
|
|
"runtime"
|
|
|
|
activities_model "code.gitea.io/gitea/models/activities"
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
const namespace = "gitea_"
|
|
|
|
// Collector implements the prometheus.Collector interface and
|
|
// exposes gitea metrics for prometheus
|
|
type Collector struct {
|
|
Accesses *prometheus.Desc
|
|
Attachments *prometheus.Desc
|
|
BuildInfo *prometheus.Desc
|
|
Comments *prometheus.Desc
|
|
Follows *prometheus.Desc
|
|
HookTasks *prometheus.Desc
|
|
Issues *prometheus.Desc
|
|
IssuesOpen *prometheus.Desc
|
|
IssuesClosed *prometheus.Desc
|
|
IssuesByLabel *prometheus.Desc
|
|
IssuesByRepository *prometheus.Desc
|
|
Labels *prometheus.Desc
|
|
LoginSources *prometheus.Desc
|
|
Milestones *prometheus.Desc
|
|
Mirrors *prometheus.Desc
|
|
Oauths *prometheus.Desc
|
|
Organizations *prometheus.Desc
|
|
Projects *prometheus.Desc
|
|
ProjectBoards *prometheus.Desc
|
|
PublicKeys *prometheus.Desc
|
|
Releases *prometheus.Desc
|
|
Repositories *prometheus.Desc
|
|
Stars *prometheus.Desc
|
|
Teams *prometheus.Desc
|
|
UpdateTasks *prometheus.Desc
|
|
Users *prometheus.Desc
|
|
Watches *prometheus.Desc
|
|
Webhooks *prometheus.Desc
|
|
}
|
|
|
|
// NewCollector returns a new Collector with all prometheus.Desc initialized
|
|
func NewCollector() Collector {
|
|
return Collector{
|
|
Accesses: prometheus.NewDesc(
|
|
namespace+"accesses",
|
|
"Number of Accesses",
|
|
nil, nil,
|
|
),
|
|
Attachments: prometheus.NewDesc(
|
|
namespace+"attachments",
|
|
"Number of Attachments",
|
|
nil, nil,
|
|
),
|
|
BuildInfo: prometheus.NewDesc(
|
|
namespace+"build_info",
|
|
"Build information",
|
|
[]string{
|
|
"goarch",
|
|
"goos",
|
|
"goversion",
|
|
"version",
|
|
}, nil,
|
|
),
|
|
Comments: prometheus.NewDesc(
|
|
namespace+"comments",
|
|
"Number of Comments",
|
|
nil, nil,
|
|
),
|
|
Follows: prometheus.NewDesc(
|
|
namespace+"follows",
|
|
"Number of Follows",
|
|
nil, nil,
|
|
),
|
|
HookTasks: prometheus.NewDesc(
|
|
namespace+"hooktasks",
|
|
"Number of HookTasks",
|
|
nil, nil,
|
|
),
|
|
Issues: prometheus.NewDesc(
|
|
namespace+"issues",
|
|
"Number of Issues",
|
|
nil, nil,
|
|
),
|
|
IssuesByLabel: prometheus.NewDesc(
|
|
namespace+"issues_by_label",
|
|
"Number of Issues",
|
|
[]string{"label"}, nil,
|
|
),
|
|
IssuesByRepository: prometheus.NewDesc(
|
|
namespace+"issues_by_repository",
|
|
"Number of Issues",
|
|
[]string{"repository"}, nil,
|
|
),
|
|
IssuesOpen: prometheus.NewDesc(
|
|
namespace+"issues_open",
|
|
"Number of open Issues",
|
|
nil, nil,
|
|
),
|
|
IssuesClosed: prometheus.NewDesc(
|
|
namespace+"issues_closed",
|
|
"Number of closed Issues",
|
|
nil, nil,
|
|
),
|
|
Labels: prometheus.NewDesc(
|
|
namespace+"labels",
|
|
"Number of Labels",
|
|
nil, nil,
|
|
),
|
|
LoginSources: prometheus.NewDesc(
|
|
namespace+"loginsources",
|
|
"Number of LoginSources",
|
|
nil, nil,
|
|
),
|
|
Milestones: prometheus.NewDesc(
|
|
namespace+"milestones",
|
|
"Number of Milestones",
|
|
nil, nil,
|
|
),
|
|
Mirrors: prometheus.NewDesc(
|
|
namespace+"mirrors",
|
|
"Number of Mirrors",
|
|
nil, nil,
|
|
),
|
|
Oauths: prometheus.NewDesc(
|
|
namespace+"oauths",
|
|
"Number of Oauths",
|
|
nil, nil,
|
|
),
|
|
Organizations: prometheus.NewDesc(
|
|
namespace+"organizations",
|
|
"Number of Organizations",
|
|
nil, nil,
|
|
),
|
|
Projects: prometheus.NewDesc(
|
|
namespace+"projects",
|
|
"Number of projects",
|
|
nil, nil,
|
|
),
|
|
ProjectBoards: prometheus.NewDesc(
|
|
namespace+"projects_boards",
|
|
"Number of project boards",
|
|
nil, nil,
|
|
),
|
|
PublicKeys: prometheus.NewDesc(
|
|
namespace+"publickeys",
|
|
"Number of PublicKeys",
|
|
nil, nil,
|
|
),
|
|
Releases: prometheus.NewDesc(
|
|
namespace+"releases",
|
|
"Number of Releases",
|
|
nil, nil,
|
|
),
|
|
Repositories: prometheus.NewDesc(
|
|
namespace+"repositories",
|
|
"Number of Repositories",
|
|
nil, nil,
|
|
),
|
|
Stars: prometheus.NewDesc(
|
|
namespace+"stars",
|
|
"Number of Stars",
|
|
nil, nil,
|
|
),
|
|
Teams: prometheus.NewDesc(
|
|
namespace+"teams",
|
|
"Number of Teams",
|
|
nil, nil,
|
|
),
|
|
UpdateTasks: prometheus.NewDesc(
|
|
namespace+"updatetasks",
|
|
"Number of UpdateTasks",
|
|
nil, nil,
|
|
),
|
|
Users: prometheus.NewDesc(
|
|
namespace+"users",
|
|
"Number of Users",
|
|
nil, nil,
|
|
),
|
|
Watches: prometheus.NewDesc(
|
|
namespace+"watches",
|
|
"Number of Watches",
|
|
nil, nil,
|
|
),
|
|
Webhooks: prometheus.NewDesc(
|
|
namespace+"webhooks",
|
|
"Number of Webhooks",
|
|
nil, nil,
|
|
),
|
|
}
|
|
}
|
|
|
|
// Describe returns all possible prometheus.Desc
|
|
func (c Collector) Describe(ch chan<- *prometheus.Desc) {
|
|
ch <- c.Accesses
|
|
ch <- c.Attachments
|
|
ch <- c.BuildInfo
|
|
ch <- c.Comments
|
|
ch <- c.Follows
|
|
ch <- c.HookTasks
|
|
ch <- c.Issues
|
|
ch <- c.IssuesByLabel
|
|
ch <- c.IssuesByRepository
|
|
ch <- c.IssuesOpen
|
|
ch <- c.IssuesClosed
|
|
ch <- c.Labels
|
|
ch <- c.LoginSources
|
|
ch <- c.Milestones
|
|
ch <- c.Mirrors
|
|
ch <- c.Oauths
|
|
ch <- c.Organizations
|
|
ch <- c.Projects
|
|
ch <- c.ProjectBoards
|
|
ch <- c.PublicKeys
|
|
ch <- c.Releases
|
|
ch <- c.Repositories
|
|
ch <- c.Stars
|
|
ch <- c.Teams
|
|
ch <- c.UpdateTasks
|
|
ch <- c.Users
|
|
ch <- c.Watches
|
|
ch <- c.Webhooks
|
|
}
|
|
|
|
// Collect returns the metrics with values
|
|
func (c Collector) Collect(ch chan<- prometheus.Metric) {
|
|
stats := activities_model.GetStatistic(db.DefaultContext)
|
|
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Accesses,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Access),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Attachments,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Attachment),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.BuildInfo,
|
|
prometheus.GaugeValue,
|
|
1,
|
|
runtime.GOARCH,
|
|
runtime.GOOS,
|
|
runtime.Version(),
|
|
setting.AppVer,
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Comments,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Comment),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Follows,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Follow),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.HookTasks,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.HookTask),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Issues,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Issue),
|
|
)
|
|
for _, il := range stats.Counter.IssueByLabel {
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.IssuesByLabel,
|
|
prometheus.GaugeValue,
|
|
float64(il.Count),
|
|
il.Label,
|
|
)
|
|
}
|
|
for _, ir := range stats.Counter.IssueByRepository {
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.IssuesByRepository,
|
|
prometheus.GaugeValue,
|
|
float64(ir.Count),
|
|
ir.OwnerName+"/"+ir.Repository,
|
|
)
|
|
}
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.IssuesClosed,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.IssueClosed),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.IssuesOpen,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.IssueOpen),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Labels,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Label),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.LoginSources,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.AuthSource),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Milestones,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Milestone),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Mirrors,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Mirror),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Oauths,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Oauth),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Organizations,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Org),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Projects,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Project),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.ProjectBoards,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.ProjectBoard),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.PublicKeys,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.PublicKey),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Releases,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Release),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Repositories,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Repo),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Stars,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Star),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Teams,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Team),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.UpdateTasks,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.UpdateTask),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Users,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.User),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Watches,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Watch),
|
|
)
|
|
ch <- prometheus.MustNewConstMetric(
|
|
c.Webhooks,
|
|
prometheus.GaugeValue,
|
|
float64(stats.Counter.Webhook),
|
|
)
|
|
}
|