diff --git a/cve-vulner-manager/cve-timed-task/tabletask/repeat_task.go b/cve-vulner-manager/cve-timed-task/tabletask/repeat_task.go index f986a1a01131c4431dc27bfa7016e366d200d1c5..a565ffdfc11fbd919f213cbe3b231d56e8d32aae 100644 --- a/cve-vulner-manager/cve-timed-task/tabletask/repeat_task.go +++ b/cve-vulner-manager/cve-timed-task/tabletask/repeat_task.go @@ -1,13 +1,15 @@ package tabletask import ( - "cvevulner/cve-timed-task/db_models" - "cvevulner/models" "fmt" - "github.com/astaxie/beego/logs" - "github.com/astaxie/beego/orm" "strconv" "time" + + "github.com/astaxie/beego/logs" + "github.com/astaxie/beego/orm" + + "cvevulner/cve-timed-task/db_models" + "cvevulner/models" ) // RepeatTask Parse the artificial CVE table, crawl the CVE official @@ -83,18 +85,8 @@ func GetPublishedDateTask() { return } for _, v := range vulnCenter { - url := fmt.Sprintf("https://nvd.nist.gov/vuln/detail/%s", v.CveNum) - crawlList, err := Crawling(url) - if err != nil { - logs.Error("Crawling error:", err, url) - continue - } - rep := GetRepair(v.CveNum) - if len(rep) > 0 { - crawlList.RepairTime = rep - } - if crawlList.RepairTime != "" { - err = db_models.UpdateCveVulnCenter(crawlList.RepairTime, strconv.FormatInt(v.CveId, 10), ormModel) + if rep := GetRepair(v.CveNum); rep != "" { + err = db_models.UpdateCveVulnCenter(rep, strconv.FormatInt(v.CveId, 10), ormModel) if err != nil { logs.Error("db_models.UpdateCveVulnCenter error:", err) continue diff --git a/cve-vulner-manager/cve-timed-task/tabletask/supplement_cve.go b/cve-vulner-manager/cve-timed-task/tabletask/supplement_cve.go index fb654ab3daa23c7f018672157be1e1dd54636dcc..1b15a3d44f2fc171469846bfda3efb5419683a02 100644 --- a/cve-vulner-manager/cve-timed-task/tabletask/supplement_cve.go +++ b/cve-vulner-manager/cve-timed-task/tabletask/supplement_cve.go @@ -110,6 +110,8 @@ func GetRepair(cve string) string { var rep string = "" resp, err := http.Get(fmt.Sprintf(CVEurl, cve)) if err != nil { + logs.Error("get repair time of %s error: %s", cve, err.Error()) + return "" } defer resp.Body.Close()