-
Notifications
You must be signed in to change notification settings - Fork 126
/
config_tracer.go
60 lines (51 loc) · 2.21 KB
/
config_tracer.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
package evergreen
import (
"context"
"github.com/pkg/errors"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
// TracerConfig configures the OpenTelemetry tracer provider. If not enabled traces will not be sent.
type TracerConfig struct {
Enabled bool `yaml:"enabled" bson:"enabled" json:"enabled"`
CollectorEndpoint string `yaml:"collector_endpoint" bson:"collector_endpoint" json:"collector_endpoint"`
CollectorAPIKey string `yaml:"collector_api_key" bson:"collector_api_key" json:"collector_api_key"`
CollectorInternalEndpoint string `yaml:"collector_internal_endpoint" bson:"collector_internal_endpoint" json:"collector_internal_endpoint"`
}
// SectionId returns the ID of this config section.
func (c *TracerConfig) SectionId() string { return "tracer" }
// Get populates the config from the database.
func (c *TracerConfig) Get(ctx context.Context) error {
res := GetEnvironment().DB().Collection(ConfigCollection).FindOne(ctx, byId(c.SectionId()))
if err := res.Err(); err != nil {
if err == mongo.ErrNoDocuments {
*c = TracerConfig{}
return nil
}
return errors.Wrapf(err, "getting config section '%s'", c.SectionId())
}
if err := res.Decode(&c); err != nil {
return errors.Wrapf(err, "decoding config section '%s'", c.SectionId())
}
return nil
}
// Set sets the document in the database to match the in-memory config struct.
func (c *TracerConfig) Set(ctx context.Context) error {
_, err := GetEnvironment().DB().Collection(ConfigCollection).UpdateOne(ctx, byId(c.SectionId()), bson.M{
"$set": bson.M{
tracerEnabledKey: c.Enabled,
tracerCollectorEndpointKey: c.CollectorEndpoint,
tracerCollectorInternalEndpointKey: c.CollectorInternalEndpoint,
tracerCollectorAPIKeyKey: c.CollectorAPIKey,
},
}, options.Update().SetUpsert(true))
return errors.Wrapf(err, "updating config section '%s'", c.SectionId())
}
// ValidateAndDefault validates the tracer configuration.
func (c *TracerConfig) ValidateAndDefault() error {
if c.Enabled && c.CollectorEndpoint == "" {
return errors.New("tracer can't be enabled without a collector endpoint")
}
return nil
}