1
#include "source/extensions/tracers/opentelemetry/samplers/parent_based/config.h"
2

            
3
#include "envoy/extensions/tracers/opentelemetry/samplers/v3/parent_based_sampler.pb.validate.h"
4
#include "envoy/server/tracer_config.h"
5

            
6
#include "source/common/config/utility.h"
7
#include "source/extensions/tracers/opentelemetry/samplers/parent_based/parent_based_sampler.h"
8
#include "source/extensions/tracers/opentelemetry/samplers/trace_id_ratio_based/config.h"
9

            
10
namespace Envoy {
11
namespace Extensions {
12
namespace Tracers {
13
namespace OpenTelemetry {
14

            
15
SamplerSharedPtr
16
ParentBasedSamplerFactory::createSampler(const Protobuf::Message& config,
17
1
                                         Server::Configuration::TracerFactoryContext& context) {
18
1
  auto mptr = Envoy::Config::Utility::translateAnyToFactoryConfig(
19
1
      dynamic_cast<const Protobuf::Any&>(config), context.messageValidationVisitor(), *this);
20
1
  const auto& proto_config = MessageUtil::downcastAndValidate<
21
1
      const envoy::extensions::tracers::opentelemetry::samplers::v3::ParentBasedSamplerConfig&>(
22
1
      *mptr, context.messageValidationVisitor());
23
1
  auto& factory =
24
1
      Envoy::Config::Utility::getAndCheckFactory<SamplerFactory>(proto_config.wrapped_sampler());
25
1
  SamplerSharedPtr wrapped_sampler =
26
1
      factory.createSampler(proto_config.wrapped_sampler().typed_config(), context);
27
1
  return std::make_shared<ParentBasedSampler>(config, context, wrapped_sampler);
28
1
}
29

            
30
/**
31
 * Static registration for the Env sampler factory. @see RegisterFactory.
32
 */
33
REGISTER_FACTORY(ParentBasedSamplerFactory, SamplerFactory);
34

            
35
} // namespace OpenTelemetry
36
} // namespace Tracers
37
} // namespace Extensions
38
} // namespace Envoy