mirror of
https://github.com/mainflux/mainflux.git
synced 2025-05-02 22:17:10 +08:00
877005538c
* Add tracing for twin repository Signed-off-by: Darko Draskovic <darko.draskovic@gmail.com> * Add state repository tracer Signed-off-by: Darko Draskovic <darko.draskovic@gmail.com> * Remove unused tracer meant for NATS publisher Signed-off-by: Darko Draskovic <darko.draskovic@gmail.com>
97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package tracing
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/mainflux/mainflux/twins"
|
|
opentracing "github.com/opentracing/opentracing-go"
|
|
)
|
|
|
|
const (
|
|
saveTwinOp = "save_twin"
|
|
updateTwinOp = "update_twin"
|
|
retrieveTwinByIDOp = "retrieve_twin_by_id"
|
|
retrieveAllTwinsOp = "retrieve_all_twins"
|
|
retrieveTwinsByAttributeOp = "retrieve_twins_by_attribute"
|
|
removeTwinOp = "remove_twin"
|
|
)
|
|
|
|
var (
|
|
_ twins.TwinRepository = (*twinRepositoryMiddleware)(nil)
|
|
)
|
|
|
|
type twinRepositoryMiddleware struct {
|
|
tracer opentracing.Tracer
|
|
repo twins.TwinRepository
|
|
}
|
|
|
|
// TwinRepositoryMiddleware tracks request and their latency, and adds spans
|
|
// to context.
|
|
func TwinRepositoryMiddleware(tracer opentracing.Tracer, repo twins.TwinRepository) twins.TwinRepository {
|
|
return twinRepositoryMiddleware{
|
|
tracer: tracer,
|
|
repo: repo,
|
|
}
|
|
}
|
|
|
|
func (trm twinRepositoryMiddleware) Save(ctx context.Context, tw twins.Twin) (string, error) {
|
|
span := createSpan(ctx, trm.tracer, saveTwinOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return trm.repo.Save(ctx, tw)
|
|
}
|
|
|
|
func (trm twinRepositoryMiddleware) Update(ctx context.Context, tw twins.Twin) error {
|
|
span := createSpan(ctx, trm.tracer, updateTwinOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return trm.repo.Update(ctx, tw)
|
|
}
|
|
|
|
func (trm twinRepositoryMiddleware) RetrieveByID(ctx context.Context, id string) (twins.Twin, error) {
|
|
span := createSpan(ctx, trm.tracer, retrieveTwinByIDOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return trm.repo.RetrieveByID(ctx, id)
|
|
}
|
|
|
|
func (trm twinRepositoryMiddleware) RetrieveAll(ctx context.Context, owner string, offset, limit uint64, name string, metadata twins.Metadata) (twins.Page, error) {
|
|
span := createSpan(ctx, trm.tracer, retrieveAllTwinsOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return trm.repo.RetrieveAll(ctx, owner, offset, limit, name, metadata)
|
|
}
|
|
|
|
func (trm twinRepositoryMiddleware) RetrieveByAttribute(ctx context.Context, channel, subtopic string) ([]string, error) {
|
|
span := createSpan(ctx, trm.tracer, retrieveAllTwinsOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return trm.repo.RetrieveByAttribute(ctx, channel, subtopic)
|
|
}
|
|
|
|
func (trm twinRepositoryMiddleware) Remove(ctx context.Context, id string) error {
|
|
span := createSpan(ctx, trm.tracer, removeTwinOp)
|
|
defer span.Finish()
|
|
ctx = opentracing.ContextWithSpan(ctx, span)
|
|
|
|
return trm.repo.Remove(ctx, id)
|
|
}
|
|
|
|
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)
|
|
}
|