支持额外输出日志

main.go

@@ -3,14 +3,18 @@ package main
 import (
 	"flag"
 	"fmt"
+	"io"
+	"log"
+	"os"
 
 	"github.com/y7ut/logtransfer/transfer"
 )
 
-const version = "2.1.1"
+const version = "2.1.2"
 
 var c = flag.String("c", "./logtransfer.conf", "使用配置文件启动")
 var v = flag.Bool("v", false, "查看当前程序版本")
+var logPath = flag.String("log", "./log/runtime.log", "日志输出")
 
 func main() {
 
@@ -22,5 +26,13 @@ func main() {
 		return
 	}
 
+	writerLog, err := os.OpenFile(*logPath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, os.ModeAppend|os.ModePerm)
+	if err != nil {
+		log.Fatalf("create file log.txt failed: %v", err)
+		return
+	}
+	log.Default().SetFlags(log.LstdFlags)
+	log.Default().SetOutput(io.MultiWriter(writerLog, os.Stderr))
+
 	transfer.Run(*c)
 }

transfer/kernel.go

@@ -62,7 +62,7 @@ func Run(confPath string) {
 			for _, topic := range currentTopics {
 
 				Close <- topic
-				log.Printf(" Customer %s unstalling...", topic)
+				log.Printf("Customer %s unstalling...", topic)
 
 			}
 			cancel()
@@ -86,7 +86,7 @@ func CollectorRegister(ctx context.Context) {
 		case closer := <-Close:
 			c, ok := source.GetCustomer(closer)
 			if !ok {
-				log.Printf(" Customer %s unstall Failed ", closer)
+				log.Printf("Customer %s unstall Failed ", closer)
 				break
 			}
 			source.UnstallManger(closer)
@@ -121,7 +121,7 @@ func TopicWatcherHandle() {
 				log.Println("Put topic but not used")
 				err := source.CreateCustomerGroup(topic.Name)
 				if err != nil {
-					log.Printf(" Create Topic Kafka customer group Failed : %s", err)
+					log.Printf("Create Topic Kafka customer group Failed : %s", err)
 					continue
 				}
 				continue
@@ -140,7 +140,7 @@ func TopicWatcherHandle() {
 			// closeWg.Add(1)
 
 			Close <- deleteTopic
-			log.Printf(" Customer %s deleting...", deleteTopic)
+			log.Printf("Customer %s deleting...", deleteTopic)
 			// closeWg.Wait()
 		}
 	}()