63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
|
package tenant
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/influxdata/influxdb"
|
||
|
|
"github.com/influxdata/influxdb/kit/metric"
|
||
|
|
"github.com/influxdata/influxdb/kit/prom"
|
||
|
|
)
|
||
|
|
|
||
|
|
type UserMetrics struct {
|
||
|
|
// RED metrics
|
||
|
|
rec *metric.REDClient
|
||
|
|
|
||
|
|
userService influxdb.UserService
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ influxdb.UserService = (*UserMetrics)(nil)
|
||
|
|
|
||
|
|
// NewUserMetrics returns a metrics service middleware for the User Service.
|
||
|
|
func NewUserMetrics(reg *prom.Registry, s influxdb.UserService) *UserMetrics {
|
||
|
|
return &UserMetrics{
|
||
|
|
rec: metric.New(reg, "user"),
|
||
|
|
userService: s,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *UserMetrics) FindUserByID(ctx context.Context, id influxdb.ID) (*influxdb.User, error) {
|
||
|
|
rec := m.rec.Record("find_user_by_id")
|
||
|
|
user, err := m.userService.FindUserByID(ctx, id)
|
||
|
|
return user, rec(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *UserMetrics) FindUser(ctx context.Context, filter influxdb.UserFilter) (*influxdb.User, error) {
|
||
|
|
rec := m.rec.Record("find_user")
|
||
|
|
user, err := m.userService.FindUser(ctx, filter)
|
||
|
|
return user, rec(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *UserMetrics) FindUsers(ctx context.Context, filter influxdb.UserFilter, opt ...influxdb.FindOptions) ([]*influxdb.User, int, error) {
|
||
|
|
rec := m.rec.Record("find_users")
|
||
|
|
users, n, err := m.userService.FindUsers(ctx, filter, opt...)
|
||
|
|
return users, n, rec(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *UserMetrics) CreateUser(ctx context.Context, u *influxdb.User) error {
|
||
|
|
rec := m.rec.Record("create_user")
|
||
|
|
err := m.userService.CreateUser(ctx, u)
|
||
|
|
return rec(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *UserMetrics) UpdateUser(ctx context.Context, id influxdb.ID, upd influxdb.UserUpdate) (*influxdb.User, error) {
|
||
|
|
rec := m.rec.Record("update_user")
|
||
|
|
updatedUser, err := m.userService.UpdateUser(ctx, id, upd)
|
||
|
|
return updatedUser, rec(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *UserMetrics) DeleteUser(ctx context.Context, id influxdb.ID) error {
|
||
|
|
rec := m.rec.Record("delete_user")
|
||
|
|
err := m.userService.DeleteUser(ctx, id)
|
||
|
|
return rec(err)
|
||
|
|
}
|