From e3587d6025ce5a777ba4e7a600c06b1e7e6f7004 Mon Sep 17 00:00:00 2001
From: ehlxr <ehlxr.me@gmail.com>
Date: Mon, 18 Nov 2019 11:07:01 +0800
Subject: [PATCH] init

---
 Makefile | 14 +++++++-------
 main.go  | 27 ++++++++++++++++-----------
 2 files changed, 23 insertions(+), 18 deletions(-)

diff --git a/Makefile b/Makefile
index 3bf3a0b..7f82d11 100644
--- a/Makefile
+++ b/Makefile
@@ -13,11 +13,11 @@ LD_GO_VERSION	:= -X '$(VERSION_PATH).GoVersion=`go version`'
 LD_VERSION		:= -X '$(VERSION_PATH).Version=$(BUILD_VERSION)'
 LD_FLAGS		:= "$(LD_APP_NAMW) $(LD_GIT_COMMIT) $(LD_BUILD_TIME) $(LD_GO_VERSION) $(LD_VERSION) -w -s"
 
-RELEASE_VERSION = $(version)
-REGISTRY_URL    = $(url)
+DOCKER_TAG 		:= $(tag)
+REGISTRY_URL	:= $(url)
 
-ifeq ("$(RELEASE_VERSION)","")
-	RELEASE_VERSION	:= $(shell echo `date "+%Y%m%d_%H%M%S"`)
+ifeq ("$(DOCKER_TAG)","")
+	DOCKER_TAG	:= $(shell echo `date "+%Y%m%d_%H%M%S"`)
 endif
 
 .PHONY : build release clean install upx docker-push docker
@@ -33,15 +33,15 @@ endif
 
 docker: build upx
 ifneq ("$(REGISTRY_URL)","")
-	@echo ========== current docker tag is: $(RELEASE_VERSION) ==========
+	@echo ========== current docker tag is: $(DOCKER_TAG) ==========
 
-	docker build -t $(REGISTRY_URL)/monitor_server:$(RELEASE_VERSION) -f Dockerfile .
+	docker build -t $(REGISTRY_URL)/monitor_server:$(DOCKER_TAG) -f Dockerfile .
 else
 	@echo "url arg should not be empty"
 endif
 
 docker-push: docker
-	docker push $(REGISTRY_URL)/monitor_server:$(RELEASE_VERSION)
+	docker push $(REGISTRY_URL)/monitor_server:$(DOCKER_TAG)
 
 clean:
 	rm -rf $(DIST_DIR)*
diff --git a/main.go b/main.go
index 1d81e0c..eb22a96 100644
--- a/main.go
+++ b/main.go
@@ -33,13 +33,17 @@ GoVersion: %s
 	bannerBase64 = "DQogX19fXyAgX19fXyAgICBfX18gIF9fX19fIA0KKCAgXyBcKCAgXyBcICAvIF9fKSggIF8gICkNCiApKF8pICkpKF8pICkoIChfLS4gKShfKSggDQooX19fXy8oX19fXy8gIFxfX18vKF9fX19fKQ0K"
 
 	opts struct {
-		MonitorFile string `short:"f" long:"monitor-file" env:"MONITOR_FILE" description:"The file to be monitored" required:"true"`
-		KeyWord     string `short:"k" long:"key-word" env:"KEY_WORD" description:"Key word to be filter" required:"true"`
-		WebHookUrl  string `short:"u" long:"webhook-url" env:"URL" description:"Webhook url of dingtalk" required:"true"`
-		Version     bool   `short:"v" long:"version" description:"Show version info"`
+		Version bool    `short:"v" long:"version" description:"Show version info"`
+		Monitor monitor `group:"MONITOR" env-namespace:"MONITOR"`
 	}
 )
 
+type monitor struct {
+	File       string `short:"f" long:"file" env:"FILE" description:"The file to be monitored" required:"true"`
+	KeyWord    string `short:"k" long:"key-word" env:"KEY_WORD" description:"Key word to be filter" required:"true"`
+	WebHookUrl string `short:"u" long:"dt-wh-url" env:"DT_WH_URL" description:"Webhook url of dingtalk" required:"true"`
+}
+
 func init() {
 	initLog()
 }
@@ -47,18 +51,19 @@ func init() {
 func main() {
 	parseArg()
 
-	tf, err := tail.TailFile(opts.MonitorFile,
+	tf, err := tail.TailFile(opts.Monitor.File,
 		tail.Config{
+			ReOpen:   true,
 			Follow:   true,
 			Location: &tail.SeekInfo{Offset: 0, Whence: 2},
 		})
 	if err != nil {
 		log.Fatal("Tail file %+v", err)
 	}
-	log.Info("monitor file %s...", opts.MonitorFile)
+	log.Info("monitor file %s...", opts.Monitor.File)
 
 	for line := range tf.Lines {
-		if ok, _ := regexp.Match(opts.KeyWord, []byte(line.Text)); ok {
+		if ok, _ := regexp.Match(opts.Monitor.KeyWord, []byte(line.Text)); ok {
 			log.Info("%s", dingToInfo(line.Text))
 		}
 	}
@@ -105,15 +110,15 @@ func dingToInfo(msg string) []byte {
 	b, _ := json.Marshal(data)
 
 	log.Info("send to %s data <%s>",
-		opts.WebHookUrl,
+		opts.Monitor.WebHookUrl,
 		b)
 
-	resp, err := http.Post(opts.WebHookUrl,
+	resp, err := http.Post(opts.Monitor.WebHookUrl,
 		"application/json",
 		bytes.NewBuffer(b))
 	if err != nil {
 		log.Error("send request to %s %+v",
-			opts.WebHookUrl,
+			opts.Monitor.WebHookUrl,
 			err)
 
 	}
@@ -123,7 +128,7 @@ func dingToInfo(msg string) []byte {
 
 	body, _ := ioutil.ReadAll(resp.Body)
 	log.Info("send to %s data <%s> result is %s",
-		opts.WebHookUrl,
+		opts.Monitor.WebHookUrl,
 		b,
 		body)
 	return body