extract and machine learning support

dmachard · Oct 26, 2023 · 38f48b9 · 38f48b9
1 parent 01322d2
commit 38f48b9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/transformers/extract.go b/transformers/extract.go
@@ -31,6 +31,10 @@ func NewExtractSubprocessor(config *dnsutils.ConfigTransformers, logger *logger.
 	return s
 }
 
+func (p *ExtractProcessor) ReloadConfig(config *dnsutils.ConfigTransformers) {
+	p.config = config
+}
+
 func (p *ExtractProcessor) InitDnsMessage(dm *dnsutils.DnsMessage) {
 	if dm.Extracted == nil {
 		dm.Extracted = &dnsutils.TransformExtracted{

diff --git a/transformers/machinelearning.go b/transformers/machinelearning.go
@@ -44,6 +44,10 @@ func NewMachineLearningSubprocessor(config *dnsutils.ConfigTransformers, logger
 	return s
 }
 
+func (p *MlProcessor) ReloadConfig(config *dnsutils.ConfigTransformers) {
+	p.config = config
+}
+
 func (p *MlProcessor) LogInfo(msg string, v ...interface{}) {
 	log := fmt.Sprintf("transformer=ml#%d - ", p.instance)
 	p.logInfo(log+msg, v...)

diff --git a/transformers/subprocessors.go b/transformers/subprocessors.go
@@ -64,6 +64,8 @@ func (p *Transforms) ReloadConfig(config *dnsutils.ConfigTransformers) {
 	p.LatencyTransform.ReloadConfig(config)
 	p.SuspiciousTransform.ReloadConfig(config)
 	p.ReducerTransform.ReloadConfig(config)
+	p.ExtractProcessor.ReloadConfig(config)
+	p.MachineLearningTransform.ReloadConfig(config)
 
 	p.Prepare()
 }