fix: fix tags

mroyme · Aug 15, 2024 · fc18f4a · fc18f4a
1 parent b7af3dd
commit fc18f4a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/cmd/dogstatsd-local/main.go b/cmd/dogstatsd-local/main.go
@@ -32,7 +32,10 @@ func main() {
 	debug := flag.Bool("debug", false, "Enable debug mode")
 	flag.Parse()
 
-	extraTags := strings.Split(*rawTags, ",")
+	var extraTags []string
+	if *rawTags != "" {
+		extraTags = strings.Split(*rawTags, ",")
+	}
 	logger := getLogger(*debug)
 
 	var formatHandler format.Handler

diff --git a/internal/format/pretty/pretty.go b/internal/format/pretty/pretty.go
@@ -119,9 +119,5 @@ func (h *Handler) StyledTags(metric messages.DogStatsDMetric, extraTags []string
 	style := lipgloss.NewStyle().
 		Foreground(h.Theme.Overlay0()).
 		Italic(true)
-	var tags []string
-	for _, tag := range append(extraTags, metric.Tags...) {
-		tags = append(tags, strings.TrimSpace(tag))
-	}
-	return style.SetString(tags...).Render()
+	return style.SetString(append(extraTags, metric.Tags...)...).Render()
 }
diff --git a/internal/messages/messages.go b/internal/messages/messages.go
@@ -171,6 +171,9 @@ func parseDogStatsDMetricMessage(buf []byte) (DogStatsDMessage, error) {
 
 		if strings.HasPrefix(piece, "#") {
 			tags := strings.Split(piece[1:], ",")
+			for i, _ := range tags {
+				tags[i] = strings.TrimSpace(tags[i])
+			}
 			metric.Tags = append(metric.Tags, tags...)
 			continue
 		}