diff --git a/publisher/htmlElementsCollector.go b/publisher/htmlElementsCollector.go
index e2f8fd2c..b9b0f4e1 100644
--- a/publisher/htmlElementsCollector.go
+++ b/publisher/htmlElementsCollector.go
@@ -122,12 +122,14 @@ func (w *cssClassCollectorWriter) Write(p []byte) (n int, err error) {
continue
}
+ key := s
+
s, tagName := w.insertStandinHTMLElement(s)
el := parseHTMLElement(s)
el.Tag = tagName
w.collector.mu.Lock()
- w.collector.elementSet[s] = true
+ w.collector.elementSet[key] = true
if el.Tag != "" {
w.collector.elements = append(w.collector.elements, el)
}
diff --git a/publisher/htmlElementsCollector_test.go b/publisher/htmlElementsCollector_test.go
index 6b5ef986..ab152928 100644
--- a/publisher/htmlElementsCollector_test.go
+++ b/publisher/htmlElementsCollector_test.go
@@ -99,3 +99,30 @@ func TestClassCollector(t *testing.T) {
}
}
+
+func BenchmarkClassCollectorWriter(b *testing.B) {
+ const benchHTML = `
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+`
+ for i := 0; i < b.N; i++ {
+ w := newHTMLElementsCollectorWriter(newHTMLElementsCollector())
+ fmt.Fprint(w, benchHTML)
+
+ }
+}