support/log: allow setting exit function

Needed for stellar/stellar-rpc#241
stellar · Feb 6, 2025 · da5bd14 · da5bd14
1 parent 5cf7ddc
commit da5bd14
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/support/log/entry.go b/support/log/entry.go
@@ -40,6 +40,10 @@ func (e *Entry) SetLevel(level logrus.Level) {
 	e.entry.Logger.SetLevel(level)
 }
 
+func (e *Entry) SetExitFun(exitFun func(int)) {
+	e.entry.Logger.ExitFunc = exitFun
+}
+
 func (e *Entry) UseJSONFormatter() {
 	formatter := new(logrus.JSONFormatter)
 	formatter.TimestampFormat = timeStampFormat