commit message

common/log.go

@@ -0,0 +1,52 @@
+package common
+
+import (
+	"fmt"
+	"os"
+)
+
+var Results = make(chan string)
+var Woker = 0
+var Start = true
+
+func LogSuccess(result string) {
+	Woker++
+	if Start {
+		go SaveLog()
+		Start = false
+	}
+	Results <- result
+}
+
+func SaveLog() {
+	for result := range Results {
+		fmt.Println(result)
+		if IsSave {
+			WriteFile(result, Outputfile)
+		}
+		Woker--
+	}
+}
+
+func WriteFile(result string, filename string) {
+	var text = []byte(result + "\n")
+	fl, err := os.OpenFile(filename, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0777)
+	if err != nil {
+		fmt.Println("Open %s error, %v", filename, err)
+		return
+	}
+	_, err = fl.Write(text)
+	fl.Close()
+	if err != nil {
+		fmt.Println("write %s error, %v", filename, err)
+	}
+}
+
+func WaitSave() {
+	for {
+		if Woker == 0 {
+			close(Results)
+			return
+		}
+	}
+}