Fix missing ref to new samplers

honeycombio · Jun 20, 2023 · 078840e · 078840e
1 parent a9533a9
commit 078840e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sample/sample.go b/sample/sample.go
@@ -50,6 +50,10 @@ func (s *SamplerFactory) GetSamplerImplementationForKey(samplerKey string, isLeg
 		sampler = &RulesBasedSampler{Config: c, Logger: s.Logger, Metrics: s.Metrics}
 	case *config.TotalThroughputSamplerConfig:
 		sampler = &TotalThroughputSampler{Config: c, Logger: s.Logger, Metrics: s.Metrics}
+	case *config.EMAThroughputSamplerConfig:
+		sampler = &EMAThroughputSampler{Config: c, Logger: s.Logger, Metrics: s.Metrics}
+	case *config.WindowedThroughputSamplerConfig:
+		sampler = &WindowedThroughputSampler{Config: c, Logger: s.Logger, Metrics: s.Metrics}
 	default:
 		s.Logger.Error().Logf("unknown sampler type %T. Exiting.", c)
 		os.Exit(1)