summaryrefslogtreecommitdiff
path: root/classes/tracer.php
blob: d788cfdbab722023d58339791b925d44033dac97 (plain)
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
<?php

use OpenTelemetry\Contrib\Otlp\OtlpHttpTransportFactory;
use OpenTelemetry\Contrib\Otlp\SpanExporter;
use OpenTelemetry\SDK\Trace\SpanProcessor\SimpleSpanProcessor;
use OpenTelemetry\SDK\Trace\TracerProvider;
use OpenTelemetry\API\Trace\Propagation\TraceContextPropagator;
use OpenTelemetry\API\Trace\Span;
use OpenTelemetry\SDK\Trace\SpanExporter\InMemoryExporter;

class Tracer {
	/** @var Tracer $instance */
	private static $instance;

	/** @var OpenTelemetry\API\Trace\TracerInterface $tracer */
	private static $tracer;

	public function __construct() {
		$opentelemetry_host = Config::get(Config::OPENTELEMETRY_HOST);

		if ($opentelemetry_host) {
			$transport = (new OtlpHttpTransportFactory())->create("http://$opentelemetry_host/v1/traces", 'application/x-protobuf');
			$exporter = new SpanExporter($transport);
		} else {
			$exporter = new InMemoryExporter();
		}

		$tracerProvider =  new TracerProvider(new SimpleSpanProcessor($exporter));
		$this->tracer = $tracerProvider->getTracer('io.opentelemetry.contrib.php');

		$context = TraceContextPropagator::getInstance()->extract(getallheaders());
		$span = $this->tracer->spanBuilder(Config::get(Config::OPENTELEMETRY_SERVICE))
			->setParent($context)
			->startSpan();

		$span->activate();

		register_shutdown_function(function() use ($span, $tracerProvider) {
			$span->end();

			$tracerProvider->shutdown();
		});
	}

	/**
	 * @param string $name
	 * @param array<string>|array<string, array<string, mixed>> $tags
	 * @param array<string> $args
	 * @return OpenTelemetry\API\Trace\SpanInterface
	 */
	private function _start(string $name, array $tags = [], array $args = []) {
		$span = $this->tracer->spanBuilder($name)->startSpan();

		foreach ($tags as $k => $v) {
			$span->setAttribute($k, $v);
		}

		$span->setAttribute("func.args", json_encode($args));

		$span->activate();

		return $span;
	}

	/**
	 * @param string $name
	 * @param array<string>|array<string, array<string, mixed>> $tags
	 * @param array<string> $args
	 * @return OpenTelemetry\API\Trace\SpanInterface
	 */
	public static function start(string $name, array $tags = [], array $args = []) {
		return self::get_instance()->_start($name, $tags, $args);
	}

	public static function get_instance() : Tracer {
		if (self::$instance == null)
			self::$instance = new self();

		return self::$instance;
	}
}