-
Notifications
You must be signed in to change notification settings - Fork 38
/
GlobalAttributesSpanAppender.java
112 lines (96 loc) · 3.82 KB
/
GlobalAttributesSpanAppender.java
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package io.opentelemetry.android;
import io.opentelemetry.api.common.Attributes;
import io.opentelemetry.api.common.AttributesBuilder;
import io.opentelemetry.context.Context;
import io.opentelemetry.sdk.trace.ReadWriteSpan;
import io.opentelemetry.sdk.trace.ReadableSpan;
import io.opentelemetry.sdk.trace.SpanProcessor;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.Consumer;
import java.util.function.Supplier;
/**
* A {@link SpanProcessor} implementation that appends a set of {@linkplain Attributes attributes}
* to every span that is exported. The attributes are supplied via Supplier. This Supplier may alter
* its results and return different attributes over time. collection is mutable, and can be updated
* by calling {@link #update(Consumer)}.
*
* <p>This class is internal and is hence not for public use. Its APIs are unstable and can change
* at any time.
*/
public final class GlobalAttributesSpanAppender implements SpanProcessor {
/**
* Returns a new {@link GlobalAttributesSpanAppender} with a given initial attributes.
*
* @param initialState The initial collection of attributes to append to every span.
*/
public static GlobalAttributesSpanAppender create(Attributes initialState) {
return create(() -> initialState);
}
/**
* Returns a new {@link GlobalAttributesSpanAppender} which calls the given supplier to populate
* the global attributes;
*
* @param attributeSupplier a Supplier of Attributes to be placed on every span.
*/
public static GlobalAttributesSpanAppender create(Supplier<Attributes> attributeSupplier) {
return new GlobalAttributesSpanAppender(attributeSupplier);
}
private final AtomicReference<Supplier<Attributes>> attributesSupplier;
private GlobalAttributesSpanAppender(Supplier<Attributes> initialState) {
this.attributesSupplier = new AtomicReference<>(initialState);
}
@Override
public void onStart(Context parentContext, ReadWriteSpan span) {
span.setAllAttributes(getAttributes());
}
private Attributes getAttributes() {
Supplier<Attributes> supplier = attributesSupplier.get();
if (supplier != null) {
Attributes result = supplier.get();
if (result != null) {
return result;
}
}
return Attributes.empty();
}
@Override
public boolean isStartRequired() {
return true;
}
@Override
public void onEnd(ReadableSpan span) {}
@Override
public boolean isEndRequired() {
return false;
}
/**
* Update the global set of attributes to be appended to every span.
*
* <p>Note: Calling this method invalidates the Supplier originally passed to this {@link
* GlobalAttributesSpanAppender} and any other previously updated Supplier.
*
* @param attributesUpdater A function which will update the current set of attributes, by
* operating on a {@link AttributesBuilder} from the current set.
*/
public void update(Consumer<AttributesBuilder> attributesUpdater) {
synchronized (attributesSupplier) {
Attributes oldAttributes = getAttributes();
AttributesBuilder builder = oldAttributes.toBuilder();
attributesUpdater.accept(builder);
Attributes newAttributes = builder.build();
attributesSupplier.set(() -> newAttributes);
}
}
/**
* Replaces the currently configured attributes Supplier with a new one.
*
* @param attributesSupplier Supplier to call to obtain Attributes for every span.
*/
public void update(Supplier<Attributes> attributesSupplier) {
this.attributesSupplier.set(attributesSupplier);
}
}