forked from absmach/magistrala
/
states.go
75 lines (58 loc) · 2.15 KB
/
states.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
// 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 (
saveStateOp = "save_state"
updateStateOp = "update_state"
countStatesOp = "count_states"
retrieveAllStatesOp = "retrieve_all_states"
retrieveLastStateOp = "retrieve_states_by_attribute"
)
var _ twins.StateRepository = (*stateRepositoryMiddleware)(nil)
type stateRepositoryMiddleware struct {
tracer opentracing.Tracer
repo twins.StateRepository
}
// StateRepositoryMiddleware tracks request and their latency, and adds spans
// to context.
func StateRepositoryMiddleware(tracer opentracing.Tracer, repo twins.StateRepository) twins.StateRepository {
return stateRepositoryMiddleware{
tracer: tracer,
repo: repo,
}
}
func (trm stateRepositoryMiddleware) Save(ctx context.Context, st twins.State) error {
span := createSpan(ctx, trm.tracer, saveStateOp)
defer span.Finish()
ctx = opentracing.ContextWithSpan(ctx, span)
return trm.repo.Save(ctx, st)
}
func (trm stateRepositoryMiddleware) Update(ctx context.Context, st twins.State) error {
span := createSpan(ctx, trm.tracer, updateStateOp)
defer span.Finish()
ctx = opentracing.ContextWithSpan(ctx, span)
return trm.repo.Update(ctx, st)
}
func (trm stateRepositoryMiddleware) Count(ctx context.Context, tw twins.Twin) (int64, error) {
span := createSpan(ctx, trm.tracer, countStatesOp)
defer span.Finish()
ctx = opentracing.ContextWithSpan(ctx, span)
return trm.repo.Count(ctx, tw)
}
func (trm stateRepositoryMiddleware) RetrieveAll(ctx context.Context, offset, limit uint64, twinID string) (twins.StatesPage, error) {
span := createSpan(ctx, trm.tracer, retrieveAllStatesOp)
defer span.Finish()
ctx = opentracing.ContextWithSpan(ctx, span)
return trm.repo.RetrieveAll(ctx, offset, limit, twinID)
}
func (trm stateRepositoryMiddleware) RetrieveLast(ctx context.Context, twinID string) (twins.State, error) {
span := createSpan(ctx, trm.tracer, retrieveAllStatesOp)
defer span.Finish()
ctx = opentracing.ContextWithSpan(ctx, span)
return trm.repo.RetrieveLast(ctx, twinID)
}