Merge pull request #6597 from lostluck/patch-15

[BEAM-3612] Add way to query input registration
apache · Oct 5, 2018 · e4965ad · e4965ad
2 parents d0810ef + 7fbf886
commit e4965ad
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