From bfda87826ae37c849488789421b589262b179eed Mon Sep 17 00:00:00 2001
From: M09Ic <m09ician@gmail.com>
Date: Thu, 17 Nov 2022 05:53:04 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E9=94=99=E8=AF=AF=E5=A4=84?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/pool.go  | 6 +++---
 internal/types.go | 6 ++++++
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/internal/pool.go b/internal/pool.go
index a1f4c45..78f0a09 100644
--- a/internal/pool.go
+++ b/internal/pool.go
@@ -299,7 +299,7 @@ func (p *Pool) BaseCompare(bl *pkg.Baseline) {
 
 	if ok {
 		if status = base.Compare(bl); status == 1 {
-			p.PutToInvalid(bl, "compare failed")
+			p.PutToInvalid(bl, ErrCompareFailed.Error())
 			return
 		}
 	}
@@ -307,13 +307,13 @@ func (p *Pool) BaseCompare(bl *pkg.Baseline) {
 	bl.Collect()
 	for _, f := range bl.Frameworks {
 		if f.Tag == "waf/cdn" {
-			p.PutToInvalid(bl, "waf")
+			p.PutToInvalid(bl, ErrWaf.Error())
 			return
 		}
 	}
 
 	if ok && status == 0 && base.FuzzyCompare(bl) {
-		p.PutToInvalid(bl, "fuzzy compare failed")
+		p.PutToInvalid(bl, ErrFuzzyCompareFailed.Error())
 		p.PutToFuzzy(bl)
 		return
 	}
diff --git a/internal/types.go b/internal/types.go
index 123429b..3b04e8d 100644
--- a/internal/types.go
+++ b/internal/types.go
@@ -8,6 +8,8 @@ const (
 	ErrRequestFailed
 	ErrWaf
 	ErrRedirect
+	ErrCompareFailed
+	ErrFuzzyCompareFailed
 )
 
 func (e ErrorType) Error() string {
@@ -22,6 +24,10 @@ func (e ErrorType) Error() string {
 		return "maybe ban of waf"
 	case ErrRedirect:
 		return "duplicate redirect url"
+	case ErrCompareFailed:
+		return "compare failed"
+	case ErrFuzzyCompareFailed:
+		return "fuzzy compare failed"
 	default:
 		return "unknown error"
 	}