103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package authorization
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/influxdata/influxdb/v2/kit/platform"
|
|
|
|
"github.com/influxdata/influxdb/v2"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type AuthLogger struct {
|
|
logger *zap.Logger
|
|
authService influxdb.AuthorizationService
|
|
}
|
|
|
|
// NewAuthLogger returns a logging service middleware for the Authorization Service.
|
|
func NewAuthLogger(log *zap.Logger, s influxdb.AuthorizationService) *AuthLogger {
|
|
return &AuthLogger{
|
|
logger: log,
|
|
authService: s,
|
|
}
|
|
}
|
|
|
|
var _ influxdb.AuthorizationService = (*AuthLogger)(nil)
|
|
|
|
func (l *AuthLogger) CreateAuthorization(ctx context.Context, a *influxdb.Authorization) (err error) {
|
|
defer func(start time.Time) {
|
|
dur := zap.Duration("took", time.Since(start))
|
|
if err != nil {
|
|
l.logger.Debug("failed to create authorization", zap.Error(err), dur)
|
|
return
|
|
}
|
|
l.logger.Debug("authorization create", dur)
|
|
}(time.Now())
|
|
return l.authService.CreateAuthorization(ctx, a)
|
|
}
|
|
|
|
func (l *AuthLogger) FindAuthorizationByID(ctx context.Context, id platform.ID) (a *influxdb.Authorization, err error) {
|
|
defer func(start time.Time) {
|
|
dur := zap.Duration("took", time.Since(start))
|
|
if err != nil {
|
|
msg := fmt.Sprintf("failed to find authorization with ID %v", id)
|
|
l.logger.Debug(msg, zap.Error(err), dur)
|
|
return
|
|
}
|
|
l.logger.Debug("auth find by ID", dur)
|
|
}(time.Now())
|
|
return l.authService.FindAuthorizationByID(ctx, id)
|
|
}
|
|
|
|
func (l *AuthLogger) FindAuthorizationByToken(ctx context.Context, t string) (a *influxdb.Authorization, err error) {
|
|
defer func(start time.Time) {
|
|
dur := zap.Duration("took", time.Since(start))
|
|
if err != nil {
|
|
l.logger.Debug("failed to find authorization with token", zap.Error(err), dur)
|
|
return
|
|
}
|
|
l.logger.Debug("auth find", dur)
|
|
|
|
}(time.Now())
|
|
return l.authService.FindAuthorizationByToken(ctx, t)
|
|
}
|
|
|
|
func (l *AuthLogger) FindAuthorizations(ctx context.Context, filter influxdb.AuthorizationFilter, opt ...influxdb.FindOptions) (as []*influxdb.Authorization, count int, err error) {
|
|
defer func(start time.Time) {
|
|
dur := zap.Duration("took", time.Since(start))
|
|
if err != nil {
|
|
l.logger.Debug("failed to find authorizations matching the given filter", zap.Error(err), dur)
|
|
return
|
|
}
|
|
l.logger.Debug("authorizations find", dur)
|
|
}(time.Now())
|
|
return l.authService.FindAuthorizations(ctx, filter)
|
|
}
|
|
|
|
func (l *AuthLogger) UpdateAuthorization(ctx context.Context, id platform.ID, upd *influxdb.AuthorizationUpdate) (a *influxdb.Authorization, err error) {
|
|
defer func(start time.Time) {
|
|
dur := zap.Duration("took", time.Since(start))
|
|
if err != nil {
|
|
l.logger.Debug("failed to update authorization", zap.Error(err), dur)
|
|
return
|
|
}
|
|
l.logger.Debug("authorization update", dur)
|
|
}(time.Now())
|
|
return l.authService.UpdateAuthorization(ctx, id, upd)
|
|
}
|
|
|
|
func (l *AuthLogger) DeleteAuthorization(ctx context.Context, id platform.ID) (err error) {
|
|
defer func(start time.Time) {
|
|
dur := zap.Duration("took", time.Since(start))
|
|
if err != nil {
|
|
msg := fmt.Sprintf("failed to delete authorization with ID %v", id)
|
|
l.logger.Debug(msg, zap.Error(err), dur)
|
|
return
|
|
}
|
|
l.logger.Debug("authorization delete", dur)
|
|
}(time.Now())
|
|
return l.authService.DeleteAuthorization(ctx, id)
|
|
}
|