SAMZA-1361; OperatorImplGraph is using wrong keys to store/retrieve OperatorImpl...
[samza.git] / samza-core / src / main / java / org / apache / samza / operators / StreamGraphImpl.java
index c0da1b2..2c2eb56 100644 (file)
@@ -170,13 +170,13 @@ public class StreamGraphImpl implements StreamGraph {
   /**
    * Get all {@link OperatorSpec}s available in this {@link StreamGraphImpl}
    *
   /**
    * Get all {@link OperatorSpec}s available in this {@link StreamGraphImpl}
    *
-   * @return  a set of all available {@link OperatorSpec}s
+   * @return  all available {@link OperatorSpec}s
    */
   public Collection<OperatorSpec> getAllOperatorSpecs() {
     Collection<InputOperatorSpec> inputOperatorSpecs = inputOperators.values();
     Set<OperatorSpec> operatorSpecs = new HashSet<>();
    */
   public Collection<OperatorSpec> getAllOperatorSpecs() {
     Collection<InputOperatorSpec> inputOperatorSpecs = inputOperators.values();
     Set<OperatorSpec> operatorSpecs = new HashSet<>();
-
     for (InputOperatorSpec inputOperatorSpec: inputOperatorSpecs) {
     for (InputOperatorSpec inputOperatorSpec: inputOperatorSpecs) {
+      operatorSpecs.add(inputOperatorSpec);
       doGetOperatorSpecs(inputOperatorSpec, operatorSpecs);
     }
     return operatorSpecs;
       doGetOperatorSpecs(inputOperatorSpec, operatorSpecs);
     }
     return operatorSpecs;