[BEAM-3612] Add way to query input registration

apache · Oct 5, 2018 · 7fbf886 · 7fbf886
1 parent 9ac7ad4
commit 7fbf886
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sdks/go/pkg/beam/core/runtime/exec/input.go b/sdks/go/pkg/beam/core/runtime/exec/input.go
@@ -60,6 +60,12 @@ func RegisterInput(t reflect.Type, maker func(ReStream) ReusableInput) {
 	inputs[t] = maker
 }
 
+// IsInputRegistered returns whether an input maker has already been registered.
+func IsInputRegistered(t reflect.Type) bool {
+	_, exists := inputs[t]
+	return exists
+}
+
 type reIterValue struct {
 	t  reflect.Type
 	s  ReStream