flusher without err

Author: Flash-LHR

sdks/community/go/pkg/encoding/sse/writer.go

@@ -67,6 +67,9 @@ func (w *SSEWriter) WriteBytes(ctx context.Context, writer io.Writer, event []by
 			return fmt.Errorf("SSE flush failed: %w", err)
 		}
 	}
+	if flusher, ok := writer.(flusherWithoutError); ok {
+		flusher.Flush()
+	}
 	return nil
 }
 
@@ -116,6 +119,9 @@ func (w *SSEWriter) WriteEventWithType(ctx context.Context, writer io.Writer, ev
 			return fmt.Errorf("SSE flush failed: %w", err)
 		}
 	}
+	if flusher, ok := writer.(flusherWithoutError); ok {
+		flusher.Flush()
+	}
 
 	return nil
 }
@@ -187,6 +193,11 @@ type flusher interface {
 	Flush() error
 }
 
+// flusherWithoutError interface for writers that support flushing
+type flusherWithoutError interface {
+	Flush()
+}
+
 // CustomEvent is a simple implementation of events.Event for error and custom events
 type CustomEvent struct {
 	events.BaseEvent