add tag to exclude

TelkomIndonesia · Jul 12, 2023 · 7be29e6 · 7be29e6
1 parent f65b8bb
commit 7be29e6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/config.go b/config.go
@@ -11,6 +11,7 @@ var maxFetchRange = strToDurationOrDefault(os.Getenv("HWC_LOGSTREAM_MAX_FETCH_RA
 var minFetchRange = strToDurationOrDefault(os.Getenv("HWC_LOGSTREAM_MIN_FETCH_RANGE"), time.Minute)
 var streamRoutine = int(strToIntOrDefault(os.Getenv("HWC_LOGSTREAM_ROUTINE"), 5))
 var streamPosTag = envOrDefault("HWC_LOGSTREAM_POSITITION_TAG", "x-hwc-logstream-pos")
+var streamExclusionTag = envOrDefault("HWC_LOGSTREAM_EXCLUSION_TAG", "x-hwc-logstream-exclude")
 
 func envOrDefault(name, def string) string {
 	v, ok := os.LookupEnv(name)

diff --git a/manager.go b/manager.go
@@ -102,6 +102,11 @@ func (m LogstreamManager) SyncStreamList(ctx context.Context) (err error) {
 		}
 
 		for _, stream := range *streams.LogStreams {
+			if _, ok := stream.Tag[streamExclusionTag]; ok {
+				log.Printf("[INFO] stream '%s' of group '%s' is excluded.\n", group.LogGroupName, stream.LogStreamName)
+				continue
+			}
+
 			s := NewLogstream(m.client, group, stream, stdout)
 			fetched[s.ID()] = struct{}{}
 			if v, ok := m.streams[s.ID()]; !ok {