From 7d8ec5f208c645fef71e2b1fd9ce1f891edfda45 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Fri, 16 Sep 2022 20:26:59 +0800
Subject: Bugfix: Ctrl-C won't stop the task.

---
 common/errors.go | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'common')

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}
 }
-- 
cgit v1.2.3