diff --git a/decor/on_abort.go b/decor/on_abort.go index 7ece120..10ff670 100644 --- a/decor/on_abort.go +++ b/decor/on_abort.go @@ -9,6 +9,9 @@ // `message` message to display on abort event // func OnAbort(decorator Decorator, message string) Decorator { + if decorator == nil { + return nil + } d := &onAbortWrapper{ Decorator: decorator, msg: message, diff --git a/decor/on_complete.go b/decor/on_complete.go index c7c9e4d..2ada2b3 100644 --- a/decor/on_complete.go +++ b/decor/on_complete.go @@ -8,6 +8,9 @@ // `message` message to display on complete event // func OnComplete(decorator Decorator, message string) Decorator { + if decorator == nil { + return nil + } d := &onCompleteWrapper{ Decorator: decorator, msg: message,