diff --git a/decor/on_compete_or_on_abort.go b/decor/on_compete_or_on_abort.go new file mode 100644 index 0000000..bba2920 --- /dev/null +++ b/decor/on_compete_or_on_abort.go @@ -0,0 +1,10 @@ +package decor + +// OnCompleteOrOnAbort wrap decorator. +// Displays provided message on complete or on abort event. +// +// `decorator` Decorator to wrap +// `message` message to display +func OnCompleteOrOnAbort(decorator Decorator, message string) Decorator { + return OnComplete(OnAbort(decorator, message), message) +}