Add flat aliases

Author: pitr-ch

lib-edge/concurrent/edge/promises.rb

@@ -16,6 +16,32 @@ def then_ask(actor)
       end
 
       include ActorIntegration
+
+      # @!macro warn.edge
+      module FlatShortcuts
+
+        def then_flat_future(*args, &block)
+          self.then(*args, &block).flat_future
+        end
+
+        alias_method :then_flat, :then_flat_future
+
+        def then_flat_future_on(executor, *args, &block)
+          self.then_on(executor, *args, &block).flat_future
+        end
+
+        alias_method :then_flat_on, :then_flat_future_on
+
+        def then_flat_event(*args, &block)
+          self.then(*args, &block).flat_event
+        end
+
+        def then_flat_event_on(executor, *args, &block)
+          self.then_on(executor, *args, &block).flat_event
+        end
+      end
+
+      include FlatShortcuts
     end
 
     class Channel < Concurrent::Synchronization::Object

spec/concurrent/edge/promises_spec.rb

@@ -445,6 +445,12 @@ def behaves_as_delay(delay, value)
         expect(::Array.new(3) { |i| Concurrent::Promises.delay { i } }.
             inject { |a, b| a.then { b }.flat }.value!(0.2)).to eq 2
       end
+
+      it 'has shortcuts' do
+        expect(fulfilled_future(1).then_flat { |v| future(v) { v + 1 } }.value!).to eq 2
+        expect(fulfilled_future(1).then_flat_event { |v| resolved_event }.wait.resolved?).to eq true
+        expect(fulfilled_future(1).then_flat_on(:fast) { |v| future(v) { v + 1 } }.value!).to eq 2
+      end
     end
 
     it 'resolves future when Exception raised' do