Add the ErrorIf function for Logger interface (#25)

base_logger.go

@@ -36,6 +36,10 @@ func (l *baseLogger) Error(err error, format string, args ...any) {
 	l.logWithErr(err, level.ErrorLevel, format, args...)
 }
 
+func (l *baseLogger) ErrorIf(err error, format string, args ...any) error {
+	return errorIf(l.Error, err, format, args...)
+}
+
 // Log write a format log
 func (l *baseLogger) Log(format string, args ...any) {
 	format = formatter.AppendRowTerminator(format)
@@ -96,3 +100,10 @@ func (l *baseLogger) setTimeFormat(f string) {
 func checkLogLevel(lvl level.Level, currentLevel level.Level) bool {
 	return currentLevel >= lvl
 }
+
+func errorIf(f func(err error, format string, args ...any), err error, format string, args ...any) error {
+	if err != nil {
+		f(err, format, args...)
+	}
+	return err
+}

default_logger.go

@@ -54,10 +54,7 @@ func Error(err error, format string, args ...any) {
 
 // ErrorIf write the error log if err is not nil
 func ErrorIf(err error, format string, args ...any) error {
-	if err != nil {
-		Error(err, format, args...)
-	}
-	return err
+	return DefaultLogger().ErrorIf(err, format, args...)
 }
 
 // DebugSample write the debug log by random sampling
@@ -82,10 +79,7 @@ func ErrorSample(err error, format string, args ...any) {
 
 // ErrorIfSample write the error log by random sampling if err is not nil
 func ErrorIfSample(err error, format string, args ...any) error {
-	if err != nil {
-		ErrorSample(err, format, args...)
-	}
-	return err
+	return DefaultSampleLogger().ErrorIf(err, format, args...)
 }
 
 // Log write the log without level

empty_logger.go

@@ -29,6 +29,10 @@ func (l *emptyLogger) Error(err error, format string, args ...any) {
 
 }
 
+func (l *emptyLogger) ErrorIf(err error, format string, args ...any) error {
+	return err
+}
+
 func (l *emptyLogger) Log(format string, args ...any) {
 
 }

logger.go

@@ -13,10 +13,16 @@ type Logger interface {
 	Writer
 	Option
 
+	// Debug write the debug log
 	Debug(format string, args ...any)
+	// Info write the info log
 	Info(format string, args ...any)
+	// Warn write the warn log
 	Warn(format string, args ...any)
+	// Error write the error log
 	Error(err error, format string, args ...any)
+	// ErrorIf write the error log if err is not nil
+	ErrorIf(err error, format string, args ...any) error
 }
 
 // Writer implement write to log

multi_logger.go

@@ -43,6 +43,10 @@ func (l *multiLogger) Error(err error, format string, args ...any) {
 	}
 }
 
+func (l *multiLogger) ErrorIf(err error, format string, args ...any) error {
+	return errorIf(l.Error, err, format, args...)
+}
+
 func (l *multiLogger) Log(format string, args ...any) {
 	for _, logger := range l.loggers {
 		logger.Log(format, args...)

sample_logger.go

@@ -50,6 +50,10 @@ func (l *sampleLogger) Error(err error, format string, args ...any) {
 	}
 }
 
+func (l *sampleLogger) ErrorIf(err error, format string, args ...any) error {
+	return errorIf(l.Error, err, format, args...)
+}
+
 func (l *sampleLogger) Log(format string, args ...any) {
 	l.logger.Log(format, args...)
 }