Expose count_workflow_executions on the temporal client

lib/temporal.rb

@@ -30,6 +30,7 @@ module Temporal
                  :list_open_workflow_executions,
                  :list_closed_workflow_executions,
                  :query_workflow_executions,
+                 :count_workflow_executions,
                  :add_custom_search_attributes,
                  :list_custom_search_attributes,
                  :remove_custom_search_attributes,

lib/temporal/client.rb

@@ -425,6 +425,17 @@ def query_workflow_executions(namespace, query, filter: {}, next_page_token: nil
       Temporal::Workflow::Executions.new(connection: connection, status: :all, request_options: { namespace: namespace, query: query, next_page_token: next_page_token, max_page_size: max_page_size }.merge(filter))
     end
 
+    # Count the number of workflows matching the provided query
+    # 
+    # @param namespace [String]
+    # @param query [String]
+    #
+    # @return [Integer] an integer count of workflows matching the query
+    def count_workflow_executions(namespace, query: nil)
+      response = connection.count_workflow_executions(namespace: namespace, query: query)
+      response.count
+    end
+
     # @param attributes [Hash[String, Symbol]] name to symbol for type, see INDEXED_VALUE_TYPE above
     # @param namespace String, required for SQL enhanced visibility, ignored for elastic search
     def add_custom_search_attributes(attributes, namespace: nil)

spec/unit/lib/temporal/client_spec.rb

@@ -1102,4 +1102,28 @@ class NamespacedWorkflow < Temporal::Workflow
       end
     end
   end
+
+  describe '#count_workflow_executions' do
+    let(:response) do
+      Temporalio::Api::WorkflowService::V1::CountWorkflowExecutionsResponse.new(
+        count: 5
+      )
+    end
+
+    before do
+      allow(connection)
+        .to receive(:count_workflow_executions)
+        .and_return(response)
+    end
+
+    it 'returns the count' do
+      resp = subject.count_workflow_executions(namespace, query: 'ExecutionStatus="Running"')
+
+      expect(connection)
+        .to have_received(:count_workflow_executions)
+        .with(namespace: namespace, query: 'ExecutionStatus="Running"')
+
+      expect(resp).to eq(5)
+    end
+  end
 end