initial commit

tee.go

@@ -0,0 +1,37 @@
+package channel
+
+// Tee returns 2 channels which both receive all values from source.
+// It's basically a copy function for channels
+func Tee[T any](source <-chan T) (<-chan T, <-chan T) {
+	outs := TeeMany(source, 2)
+	return outs[0], outs[1]
+}
+
+// TeeMany returns a given amount of channels which all receive all values from source.
+// It's basically a copy function for channels
+func TeeMany[T any](source <-chan T, amount int) []<-chan T {
+	outputs := make([]chan T, amount)
+	for i := range outputs {
+		outputs[i] = make(chan T, cap(source))
+	}
+
+	go func() {
+		defer func() {
+			for _, out := range outputs {
+				close(out)
+			}
+		}()
+
+		for value := range source {
+			for _, out := range outputs {
+				out <- value
+			}
+		}
+	}()
+
+	readOnlyOutputs := make([]<-chan T, 0, amount)
+	for _, out := range outputs {
+		readOnlyOutputs = append(readOnlyOutputs, out)
+	}
+	return readOnlyOutputs
+}