From 08b96819ef1e4299335522ef99aa9d6f7dbf33e4 Mon Sep 17 00:00:00 2001
From: kingecg <kingecg@hotmail.com>
Date: Sat, 2 Dec 2023 20:14:48 +0800
Subject: [PATCH] if got zero value or invalid or nil, not pass through

---
 main.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index cbf0487..4b1fa71 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,9 @@
 package gostream
 
-import "errors"
+import (
+	"errors"
+	"reflect"
+)
 
 type StreamOption struct {
 	WaterMark int // buffer len
@@ -124,7 +127,11 @@ func (tr *TransformStream[T, K]) OnData(cb func(K)) error {
 		go func() {
 			for k := range tr.writer {
 				transfered := tr.ITransformer.Transform(k)
-				tr.reader <- transfered
+				tvalue := reflect.ValueOf(transfered)
+				if !tvalue.IsZero() && !tvalue.IsValid() {
+					tr.reader <- transfered
+				}
+
 			}
 		}()
 	}