소스 검색

Enable poller

Brendan Abolivier 6 년 전
부모
커밋
c264f267df
로그인 계정: Brendan Abolivier <contact@brendanabolivier.com> GPG 키 ID: 8EF1500759F70623
1개의 변경된 파일10개의 추가작업 그리고 1개의 파일을 삭제
  1. 10
    1
      src/pusher/main.go

+ 10
- 1
src/pusher/main.go 파일 보기

@@ -6,6 +6,7 @@ import (
6 6
 	"config"
7 7
 	"grafana"
8 8
 	"logger"
9
+	"pusher/poller"
9 10
 	"pusher/webhook"
10 11
 
11 12
 	"github.com/sirupsen/logrus"
@@ -30,7 +31,15 @@ func main() {
30 31
 
31 32
 	grafanaClient := grafana.NewClient(cfg.Grafana.BaseURL, cfg.Grafana.APIKey)
32 33
 
33
-	if err = webhook.Setup(cfg, grafanaClient, *deleteRemoved); err != nil {
34
+	switch cfg.Pusher.Mode {
35
+	case "webhook":
36
+		err = webhook.Setup(cfg, grafanaClient, *deleteRemoved)
37
+		break
38
+	case "git-pull":
39
+		err = poller.Setup(cfg, grafanaClient, *deleteRemoved)
40
+	}
41
+
42
+	if err != nil {
34 43
 		logrus.Panic(err)
35 44
 	}
36 45
 }