Less allocations in mask plugin (#656)

* Add flag --------- Co-authored-by: george pogosyan <[email protected]>
ozontech · Aug 20, 2024 · 9dea20e · 9dea20e
1 parent 2533c1b
commit 9dea20e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/plugin/action/mask/mask.go b/plugin/action/mask/mask.go
@@ -502,7 +502,9 @@ func (p *Plugin) Do(event *pipeline.Event) pipeline.ActionResult {
 		}
 
 		p.sourceBuf = append(p.sourceBuf[:0], value...)
-		p.maskBuf = append(p.maskBuf[:0], p.sourceBuf...)
+		// valueMasked is not the same as maskApplied;
+		// it shows if current node is masked and really needs to be changed
+		valueMasked := false
 		for i := range p.config.Masks {
 			mask := &p.config.Masks[i]
 			if mask.Re != "" && !valueIsCommonMatched {
@@ -515,6 +517,8 @@ func (p *Plugin) Do(event *pipeline.Event) pipeline.ActionResult {
 			if !locApplied {
 				continue
 			}
+			valueMasked = true
+
 			if mask.AppliedField != "" {
 				event.Root.AddFieldNoAlloc(event.Root, mask.AppliedField).MutateToString(mask.AppliedValue)
 			}
@@ -524,7 +528,10 @@ func (p *Plugin) Do(event *pipeline.Event) pipeline.ActionResult {
 
 			maskApplied = true
 		}
-		v.MutateToString(string(p.maskBuf))
+
+		if valueMasked {
+			v.MutateToString(string(p.maskBuf))
+		}
 	}
 
 	if p.config.MaskAppliedField != "" && maskApplied {