Merge pull request #182 from malukenho/enhancement/transaction

Introduce EventStore#transactional(callable $callable) API

src/EventStore.php

@@ -295,6 +295,29 @@ public function replay(array $streamNames, DateTimeInterface $since = null, arra
         });
     }
 
+    /**
+     * @param callable $callable
+     *
+     * @throws \Exception
+     *
+     * @return mixed
+     */
+    public function transactional(callable $callable)
+    {
+        $this->beginTransaction();
+
+        try {
+            $result = $callable($this);
+            $this->commit();
+        } catch (\Exception $e) {
+            $this->rollback();
+
+            throw $e;
+        }
+
+        return $result ?: true;
+    }
+
     /**
      * Begin transaction
      *

tests/EventStoreTest.php

@@ -18,6 +18,7 @@
 use Prooph\EventStore\EventStore;
 use Prooph\EventStore\Stream\Stream;
 use Prooph\EventStore\Stream\StreamName;
+use ProophTest\EventStore\Mock\TransactionalInMemoryAdapterMock;
 use ProophTest\EventStore\Mock\PostCreated;
 use ProophTest\EventStore\Mock\TestDomainEvent;
 use ProophTest\EventStore\Mock\UserCreated;
@@ -586,6 +587,76 @@ public function it_makes_rollback_when_event_is_stopped_during_commit()
         $this->eventStore->load($stream->streamName());
     }
 
+    /**
+     * @test
+     */
+    public function it_should_rollback_and_throw_exception_in_case_of_transaction_fail()
+    {
+        $eventStore = new EventStore(new TransactionalInMemoryAdapterMock(), new ProophActionEventEmitter());
+
+        $this->setExpectedException(\Exception::class, 'Transaction failed');
+
+        $eventStore->transactional(function (EventStore $es) use ($eventStore) {
+            self::assertSame($es, $eventStore);
+
+            throw new \Exception('Transaction failed');
+        });
+    }
+
+    /**
+     * @test
+     */
+    public function it_should_return_true_by_default_if_transaction_is_used()
+    {
+        $transactionResult = $this->eventStore->transactional(function (EventStore $eventStore) {
+            $this->eventStore->create($this->getTestStream());
+
+            self::assertSame($this->eventStore, $eventStore);
+        });
+
+        self::assertTrue($transactionResult);
+    }
+
+    /**
+     * @test
+     */
+    public function it_wraps_up_code_in_transaction_properly()
+    {
+        $recordedEvents = [];
+
+        $this->eventStore->getActionEventEmitter()->attachListener('commit.post', function (ActionEvent $event) use (&$recordedEvents) {
+            foreach ($event->getParam('recordedEvents', new \ArrayIterator()) as $recordedEvent) {
+                $recordedEvents[] = $recordedEvent;
+            }
+        });
+
+        $transactionResult = $this->eventStore->transactional(function (EventStore $eventStore) {
+            $this->eventStore->create($this->getTestStream());
+
+            self::assertSame($this->eventStore, $eventStore);
+
+            return 'Result';
+        });
+
+        self::assertSame('Result', $transactionResult);
+
+        $secondStreamEvent = UsernameChanged::with(
+            ['new_name' => 'John Doe'],
+            2
+        );
+
+        $transactionResult = $this->eventStore->transactional(function (EventStore $eventStore) use ($secondStreamEvent) {
+            $this->eventStore->appendTo(new StreamName('user'), new ArrayIterator([$secondStreamEvent]));
+
+            self::assertSame($this->eventStore, $eventStore);
+
+            return 'Second Result';
+        });
+
+        self::assertSame('Second Result', $transactionResult);
+        self::assertCount(2, $recordedEvents);
+    }
+
     /**
      * @test
      */

tests/Mock/TransactionalInMemoryAdapterMock.php

@@ -0,0 +1,44 @@
+<?php
+/**
+ * This file is part of the prooph/service-bus.
+ * (c) 2014-2016 prooph software GmbH <[email protected]>
+ * (c) 2015-2016 Sascha-Oliver Prolic <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace ProophTest\EventStore\Mock;
+
+use Prooph\EventStore\Adapter\Feature\CanHandleTransaction;
+use Prooph\EventStore\Adapter\InMemoryAdapter;
+
+/**
+ * Class TransactionalInMemoryAdapterMock
+ *
+ * @package ProophTest\EventStore
+ * @author Jefersson Nathan <[email protected]>
+ */
+final class TransactionalInMemoryAdapterMock extends InMemoryAdapter implements CanHandleTransaction
+{
+    /**
+     * {@inheritDoc}
+     */
+    public function beginTransaction()
+    {
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function commit()
+    {
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function rollback()
+    {
+    }
+}