optimized prompt message

Greyh4t · Jun 10, 2020 · e0dd098 · e0dd098
1 parent 6cec87f
commit e0dd098
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -193,7 +193,7 @@ func download(in interface{}) {
 
 	statusCode, data, err := ZHTTP.Get(segment.URI, headers, conf.Retry)
 	if err != nil {
-		log.Fatalln("[-] Download failed:", err)
+		log.Fatalln("[-] Download failed:", id, err)
 	}
 
 	if statusCode/100 != 2 || len(data) == 0 {
@@ -231,7 +231,7 @@ func download(in interface{}) {
 		}
 	}
 
-	log.Println("[+] Download succed:", segment.URI)
+	log.Println("[+] Download succed:", id, segment.URI)
 
 	JOINER.Join(id, data)
 }
@@ -266,6 +266,8 @@ func main() {
 		log.Fatalln("[-] Init failed:", err)
 	}
 
+	t := time.Now()
+
 	var data []byte
 	if conf.File != "" {
 		data, err = ioutil.ReadFile(conf.File)
@@ -307,6 +309,6 @@ func main() {
 		if err != nil {
 			log.Fatalln("[-] Write to file failed:", err)
 		}
-		log.Println("[+] Download succed, saved to", JOINER.Name())
+		log.Println("[+] Download succed, saved to", JOINER.Name(), "cost:", time.Now().Sub(t))
 	}
 }