From 68069dec341e60be577a28ea7aecb2842ec759b9 Mon Sep 17 00:00:00 2001
From: zhiqiang feng <fengzhq@it2000.com.cn>
Date: Mon, 28 Dec 2020 16:36:59 +0800
Subject: [PATCH] add rawlog

---
 cmd/root.go       | 9 +++++++--
 utils/telegram.go | 4 ++++
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/cmd/root.go b/cmd/root.go
index e37fd07..201d2d3 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -58,7 +58,6 @@ Read the telegram from `,
 	Mode = os.Getenv(ModeName)
 
 	logger *zap.Logger
-	// l      *zap.SugaredLogger
 )
 
 // Execute adds all child commands to the root command and sets flags appropriately.
@@ -119,7 +118,7 @@ func initConfig() {
 }
 
 func initLog() {
-	logFile := "./log/telegram-%Y-%m-%d-%H.log"
+	logFile := "./logs/telegram-%Y-%m-%d-%H.log"
 	rotator, err := rotatelogs.New(
 		logFile,
 		rotatelogs.WithMaxAge(60*24*time.Hour),
@@ -128,6 +127,12 @@ func initLog() {
 		panic(err)
 	}
 
+	rawFile := "./logs/raw-%Y-%m-%d-%H.txt"
+	utils.RawLog, err = rotatelogs.New(
+		rawFile,
+		rotatelogs.WithMaxAge(60*24*time.Hour),
+		rotatelogs.WithRotationTime(time.Hour))
+
 	priority := zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
 		return lvl >= zapcore.DebugLevel
 	})
diff --git a/utils/telegram.go b/utils/telegram.go
index 2a4d70a..684c1fe 100644
--- a/utils/telegram.go
+++ b/utils/telegram.go
@@ -3,6 +3,8 @@ package utils
 import (
 	"regexp"
 	"strings"
+
+	rotatelogs "github.com/lestrrat-go/file-rotatelogs"
 )
 
 const (
@@ -11,6 +13,7 @@ const (
 )
 
 var buffer []byte
+var RawLog *rotatelogs.RotateLogs
 
 func Append(data []byte) {
 	buffer = append(buffer, data...)
@@ -28,6 +31,7 @@ func check() {
 
 		if len(telegram) > 0 {
 			// Log.Info("telegram ", zap.String("text", telegram))
+			RawLog.Write([]byte(telegram))
 			buffer = nil
 			InsertTelegram(telegram)
 		}