mirror of
https://github.com/mainflux/mainflux.git
synced 2025-05-02 22:17:10 +08:00
48e368b59a
* Initial commit Signed-off-by: b1ackd0t <blackd0t@protonmail.com> * Initial commit Signed-off-by: b1ackd0t <blackd0t@protonmail.com> * Fix CI Signed-off-by: b1ackd0t <blackd0t@protonmail.com> * fix tests: add enabled status key Signed-off-by: rodneyosodo <socials@rodneyosodo.com> * start with token Signed-off-by: rodneyosodo <socials@rodneyosodo.com> Signed-off-by: b1ackd0t <blackd0t@protonmail.com> Signed-off-by: 0x6f736f646f <blackd0t@protonmail.com> Signed-off-by: rodneyosodo <socials@rodneyosodo.com> Co-authored-by: rodneyosodo <socials@rodneyosodo.com>
103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package tracing contains middlewares that will add spans
|
|
// to existing traces.
|
|
package tracing
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/mainflux/mainflux/users"
|
|
opentracing "github.com/opentracing/opentracing-go"
|
|
)
|
|
|
|
const (
|
|
saveOp = "save_op"
|
|
retrieveByEmailOp = "retrieve_by_email"
|
|
updatePassword = "update_password"
|
|
members = "members"
|
|
)
|
|
|
|
var _ users.UserRepository = (*userRepositoryMiddleware)(nil)
|
|
|
|
type userRepositoryMiddleware struct {
|
|
tracer opentracing.Tracer
|
|
repo users.UserRepository
|
|
}
|
|
|
|
// UserRepositoryMiddleware tracks request and their latency, and adds spans
|
|
// to context.
|
|
func UserRepositoryMiddleware(repo users.UserRepository, tracer opentracing.Tracer) users.UserRepository {
|
|
return userRepositoryMiddleware{
|
|
tracer: tracer,
|
|
repo: repo,
|
|
}
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) Save(ctx context.Context, user users.User) (string, error) {
|
|
span := createSpan(ctx, urm.tracer, saveOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.Save(ctx, user)
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) UpdateUser(ctx context.Context, user users.User) error {
|
|
span := createSpan(ctx, urm.tracer, saveOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.UpdateUser(ctx, user)
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) RetrieveByEmail(ctx context.Context, email string) (users.User, error) {
|
|
span := createSpan(ctx, urm.tracer, retrieveByEmailOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.RetrieveByEmail(ctx, email)
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) RetrieveByID(ctx context.Context, id string) (users.User, error) {
|
|
span := createSpan(ctx, urm.tracer, retrieveByEmailOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.RetrieveByID(ctx, id)
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) UpdatePassword(ctx context.Context, email, password string) error {
|
|
span := createSpan(ctx, urm.tracer, updatePassword)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.UpdatePassword(ctx, email, password)
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) RetrieveAll(ctx context.Context, ids []string, pm users.PageMetadata) (users.UserPage, error) {
|
|
span := createSpan(ctx, urm.tracer, members)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.RetrieveAll(ctx, ids, pm)
|
|
}
|
|
|
|
func (urm userRepositoryMiddleware) ChangeStatus(ctx context.Context, id, status string) error {
|
|
span := createSpan(ctx, urm.tracer, members)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return urm.repo.ChangeStatus(ctx, id, status)
|
|
}
|
|
|
|
func createSpan(ctx context.Context, tracer opentracing.Tracer, opName string) opentracing.Span {
|
|
if parentSpan := opentracing.SpanFromContext(ctx); parentSpan != nil {
|
|
return tracer.StartSpan(
|
|
opName,
|
|
opentracing.ChildOf(parentSpan.Context()),
|
|
)
|
|
}
|
|
return tracer.StartSpan(opName)
|
|
}
|