feat: add benchmark test for sse (#7)

* feat: add benchmark for sse

* update benchmarks

sse_test.go

@@ -106,6 +106,20 @@ func TestNewStream(t *testing.T) {
 	assert.NotNil(t, s)
 }
 
+func BenchmarkNewStream(b *testing.B) {
+	var c app.RequestContext
+	var s *Stream
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		s = NewStream(&c)
+	}
+
+	assert.DeepEqual(b, ContentType, string(c.Response.Header.ContentType()))
+	assert.DeepEqual(b, noCache, c.Response.Header.Get(cacheControl))
+	assert.NotNil(b, c.Response.GetHijackWriter())
+	assert.NotNil(b, s)
+}
+
 type BufferExtWriter struct {
 	buffer *bytes.Buffer
 }
@@ -136,6 +150,21 @@ func TestStreamPublish(t *testing.T) {
 	assert.DeepEqual(t, "data:hertz\n\n", buffer.String())
 }
 
+func BenchmarkStream_Publish(b *testing.B) {
+	var c app.RequestContext
+	s := NewStream(&c)
+	var buffer bytes.Buffer
+	s.w = &BufferExtWriter{
+		buffer: &buffer,
+	}
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		s.Publish(&Event{
+			Data: []byte("hertz"),
+		})
+	}
+}
+
 func TestLastEventID(t *testing.T) {
 	var c app.RequestContext
 	c.Request.Header.Set(LastEventID, "1")