From 7893d763d1b1d5eeeb62d8a4beb2175acaa13b65 Mon Sep 17 00:00:00 2001
From: Dmitrii Stepanov <d.stepanov@yadro.com>
Date: Thu, 13 Mar 2025 10:37:29 +0300
Subject: [PATCH] [#1673] logger: Add sampling for journald logger

Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
---
 pkg/util/logger/logger.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/pkg/util/logger/logger.go b/pkg/util/logger/logger.go
index a0b2728c9..2eb5e5538 100644
--- a/pkg/util/logger/logger.go
+++ b/pkg/util/logger/logger.go
@@ -2,6 +2,7 @@ package logger
 
 import (
 	"fmt"
+	"time"
 
 	"git.frostfs.info/TrueCloudLab/zapjournald"
 	"github.com/ssgreg/journald"
@@ -166,7 +167,18 @@ func newJournaldLogger(prm *Prm) (*Logger, error) {
 		zapjournald.SyslogPid(),
 	})
 
-	lZap := zap.New(coreWithContext, zap.AddStacktrace(zap.NewAtomicLevelAt(zap.FatalLevel)), zap.AddCallerSkip(1))
+	var samplerOpts []zapcore.SamplerOption
+	if c.Sampling.Hook != nil {
+		samplerOpts = append(samplerOpts, zapcore.SamplerHook(c.Sampling.Hook))
+	}
+	samplingCore := zapcore.NewSamplerWithOptions(
+		coreWithContext,
+		time.Second,
+		c.Sampling.Initial,
+		c.Sampling.Thereafter,
+		samplerOpts...,
+	)
+	lZap := zap.New(samplingCore, zap.AddStacktrace(zap.NewAtomicLevelAt(zap.FatalLevel)), zap.AddCallerSkip(1))
 
 	l := &Logger{z: lZap, lvl: lvl}
 	prm._log = l