perf: 使用协程

Author: msojocs

tool/fe/fe.go

@@ -2,7 +2,6 @@ package fe
 
 import (
 	"bufio"
-	"errors"
 	"fe-tool/common"
 	"io"
 	"log"
@@ -13,20 +12,6 @@ import (
 const TargetDir = "FiddlerEverywhere"
 
 func Download() {
-	_, err := os.Stat("cache")
-	if err != nil {
-		if errors.Is(err, os.ErrNotExist) {
-			// 不存在
-			err := os.Mkdir("cache", 0755)
-			if err != nil {
-				log.Fatalln("Create dir error:", err)
-			} else {
-				log.Println("Create dir ok.")
-			}
-		} else {
-			log.Fatalln("Check cache dir error:", err)
-		}
-	}
 	if s, err := os.Stat("cache/fe.exe"); err == nil && !s.IsDir() {
 		log.Println("cache/fe.exe exists.")
 		return

tool/main.go

@@ -1,19 +1,47 @@
 package main
 
 import (
+	"errors"
 	"fe-tool/fe"
 	"fe-tool/patch"
 	"fe-tool/server"
+	"log"
+	"os"
+	"sync"
 )
 
 func main() {
+
+	_, err := os.Stat("cache")
+	if err != nil {
+		if errors.Is(err, os.ErrNotExist) {
+			// 不存在
+			err := os.Mkdir("cache", 0755)
+			if err != nil {
+				log.Fatalln("Create dir error:", err)
+			} else {
+				log.Println("Create dir ok.")
+			}
+		} else {
+			log.Fatalln("Check cache dir error:", err)
+		}
+	}
 	// 1.下载FE，解压
-	fe.Download()
-	server.Download()
-	patch.Download()
-	fe.Extract()
-	// 3.下载server数据，解压
-	server.Extract()
+	sg := sync.WaitGroup{}
+	sg.Add(3)
+	go func() {
+		fe.Download()
+		fe.Extract()
+		sg.Done()
+	}()
+	go func() {
+		// 3.下载server数据，解压
+		server.Download()
+		server.Extract()
+		sg.Done()
+	}()
+	go patch.Download(&sg)
+	sg.Wait()
 	// 4.patch
 	patch.Apply()
 }

tool/patch/patch.go

@@ -7,6 +7,7 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"sync"
 )
 
 func Apply() {
@@ -55,7 +56,7 @@ func Apply() {
 	log.Println("Apply end.")
 }
 
-func Download() {
+func Download(sg *sync.WaitGroup) {
 	_, err := os.Stat("cache")
 	if err != nil {
 		if errors.Is(err, os.ErrNotExist) {
@@ -99,4 +100,5 @@ func Download() {
 		log.Fatalln("Rename fiddler.dll.tmp error", err)
 	}
 	log.Println("Download end, file size:", fileSize)
+	sg.Done()
 }

tool/server/server.go

@@ -2,7 +2,6 @@ package server
 
 import (
 	"bufio"
-	"errors"
 	"fe-tool/common"
 	"io"
 	"log"
@@ -13,20 +12,6 @@ import (
 const TargetDir = "."
 
 func Download() {
-	_, err := os.Stat("cache")
-	if err != nil {
-		if errors.Is(err, os.ErrNotExist) {
-			// 不存在
-			err := os.Mkdir("cache", 0755)
-			if err != nil {
-				log.Fatalln("Create dir error:", err)
-			} else {
-				log.Println("Create dir ok.")
-			}
-		} else {
-			log.Fatalln("Check cache dir error:", err)
-		}
-	}
 	if s, err := os.Stat("cache/server.zip"); err == nil && !s.IsDir() {
 		log.Println("cache/server.zip exists.")
 		return