lib/graphql_grpc/schema.rb in graphql_grpc-0.1.5 vs lib/graphql_grpc/schema.rb in graphql_grpc-0.1.6

@@ -30,15 +30,23 @@
     def query?(name_sym, rpc_desc)
       name_sym.to_s.start_with?('get') ||
         rpc_desc.rpc_desc.input == Google::Protobuf::Empty
     end
 
+    def streaming_response?(rpc_desc)
+      rpc_desc&.rpc_desc&.output.class == GRPC::RpcDesc::Stream
+    end
+
     def gql_mutations
-      @function_map.reject { |name_sym, rpc_desc| query?(name_sym, rpc_desc) }
+      @function_map.reject do |name_sym, rpc_desc|
+        query?(name_sym, rpc_desc) || streaming_response?(rpc_desc)
+      end
     end
 
     def gql_queries
-      @function_map.select { |name_sym, rpc_desc| query?(name_sym, rpc_desc) }
+      @function_map.select do |name_sym, rpc_desc|
+        query?(name_sym, rpc_desc) && !streaming_response?(rpc_desc)
+      end
     end
 
     def to_schema_types
       function_output_types = @function_map.values.map do |function|
         function.rpc_desc.output