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
|
<?php
declare(strict_types=1);
namespace OpenTelemetry\API\Baggage\Propagation;
use OpenTelemetry\API\Baggage\Baggage;
use OpenTelemetry\API\Baggage\BaggageBuilderInterface;
use OpenTelemetry\API\Baggage\Entry; /** @phan-suppress-current-line PhanUnreferencedUseNormal */
use OpenTelemetry\Context\Context;
use OpenTelemetry\Context\ContextInterface;
use OpenTelemetry\Context\Propagation\ArrayAccessGetterSetter;
use OpenTelemetry\Context\Propagation\PropagationGetterInterface;
use OpenTelemetry\Context\Propagation\PropagationSetterInterface;
use OpenTelemetry\Context\Propagation\TextMapPropagatorInterface;
use function rtrim;
use function urlencode;
/**
* @see https://www.w3.org/TR/baggage
*/
final class BaggagePropagator implements TextMapPropagatorInterface
{
public const BAGGAGE = 'baggage';
private static ?self $instance = null;
public static function getInstance(): self
{
if (null === self::$instance) {
self::$instance = new self();
}
return self::$instance;
}
public function fields(): array
{
return [self::BAGGAGE];
}
public function inject(&$carrier, PropagationSetterInterface $setter = null, ContextInterface $context = null): void
{
$setter ??= ArrayAccessGetterSetter::getInstance();
$context ??= Context::getCurrent();
$baggage = Baggage::fromContext($context);
if ($baggage->isEmpty()) {
return;
}
$headerString = '';
/** @var Entry $entry */
foreach ($baggage->getAll() as $key => $entry) {
$value = urlencode($entry->getValue());
$headerString.= "{$key}={$value}";
if (($metadata = $entry->getMetadata()->getValue()) !== '' && ($metadata = $entry->getMetadata()->getValue()) !== '0') {
$headerString .= ";{$metadata}";
}
$headerString .= ',';
}
if ($headerString !== '' && $headerString !== '0') {
$headerString = rtrim($headerString, ',');
$setter->set($carrier, self::BAGGAGE, $headerString);
}
}
public function extract($carrier, PropagationGetterInterface $getter = null, ContextInterface $context = null): ContextInterface
{
$getter ??= ArrayAccessGetterSetter::getInstance();
$context ??= Context::getCurrent();
if (!$baggageHeader = $getter->get($carrier, self::BAGGAGE)) {
return $context;
}
$baggageBuilder = Baggage::getBuilder();
$this->extractValue($baggageHeader, $baggageBuilder);
return $context->withContextValue($baggageBuilder->build());
}
private function extractValue(string $baggageHeader, BaggageBuilderInterface $baggageBuilder): void
{
(new Parser($baggageHeader))->parseInto($baggageBuilder);
}
}
|
