From e6aabe44c5dfac34fc0c0a0bfdea3f64e37ee1de Mon Sep 17 00:00:00 2001
From: M09Ic <m09ician@gmail.com>
Date: Tue, 3 Jan 2023 18:27:06 +0800
Subject: [PATCH] =?UTF-8?q?baseline=E6=B7=BB=E5=8A=A0source=E5=AD=97?=
 =?UTF-8?q?=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/pool.go | 1 +
 pkg/baseline.go  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/internal/pool.go b/internal/pool.go
index d2fa208..dd0a721 100644
--- a/internal/pool.go
+++ b/internal/pool.go
@@ -90,6 +90,7 @@ func NewPool(ctx context.Context, config *pkg.Config) (*Pool, error) {
 			}
 		}
 
+		bl.Source = int(unit.source)
 		bl.ReqDepth = unit.depth
 		bl.Spended = time.Since(start).Milliseconds()
 		switch unit.source {
diff --git a/pkg/baseline.go b/pkg/baseline.go
index a1f7edf..c816aa0 100644
--- a/pkg/baseline.go
+++ b/pkg/baseline.go
@@ -85,6 +85,7 @@ type Baseline struct {
 	IsValid      bool       `json:"valid"`
 	IsFuzzy      bool       `json:"fuzzy"`
 	URLs         []string   `json:"urls"`
+	Source       int        `json:"source"`
 	RecuDepth    int        `json:"-"`
 	ReqDepth     int        `json:"depth"`
 	Recu         bool       `json:"-"`