Fix: detecting goroutine leak (#191)

* test for detecting goroutine leak

* fix goroutine leak

input_spliter.go

@@ -24,13 +24,16 @@ func split(ctx context.Context, r io.Reader) (<-chan string, <-chan error) {
 		for sc.Scan() {
 			select {
 			case <-ctx.Done():
-				errc <- ctx.Err()
 				return
 			default:
 				l := sc.Text()
 				if isRootBlockBeginning(l) {
 					if len(block) != 0 {
-						blockc <- block
+						select {
+						case <-ctx.Done():
+							return
+						case blockc <- block:
+						}
 					}
 					block = ""
 				}
@@ -41,8 +44,12 @@ func split(ctx context.Context, r io.Reader) (<-chan string, <-chan error) {
 			errc <- err
 			return
 		}
-		blockc <- block // 最後のRootブロック送出
-		return
+		select {
+		case <-ctx.Done():
+			return
+		case blockc <- block: // 最後のRootブロック送出
+			return
+		}
 	}()
 
 	return blockc, errc

pipeline_tree_grower.go

@@ -57,7 +57,11 @@ func (dg *defaultGrowerPipeline) worker(ctx context.Context, wg *sync.WaitGroup,
 				errc <- err
 				return
 			}
-			nodes <- root
+			select {
+			case <-ctx.Done():
+				return
+			case nodes <- root:
+			}
 		}
 	}
 }

root_generator.go

@@ -134,8 +134,11 @@ func (rg *rootGeneratorPipeline) worker(ctx context.Context, wg *sync.WaitGroup,
 				errc <- err
 				return
 			}
-
-			rootc <- root
+			select {
+			case <-ctx.Done():
+				return
+			case rootc <- root:
+			}
 		}
 	}
 }

tree_handler_output_test.go

@@ -10,6 +10,7 @@ import (
 	"os"
 	"strings"
 	"testing"
+	"time"
 
 	"github.com/ddddddO/gtree"
 	tu "github.com/ddddddO/gtree/testutil"
@@ -1192,3 +1193,15 @@ children:
 		})
 	}
 }
+
+func TestOutput_detecting_goroutinelerk(t *testing.T) {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Duration(500*time.Millisecond))
+	defer cancel()
+	w := &strings.Builder{}
+	r := strings.NewReader(tu.TwentyThousandRoots)
+	if gotErr := gtree.Output(w, r, gtree.WithMassive(ctx)); gotErr != nil {
+		if gotErr != context.DeadlineExceeded {
+			t.Errorf("\ngotErr: \n%v\nwantErr: \n%v", gotErr, context.DeadlineExceeded)
+		}
+	}
+}