Bugfix: Ctrl-C won't stop the task.

1 files changed, 8 insertions, 0 deletions

diff --git a/common/errors.go b/common/errors.go
index fa2d492..62899c1 100644
--- a/common/errors.go
+++ b/common/errors.go
@@ -42,6 +42,10 @@ func (e *RecoverableTaskError) Error() string {
 	return fmt.Sprintf("%v: %v", e.message, e.err)
 }
 
+func (e *RecoverableTaskError) Unwrap() error {
+	return e.err
+}
+
 func NewRecoverableTaskError(message string, err error) error {
 	return &RecoverableTaskError{message: message, err: err}
 }
@@ -55,6 +59,10 @@ func (e *UnrecoverableTaskError) Error() string {
 	return fmt.Sprintf("%v: %v", e.message, e.err)
 }
 
+func (e *UnrecoverableTaskError) Unwrap() error {
+	return e.err
+}
+
 func NewUnrecoverableTaskError(message string, err error) error {
 	return &UnrecoverableTaskError{message: message, err: err}
 }