diff --git a/assets/templates/middleware/metrics b/assets/templates/middleware/metrics new file mode 100644 index 0000000000000000000000000000000000000000..9dd4047d8908de2d094acc3bc4170bfd5552ec13 --- /dev/null +++ b/assets/templates/middleware/metrics @@ -0,0 +1,39 @@ +import ( + "context" + + "github.com/prometheus/client_golang/prometheus" +) + +{{ $decorator := (or .Vars.DecoratorName "metricsMiddleware") }} +{{ $funcName := (or .Vars.FuncName ("MetricsMiddleware")) }} + +// {{ $decorator }} implements {{ .Interface.Type }} that is instrumented with metrics +type {{ $decorator }} struct { + requestMetrics *metrics.RequestMetrics + next {{ .Interface.Type }} +} + +// {{ $funcName }} instruments an implementation of the {{ $decorator }} with metrics +func {{ $funcName }} (requestMetrics *metrics.RequestMetrics) Middleware { + return func(next {{ .Interface.Type }}) {{ .Interface.Type }} { + return &{{ $decorator }}{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +{{ range $method := .Interface.Methods }} + // {{ $method.Name }} implements {{ $.Interface.Type }} + func (m {{ $decorator }}) {{ $method.Declaration }} { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("{{ $method.Name }}")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("{{ $method.Name }}").Inc() + {{ $method.ResultsNames }} = m.next.{{ $method.Call }} + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("{{ $method.Name }}").Inc() + return + } + return + } +{{ end }} \ No newline at end of file diff --git a/images/middleware/metrics_middleware.go b/images/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..f583718d99a1f2bad613f279c447307c6b715c78 --- /dev/null +++ b/images/middleware/metrics_middleware.go @@ -0,0 +1,45 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/images -i Images -t ../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/images" + "git.perx.ru/perxis/perxis-go/pkg/files" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements images.Images that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next images.Images +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next images.Images) images.Images { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Get implements images.Images +func (m metricsMiddleware) Get(ctx context.Context, source *files.File, opts *images.GetOptions) (result *files.File, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + result, err = m.next.Get(ctx, source, opts) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} diff --git a/pkg/clients/middleware/metrics_middleware.go b/pkg/clients/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..b96bac0685734880625dda7566588bc93d2345c1 --- /dev/null +++ b/pkg/clients/middleware/metrics_middleware.go @@ -0,0 +1,122 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/clients -i Clients -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/clients" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements clients.Clients that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next clients.Clients +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next clients.Clients) clients.Clients { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements clients.Clients +func (m metricsMiddleware) Create(ctx context.Context, client *clients.Client) (created *clients.Client, err error) { + m.requestMetrics.Total.WithLabelValues("Create").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + } + }() + return m.next.Create(ctx, client) +} + +// Delete implements clients.Clients +func (m metricsMiddleware) Delete(ctx context.Context, spaceId string, id string) (err error) { + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + } + }() + return m.next.Delete(ctx, spaceId, id) +} + +// Enable implements clients.Clients +func (m metricsMiddleware) Enable(ctx context.Context, spaceId string, id string, enable bool) (err error) { + m.requestMetrics.Total.WithLabelValues("Enable").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Enable")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Enable").Inc() + } + }() + return m.next.Enable(ctx, spaceId, id, enable) +} + +// Get implements clients.Clients +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, id string) (client *clients.Client, err error) { + m.requestMetrics.Total.WithLabelValues("Get").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + } + }() + return m.next.Get(ctx, spaceId, id) +} + +// GetBy implements clients.Clients +func (m metricsMiddleware) GetBy(ctx context.Context, spaceId string, params *clients.GetByParams) (client *clients.Client, err error) { + m.requestMetrics.Total.WithLabelValues("GetBy").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("GetBy")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("GetBy").Inc() + } + }() + return m.next.GetBy(ctx, spaceId, params) +} + +// List implements clients.Clients +func (m metricsMiddleware) List(ctx context.Context, spaceId string) (clients []*clients.Client, err error) { + m.requestMetrics.Total.WithLabelValues("List").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("List")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("List").Inc() + } + }() + return m.next.List(ctx, spaceId) +} + +// Update implements clients.Clients +func (m metricsMiddleware) Update(ctx context.Context, client *clients.Client) (err error) { + m.requestMetrics.Total.WithLabelValues("Update").Inc() + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer func() { + timer.ObserveDuration() + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + } + }() + return m.next.Update(ctx, client) +} diff --git a/pkg/collaborators/middleware/metrics_middleware.go b/pkg/collaborators/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..0e4dcc986e394f6ea1f12d49b7fbfd7303987f01 --- /dev/null +++ b/pkg/collaborators/middleware/metrics_middleware.go @@ -0,0 +1,96 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/collaborators -i Collaborators -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/collaborators" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements collaborators.Collaborators that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next collaborators.Collaborators +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next collaborators.Collaborators) collaborators.Collaborators { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Get implements collaborators.Collaborators +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, subject string) (role string, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + role, err = m.next.Get(ctx, spaceId, subject) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// ListCollaborators implements collaborators.Collaborators +func (m metricsMiddleware) ListCollaborators(ctx context.Context, spaceId string) (collaborators []*collaborators.Collaborator, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("ListCollaborators")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("ListCollaborators").Inc() + collaborators, err = m.next.ListCollaborators(ctx, spaceId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("ListCollaborators").Inc() + return + } + return +} + +// ListSpaces implements collaborators.Collaborators +func (m metricsMiddleware) ListSpaces(ctx context.Context, subject string) (spaces []*collaborators.Collaborator, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("ListSpaces")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("ListSpaces").Inc() + spaces, err = m.next.ListSpaces(ctx, subject) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("ListSpaces").Inc() + return + } + return +} + +// Remove implements collaborators.Collaborators +func (m metricsMiddleware) Remove(ctx context.Context, spaceId string, subject string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Remove")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Remove").Inc() + err = m.next.Remove(ctx, spaceId, subject) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Remove").Inc() + return + } + return +} + +// Set implements collaborators.Collaborators +func (m metricsMiddleware) Set(ctx context.Context, spaceId string, subject string, role string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Set")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Set").Inc() + err = m.next.Set(ctx, spaceId, subject, role) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Set").Inc() + return + } + return +} diff --git a/pkg/collections/middleware/metrics_middleware.go b/pkg/collections/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..012e11270f8a9ceb9ae11b762c1e2bb7b781d310 --- /dev/null +++ b/pkg/collections/middleware/metrics_middleware.go @@ -0,0 +1,123 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/collections -i Collections -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/collections" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/schema" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements collections.Collections that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next collections.Collections +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next collections.Collections) collections.Collections { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements collections.Collections +func (m metricsMiddleware) Create(ctx context.Context, collection *collections.Collection) (created *collections.Collection, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, collection) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements collections.Collections +func (m metricsMiddleware) Delete(ctx context.Context, spaceId string, envId string, collectionId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, spaceId, envId, collectionId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Get implements collections.Collections +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, envId string, collectionId string, options ...*collections.GetOptions) (collection *collections.Collection, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + collection, err = m.next.Get(ctx, spaceId, envId, collectionId, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// List implements collections.Collections +func (m metricsMiddleware) List(ctx context.Context, spaceId string, envId string, filter *collections.Filter) (collections []*collections.Collection, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("List")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("List").Inc() + collections, err = m.next.List(ctx, spaceId, envId, filter) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("List").Inc() + return + } + return +} + +// SetSchema implements collections.Collections +func (m metricsMiddleware) SetSchema(ctx context.Context, spaceId string, envId string, collectionId string, schema *schema.Schema) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("SetSchema")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("SetSchema").Inc() + err = m.next.SetSchema(ctx, spaceId, envId, collectionId, schema) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("SetSchema").Inc() + return + } + return +} + +// SetState implements collections.Collections +func (m metricsMiddleware) SetState(ctx context.Context, spaceId string, envId string, collectionId string, state *collections.StateInfo) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("SetState")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("SetState").Inc() + err = m.next.SetState(ctx, spaceId, envId, collectionId, state) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("SetState").Inc() + return + } + return +} + +// Update implements collections.Collections +func (m metricsMiddleware) Update(ctx context.Context, coll *collections.Collection) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, coll) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +} diff --git a/pkg/environments/middleware/metrics_middleware.go b/pkg/environments/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..3f4723b64bad3b651b7940cda9f6ee77ae74939e --- /dev/null +++ b/pkg/environments/middleware/metrics_middleware.go @@ -0,0 +1,135 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/environments -i Environments -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/environments" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements environments.Environments that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next environments.Environments +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next environments.Environments) environments.Environments { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements environments.Environments +func (m metricsMiddleware) Create(ctx context.Context, env *environments.Environment) (created *environments.Environment, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, env) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements environments.Environments +func (m metricsMiddleware) Delete(ctx context.Context, spaceId string, envId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, spaceId, envId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Get implements environments.Environments +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, envId string) (env *environments.Environment, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + env, err = m.next.Get(ctx, spaceId, envId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// List implements environments.Environments +func (m metricsMiddleware) List(ctx context.Context, spaceId string) (envs []*environments.Environment, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("List")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("List").Inc() + envs, err = m.next.List(ctx, spaceId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("List").Inc() + return + } + return +} + +// Migrate implements environments.Environments +func (m metricsMiddleware) Migrate(ctx context.Context, spaceId string, envId string, options ...*environments.MigrateOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Migrate")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Migrate").Inc() + err = m.next.Migrate(ctx, spaceId, envId, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Migrate").Inc() + return + } + return +} + +// RemoveAlias implements environments.Environments +func (m metricsMiddleware) RemoveAlias(ctx context.Context, spaceId string, envId string, alias string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("RemoveAlias")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("RemoveAlias").Inc() + err = m.next.RemoveAlias(ctx, spaceId, envId, alias) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("RemoveAlias").Inc() + return + } + return +} + +// SetAlias implements environments.Environments +func (m metricsMiddleware) SetAlias(ctx context.Context, spaceId string, envId string, alias string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("SetAlias")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("SetAlias").Inc() + err = m.next.SetAlias(ctx, spaceId, envId, alias) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("SetAlias").Inc() + return + } + return +} + +// Update implements environments.Environments +func (m metricsMiddleware) Update(ctx context.Context, env *environments.Environment) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, env) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +} diff --git a/pkg/invitations/middleware/metrics_middleware.go b/pkg/invitations/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..ac82571ec0416446956e6e684b1d92f0e5db507a --- /dev/null +++ b/pkg/invitations/middleware/metrics_middleware.go @@ -0,0 +1,97 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/invitations -i Invitations -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/invitations" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/options" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements invitations.Invitations that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next invitations.Invitations +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next invitations.Invitations) invitations.Invitations { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Accept implements invitations.Invitations +func (m metricsMiddleware) Accept(ctx context.Context, invitationId string, userId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Accept")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Accept").Inc() + err = m.next.Accept(ctx, invitationId, userId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Accept").Inc() + return + } + return +} + +// Create implements invitations.Invitations +func (m metricsMiddleware) Create(ctx context.Context, invitation *invitations.Invitation) (created *invitations.Invitation, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, invitation) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements invitations.Invitations +func (m metricsMiddleware) Delete(ctx context.Context, invitationId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, invitationId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Find implements invitations.Invitations +func (m metricsMiddleware) Find(ctx context.Context, filter *invitations.Filter, opts *options.FindOptions) (invitations []*invitations.Invitation, total int, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Find")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Find").Inc() + invitations, total, err = m.next.Find(ctx, filter, opts) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Find").Inc() + return + } + return +} + +// Get implements invitations.Invitations +func (m metricsMiddleware) Get(ctx context.Context, invitationId string) (invitation *invitations.Invitation, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + invitation, err = m.next.Get(ctx, invitationId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} diff --git a/pkg/items/middleware/metrics_middleware.go b/pkg/items/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..a9f7b8f9099750db9dca96a06442f594cd3e0681 --- /dev/null +++ b/pkg/items/middleware/metrics_middleware.go @@ -0,0 +1,266 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/items -i Items -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/items" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/schema" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements items.Items that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next items.Items +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next items.Items) items.Items { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Aggregate implements items.Items +func (m metricsMiddleware) Aggregate(ctx context.Context, spaceId string, envId string, collectionId string, filter *items.Filter, options ...*items.AggregateOptions) (result map[string]interface{}, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Aggregate")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Aggregate").Inc() + result, err = m.next.Aggregate(ctx, spaceId, envId, collectionId, filter, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Aggregate").Inc() + return + } + return +} + +// AggregatePublished implements items.Items +func (m metricsMiddleware) AggregatePublished(ctx context.Context, spaceId string, envId string, collectionId string, filter *items.Filter, options ...*items.AggregatePublishedOptions) (result map[string]interface{}, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("AggregatePublished")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("AggregatePublished").Inc() + result, err = m.next.AggregatePublished(ctx, spaceId, envId, collectionId, filter, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("AggregatePublished").Inc() + return + } + return +} + +// Archive implements items.Items +func (m metricsMiddleware) Archive(ctx context.Context, item *items.Item, options ...*items.ArchiveOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Archive")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Archive").Inc() + err = m.next.Archive(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Archive").Inc() + return + } + return +} + +// Create implements items.Items +func (m metricsMiddleware) Create(ctx context.Context, item *items.Item, opts ...*items.CreateOptions) (created *items.Item, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, item, opts...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements items.Items +func (m metricsMiddleware) Delete(ctx context.Context, item *items.Item, options ...*items.DeleteOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Find implements items.Items +func (m metricsMiddleware) Find(ctx context.Context, spaceId string, envId string, collectionId string, filter *items.Filter, options ...*items.FindOptions) (items []*items.Item, total int, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Find")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Find").Inc() + items, total, err = m.next.Find(ctx, spaceId, envId, collectionId, filter, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Find").Inc() + return + } + return +} + +// FindArchived implements items.Items +func (m metricsMiddleware) FindArchived(ctx context.Context, spaceId string, envId string, collectionId string, filter *items.Filter, options ...*items.FindArchivedOptions) (items []*items.Item, total int, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("FindArchived")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("FindArchived").Inc() + items, total, err = m.next.FindArchived(ctx, spaceId, envId, collectionId, filter, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("FindArchived").Inc() + return + } + return +} + +// FindPublished implements items.Items +func (m metricsMiddleware) FindPublished(ctx context.Context, spaceId string, envId string, collectionId string, filter *items.Filter, options ...*items.FindPublishedOptions) (items []*items.Item, total int, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("FindPublished")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("FindPublished").Inc() + items, total, err = m.next.FindPublished(ctx, spaceId, envId, collectionId, filter, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("FindPublished").Inc() + return + } + return +} + +// Get implements items.Items +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, envId string, collectionId string, itemId string, options ...*items.GetOptions) (item *items.Item, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + item, err = m.next.Get(ctx, spaceId, envId, collectionId, itemId, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// GetPublished implements items.Items +func (m metricsMiddleware) GetPublished(ctx context.Context, spaceId string, envId string, collectionId string, itemId string, options ...*items.GetPublishedOptions) (item *items.Item, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("GetPublished")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("GetPublished").Inc() + item, err = m.next.GetPublished(ctx, spaceId, envId, collectionId, itemId, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("GetPublished").Inc() + return + } + return +} + +// GetRevision implements items.Items +func (m metricsMiddleware) GetRevision(ctx context.Context, spaceId string, envId string, collectionId string, itemId string, revisionId string, options ...*items.GetRevisionOptions) (item *items.Item, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("GetRevision")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("GetRevision").Inc() + item, err = m.next.GetRevision(ctx, spaceId, envId, collectionId, itemId, revisionId, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("GetRevision").Inc() + return + } + return +} + +// Introspect implements items.Items +func (m metricsMiddleware) Introspect(ctx context.Context, item *items.Item, opts ...*items.IntrospectOptions) (itm *items.Item, sch *schema.Schema, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Introspect")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Introspect").Inc() + itm, sch, err = m.next.Introspect(ctx, item, opts...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Introspect").Inc() + return + } + return +} + +// ListRevisions implements items.Items +func (m metricsMiddleware) ListRevisions(ctx context.Context, spaceId string, envId string, collectionId string, itemId string, options ...*items.ListRevisionsOptions) (items []*items.Item, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("ListRevisions")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("ListRevisions").Inc() + items, err = m.next.ListRevisions(ctx, spaceId, envId, collectionId, itemId, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("ListRevisions").Inc() + return + } + return +} + +// Publish implements items.Items +func (m metricsMiddleware) Publish(ctx context.Context, item *items.Item, options ...*items.PublishOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Publish")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Publish").Inc() + err = m.next.Publish(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Publish").Inc() + return + } + return +} + +// Unarchive implements items.Items +func (m metricsMiddleware) Unarchive(ctx context.Context, item *items.Item, options ...*items.UnarchiveOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Unarchive")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Unarchive").Inc() + err = m.next.Unarchive(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Unarchive").Inc() + return + } + return +} + +// Undelete implements items.Items +func (m metricsMiddleware) Undelete(ctx context.Context, item *items.Item, options ...*items.UndeleteOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Undelete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Undelete").Inc() + err = m.next.Undelete(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Undelete").Inc() + return + } + return +} + +// Unpublish implements items.Items +func (m metricsMiddleware) Unpublish(ctx context.Context, item *items.Item, options ...*items.UnpublishOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Unpublish")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Unpublish").Inc() + err = m.next.Unpublish(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Unpublish").Inc() + return + } + return +} + +// Update implements items.Items +func (m metricsMiddleware) Update(ctx context.Context, item *items.Item, options ...*items.UpdateOptions) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, item, options...) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +} diff --git a/pkg/locales/middleware/metrics_middleware.go b/pkg/locales/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..a8296a4c728ab629f1fde80b293cd1edfc1990e9 --- /dev/null +++ b/pkg/locales/middleware/metrics_middleware.go @@ -0,0 +1,70 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/locales -i Locales -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/locales" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements locales.Locales that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next locales.Locales +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next locales.Locales) locales.Locales { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements locales.Locales +func (m metricsMiddleware) Create(ctx context.Context, locale *locales.Locale) (created *locales.Locale, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, locale) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements locales.Locales +func (m metricsMiddleware) Delete(ctx context.Context, spaceId string, localeId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, spaceId, localeId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// List implements locales.Locales +func (m metricsMiddleware) List(ctx context.Context, spaceId string) (locales []*locales.Locale, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("List")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("List").Inc() + locales, err = m.next.List(ctx, spaceId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("List").Inc() + return + } + return +} diff --git a/pkg/members/middleware/metrics_middleware.go b/pkg/members/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..8e947287343e7801d486edfd8cacff182fc158cd --- /dev/null +++ b/pkg/members/middleware/metrics_middleware.go @@ -0,0 +1,109 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/members -i Members -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/members" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements members.Members that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next members.Members +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next members.Members) members.Members { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Get implements members.Members +func (m metricsMiddleware) Get(ctx context.Context, orgId string, userId string) (role members.Role, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + role, err = m.next.Get(ctx, orgId, userId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// ListMembers implements members.Members +func (m metricsMiddleware) ListMembers(ctx context.Context, orgId string) (members []*members.Member, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("ListMembers")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("ListMembers").Inc() + members, err = m.next.ListMembers(ctx, orgId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("ListMembers").Inc() + return + } + return +} + +// ListOrganizations implements members.Members +func (m metricsMiddleware) ListOrganizations(ctx context.Context, userId string) (organizations []*members.Member, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("ListOrganizations")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("ListOrganizations").Inc() + organizations, err = m.next.ListOrganizations(ctx, userId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("ListOrganizations").Inc() + return + } + return +} + +// Remove implements members.Members +func (m metricsMiddleware) Remove(ctx context.Context, orgId string, userId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Remove")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Remove").Inc() + err = m.next.Remove(ctx, orgId, userId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Remove").Inc() + return + } + return +} + +// RemoveAll implements members.Members +func (m metricsMiddleware) RemoveAll(ctx context.Context, orgId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("RemoveAll")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("RemoveAll").Inc() + err = m.next.RemoveAll(ctx, orgId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("RemoveAll").Inc() + return + } + return +} + +// Set implements members.Members +func (m metricsMiddleware) Set(ctx context.Context, orgId string, userId string, role members.Role) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Set")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Set").Inc() + err = m.next.Set(ctx, orgId, userId, role) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Set").Inc() + return + } + return +} diff --git a/pkg/metrics/request.go b/pkg/metrics/request.go new file mode 100644 index 0000000000000000000000000000000000000000..4c5ff719e4fa85dd33a1a270e2079ae99d644d98 --- /dev/null +++ b/pkg/metrics/request.go @@ -0,0 +1,50 @@ +package metrics + +import ( + "github.com/prometheus/client_golang/prometheus" +) + +type RequestMetrics struct { + Total *prometheus.CounterVec + FailedTotal *prometheus.CounterVec + DurationSeconds *prometheus.HistogramVec +} + +// NewRequestMetrics возвращает метрики для подсчета количества удачных/неудачных запросов, а так же длительности ответов. +// +// subsystem указывает подсистему, к которой принадлежат метрики. +// Значение должно быть уникальным, совпадение разрешено только при совпадении ключей labels. Пустое значение допустимо. +// +// labels - список меток, где каждый элемент метки соответствует парам ключ-значение. Отсутствие допустимо. +// Значения меток должны быть уникальными в рамках одной subsystem. +// +// Метрики записываются в prometheus.DefaultRegisterer +func NewRequestMetrics(subsystem string, durationBuckets []float64, labels ...string) *RequestMetrics { + if len(durationBuckets) == 0 { + durationBuckets = prometheus.DefBuckets + } + metrics := &RequestMetrics{ + Total: prometheus.NewCounterVec(prometheus.CounterOpts{ + Subsystem: subsystem, + Name: "service_requests_total", + Help: "Количество запросов.", + }, []string{"method"}), + FailedTotal: prometheus.NewCounterVec(prometheus.CounterOpts{ + Subsystem: subsystem, + Name: "service_requests_failed_total", + Help: "Количество запросов, вернувших ошибку.", + }, []string{"method"}), + DurationSeconds: prometheus.NewHistogramVec(prometheus.HistogramOpts{ + Subsystem: subsystem, + Name: "service_request_duration_seconds", + Help: "Длительность обработки запроса.", + Buckets: durationBuckets, + }, []string{"method"}), + } + prometheus.WrapRegistererWith(GetLabelsFromKV(labels), prometheus.DefaultRegisterer).MustRegister( + metrics.Total, + metrics.FailedTotal, + metrics.DurationSeconds, + ) + return metrics +} diff --git a/pkg/organizations/middleware/metrics_middleware.go b/pkg/organizations/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..dc1110733b630a3383c6c1dd49493ff237daec00 --- /dev/null +++ b/pkg/organizations/middleware/metrics_middleware.go @@ -0,0 +1,97 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/organizations -i Organizations -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/options" + "git.perx.ru/perxis/perxis-go/pkg/organizations" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements organizations.Organizations that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next organizations.Organizations +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next organizations.Organizations) organizations.Organizations { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements organizations.Organizations +func (m metricsMiddleware) Create(ctx context.Context, org *organizations.Organization) (created *organizations.Organization, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, org) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements organizations.Organizations +func (m metricsMiddleware) Delete(ctx context.Context, orgId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, orgId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Find implements organizations.Organizations +func (m metricsMiddleware) Find(ctx context.Context, filter *organizations.Filter, opts *options.FindOptions) (orgs []*organizations.Organization, total int, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Find")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Find").Inc() + orgs, total, err = m.next.Find(ctx, filter, opts) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Find").Inc() + return + } + return +} + +// Get implements organizations.Organizations +func (m metricsMiddleware) Get(ctx context.Context, orgId string) (org *organizations.Organization, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + org, err = m.next.Get(ctx, orgId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// Update implements organizations.Organizations +func (m metricsMiddleware) Update(ctx context.Context, org *organizations.Organization) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, org) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +} diff --git a/pkg/references/middleware/metrics_middleware.go b/pkg/references/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..520dd7683f81531a9a206cf2e4207cadca302252 --- /dev/null +++ b/pkg/references/middleware/metrics_middleware.go @@ -0,0 +1,58 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/references -i References -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/items" + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/references" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements references.References that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next references.References +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next references.References) references.References { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Get implements references.References +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, envId string, references []*references.Reference) (items []*items.Item, notfound []*references.Reference, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + items, notfound, err = m.next.Get(ctx, spaceId, envId, references) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// Publish implements references.References +func (m metricsMiddleware) Publish(ctx context.Context, spaceId string, envId string, references []*references.Reference, recursive bool, force bool) (published []*references.Reference, notfound []*references.Reference, unpublished []*references.Reference, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Publish")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Publish").Inc() + published, notfound, unpublished, err = m.next.Publish(ctx, spaceId, envId, references, recursive, force) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Publish").Inc() + return + } + return +} diff --git a/pkg/roles/middleware/metrics_middleware.go b/pkg/roles/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..75e242ef5235e55cf9021e06ab7e9d1c9ed7f30c --- /dev/null +++ b/pkg/roles/middleware/metrics_middleware.go @@ -0,0 +1,96 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/roles -i Roles -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/roles" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements roles.Roles that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next roles.Roles +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next roles.Roles) roles.Roles { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements roles.Roles +func (m metricsMiddleware) Create(ctx context.Context, role *roles.Role) (created *roles.Role, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, role) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements roles.Roles +func (m metricsMiddleware) Delete(ctx context.Context, spaceId string, roleId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, spaceId, roleId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Get implements roles.Roles +func (m metricsMiddleware) Get(ctx context.Context, spaceId string, roleId string) (role *roles.Role, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + role, err = m.next.Get(ctx, spaceId, roleId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// List implements roles.Roles +func (m metricsMiddleware) List(ctx context.Context, spaceId string) (roles []*roles.Role, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("List")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("List").Inc() + roles, err = m.next.List(ctx, spaceId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("List").Inc() + return + } + return +} + +// Update implements roles.Roles +func (m metricsMiddleware) Update(ctx context.Context, role *roles.Role) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, role) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +} diff --git a/pkg/spaces/middleware/metrics_middleware.go b/pkg/spaces/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..df81e592174cfc7d3c000ae7d617071d38ab5250 --- /dev/null +++ b/pkg/spaces/middleware/metrics_middleware.go @@ -0,0 +1,161 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/spaces -i Spaces -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/spaces" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements spaces.Spaces that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next spaces.Spaces +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next spaces.Spaces) spaces.Spaces { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// AbortTransfer implements spaces.Spaces +func (m metricsMiddleware) AbortTransfer(ctx context.Context, spaceID string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("AbortTransfer")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("AbortTransfer").Inc() + err = m.next.AbortTransfer(ctx, spaceID) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("AbortTransfer").Inc() + return + } + return +} + +// Create implements spaces.Spaces +func (m metricsMiddleware) Create(ctx context.Context, space *spaces.Space) (created *spaces.Space, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + created, err = m.next.Create(ctx, space) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements spaces.Spaces +func (m metricsMiddleware) Delete(ctx context.Context, spaceId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, spaceId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Get implements spaces.Spaces +func (m metricsMiddleware) Get(ctx context.Context, spaceId string) (space *spaces.Space, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + space, err = m.next.Get(ctx, spaceId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// List implements spaces.Spaces +func (m metricsMiddleware) List(ctx context.Context, orgId string) (spaces []*spaces.Space, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("List")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("List").Inc() + spaces, err = m.next.List(ctx, orgId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("List").Inc() + return + } + return +} + +// ListTransfers implements spaces.Spaces +func (m metricsMiddleware) ListTransfers(ctx context.Context, orgID string) (spaces []*spaces.Space, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("ListTransfers")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("ListTransfers").Inc() + spaces, err = m.next.ListTransfers(ctx, orgID) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("ListTransfers").Inc() + return + } + return +} + +// Move implements spaces.Spaces +func (m metricsMiddleware) Move(ctx context.Context, spaceID string, orgID string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Move")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Move").Inc() + err = m.next.Move(ctx, spaceID, orgID) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Move").Inc() + return + } + return +} + +// Transfer implements spaces.Spaces +func (m metricsMiddleware) Transfer(ctx context.Context, spaceID string, transferToOrg string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Transfer")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Transfer").Inc() + err = m.next.Transfer(ctx, spaceID, transferToOrg) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Transfer").Inc() + return + } + return +} + +// Update implements spaces.Spaces +func (m metricsMiddleware) Update(ctx context.Context, space *spaces.Space) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, space) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +} + +// UpdateConfig implements spaces.Spaces +func (m metricsMiddleware) UpdateConfig(ctx context.Context, spaceId string, config *spaces.Config) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("UpdateConfig")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("UpdateConfig").Inc() + err = m.next.UpdateConfig(ctx, spaceId, config) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("UpdateConfig").Inc() + return + } + return +} diff --git a/pkg/users/middleware/metrics_middleware.go b/pkg/users/middleware/metrics_middleware.go new file mode 100644 index 0000000000000000000000000000000000000000..f1aa52ffc4758a41abe5560f55a1b9ce8616dce7 --- /dev/null +++ b/pkg/users/middleware/metrics_middleware.go @@ -0,0 +1,110 @@ +// Code generated by gowrap. DO NOT EDIT. +// template: ../../../assets/templates/middleware/metrics +// gowrap: http://github.com/hexdigest/gowrap + +package middleware + +//go:generate gowrap gen -p git.perx.ru/perxis/perxis-go/pkg/users -i Users -t ../../../assets/templates/middleware/metrics -o metrics_middleware.go -l "" + +import ( + "context" + + "git.perx.ru/perxis/perxis-go/pkg/metrics" + "git.perx.ru/perxis/perxis-go/pkg/options" + "git.perx.ru/perxis/perxis-go/pkg/users" + "github.com/prometheus/client_golang/prometheus" +) + +// metricsMiddleware implements users.Users that is instrumented with metrics +type metricsMiddleware struct { + requestMetrics *metrics.RequestMetrics + next users.Users +} + +// MetricsMiddleware instruments an implementation of the metricsMiddleware with metrics +func MetricsMiddleware(requestMetrics *metrics.RequestMetrics) Middleware { + return func(next users.Users) users.Users { + return &metricsMiddleware{ + requestMetrics: requestMetrics, + next: next, + } + } +} + +// Create implements users.Users +func (m metricsMiddleware) Create(ctx context.Context, create *users.User) (user *users.User, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Create")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Create").Inc() + user, err = m.next.Create(ctx, create) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Create").Inc() + return + } + return +} + +// Delete implements users.Users +func (m metricsMiddleware) Delete(ctx context.Context, userId string) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Delete")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Delete").Inc() + err = m.next.Delete(ctx, userId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Delete").Inc() + return + } + return +} + +// Find implements users.Users +func (m metricsMiddleware) Find(ctx context.Context, filter *users.Filter, options *options.FindOptions) (users []*users.User, total int, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Find")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Find").Inc() + users, total, err = m.next.Find(ctx, filter, options) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Find").Inc() + return + } + return +} + +// Get implements users.Users +func (m metricsMiddleware) Get(ctx context.Context, userId string) (user *users.User, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Get")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Get").Inc() + user, err = m.next.Get(ctx, userId) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Get").Inc() + return + } + return +} + +// GetByIdentity implements users.Users +func (m metricsMiddleware) GetByIdentity(ctx context.Context, identity string) (user *users.User, err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("GetByIdentity")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("GetByIdentity").Inc() + user, err = m.next.GetByIdentity(ctx, identity) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("GetByIdentity").Inc() + return + } + return +} + +// Update implements users.Users +func (m metricsMiddleware) Update(ctx context.Context, update *users.User) (err error) { + timer := prometheus.NewTimer(m.requestMetrics.DurationSeconds.WithLabelValues("Update")) + defer timer.ObserveDuration() + m.requestMetrics.Total.WithLabelValues("Update").Inc() + err = m.next.Update(ctx, update) + if err != nil { + m.requestMetrics.FailedTotal.WithLabelValues("Update").Inc() + return + } + return +}