Merge pull request #282 from Johnabell/support-created-at-override

Add support for overriding created_at timestamps for copy transform workflows

lib/event_store.ex

@@ -294,13 +294,17 @@ defmodule EventStore do
         Supervisor.stop(supervisor, :normal, timeout)
       end
 
+      @accepted_overrides_append_to_stream [:created_at_override]
+
       def append_to_stream(stream_uuid, expected_version, events, opts \\ [])
 
       def append_to_stream(@all_stream, _expected_version, _events, _opts),
         do: {:error, :cannot_append_to_all_stream}
 
       def append_to_stream(stream_uuid, expected_version, events, opts) do
+        overrides = Keyword.take(opts, @accepted_overrides_append_to_stream)
         {conn, opts} = parse_opts(opts)
+        opts = Keyword.merge(opts, overrides)
 
         Stream.append_to_stream(conn, stream_uuid, expected_version, events, opts)
       end

lib/event_store/sql/statements.ex

@@ -11,8 +11,8 @@ defmodule EventStore.Sql.Statements do
   for {fun, args} <- [
         {:count_streams, [:schema]},
         {:create_stream, [:schema]},
-        {:insert_events, [:schema, :stream_id, :number_of_events]},
-        {:insert_events_any_version, [:schema, :stream_id, :number_of_events]},
+        {:insert_events, [:schema, :stream_id, :number_of_events, :created_at]},
+        {:insert_events_any_version, [:schema, :stream_id, :number_of_events, :created_at]},
         {:insert_link_events, [:schema, :number_of_events]},
         {:soft_delete_stream, [:schema]},
         {:hard_delete_stream, [:schema]},

lib/event_store/sql/statements/insert_events.sql.eex

@@ -24,12 +24,17 @@ WITH
     <% end %>
   ),
   stream AS (
-    <%= if stream_id do %>
+    <%= cond do %>
+    <% stream_id -> %>
       UPDATE "<%= schema %>".streams
       SET stream_version = stream_version + $2::bigint
       WHERE stream_id = $1::bigint
       returning stream_id
-    <% else %>
+    <% created_at -> %>
+      INSERT INTO "<%= schema %>".streams (stream_uuid, stream_version, created_at)
+      VALUES ($1, $2::bigint, $<%= number_of_events*9 + 3 %>)
+      returning stream_id
+    <% true -> %>
       INSERT INTO "<%= schema %>".streams (stream_uuid, stream_version)
       VALUES ($1, $2::bigint)
       returning stream_id

lib/event_store/sql/statements/insert_events_any_version.sql.eex

@@ -1,11 +1,16 @@
 WITH
   stream AS (
-    <%= if stream_id do %>
+    <%= cond do %>
+    <% stream_id -> %>
       UPDATE "<%= schema %>".streams
       SET stream_version = stream_version + $2::bigint
       WHERE stream_id = $1::bigint
       returning stream_id, stream_version - $2::bigint as initial_stream_version
-    <% else %>
+    <% created_at -> %>
+      INSERT INTO "<%= schema %>".streams (stream_uuid, stream_version, created_at)
+      VALUES ($1, $2::bigint, $<%= number_of_events*9 + 3 %>)
+      returning stream_id, stream_version - $2::bigint as initial_stream_version
+    <% true -> %>
       INSERT INTO "<%= schema %>".streams (stream_uuid, stream_version)
       VALUES ($1, $2::bigint)
       returning stream_id, stream_version - $2::bigint as initial_stream_version

lib/event_store/storage/appender.ex

@@ -101,15 +101,23 @@ defmodule EventStore.Storage.Appender do
   defp insert_event_batch(conn, stream_id, stream_uuid, events, event_count, opts) do
     {schema, opts} = Keyword.pop(opts, :schema)
     {expected_version, opts} = Keyword.pop(opts, :expected_version)
+    {created_at, opts} = Keyword.pop(opts, :created_at_override)
 
     statement =
       case expected_version do
-        :any_version -> Statements.insert_events_any_version(schema, stream_id, event_count)
-        _expected_version -> Statements.insert_events(schema, stream_id, event_count)
+        :any_version ->
+          Statements.insert_events_any_version(schema, stream_id, event_count, created_at)
+
+        _expected_version ->
+          Statements.insert_events(schema, stream_id, event_count, created_at)
       end
 
     stream_id_or_uuid = stream_id || stream_uuid
-    params = [stream_id_or_uuid, event_count] ++ build_insert_parameters(events)
+
+    params =
+      [stream_id_or_uuid, event_count]
+      |> Enum.concat(build_insert_parameters(events))
+      |> append_if(!stream_id, created_at)
 
     case Postgrex.query(conn, statement, params, opts) do
       {:ok, %Postgrex.Result{num_rows: 0}} ->
@@ -123,6 +131,9 @@ defmodule EventStore.Storage.Appender do
     end
   end
 
+  defp append_if(params, true, value) when not is_nil(value), do: params ++ [value]
+  defp append_if(params, _, _), do: params
+
   defp build_insert_parameters(events) do
     events
     |> Enum.with_index(1)

lib/event_store/streams/stream.ex

@@ -145,16 +145,16 @@ defmodule EventStore.Streams.Stream do
          serializer,
          opts
        ) do
-    prepared_events = prepare_events(events, stream, serializer)
+    prepared_events = prepare_events(events, stream, serializer, opts)
 
     write_to_stream(conn, prepared_events, stream, expected_version, opts)
   end
 
-  defp prepare_events(events, %StreamInfo{} = stream, serializer) do
+  defp prepare_events(events, %StreamInfo{} = stream, serializer, opts) do
     %StreamInfo{stream_uuid: stream_uuid, stream_version: stream_version} = stream
 
     events
-    |> Enum.map(&map_to_recorded_event(&1, utc_now(), serializer))
+    |> Enum.map(&map_to_recorded_event(&1, opts[:created_at_override] || utc_now(), serializer))
     |> Enum.with_index(1)
     |> Enum.map(fn {recorded_event, index} ->
       %RecordedEvent{

test/event_store_test.exs

@@ -203,6 +203,84 @@ defmodule EventStore.EventStoreTest do
     assert recorded_event.data.event == unicode_text
   end
 
+  test "override created_at" do
+    created_at = DateTime.utc_now() |> DateTime.add(-1, :day)
+    stream_uuid = UUID.uuid4()
+    events = EventFactory.create_events(1)
+
+    :ok = EventStore.append_to_stream(stream_uuid, 0, events, created_at_override: created_at)
+
+    [recorded_event] = EventStore.stream_all_forward() |> Enum.to_list()
+    {:ok, stream_info} = EventStore.stream_info(stream_uuid)
+
+    assert recorded_event.created_at == created_at
+    assert stream_info.created_at == created_at
+  end
+
+  test "override created_at existing stream" do
+    created_at = DateTime.utc_now() |> DateTime.add(-1, :day)
+    created_at2 = DateTime.utc_now() |> DateTime.add(-1, :hour)
+    stream_uuid = UUID.uuid4()
+    events = EventFactory.create_events(1)
+    events2 = EventFactory.create_events(1)
+
+    :ok = EventStore.append_to_stream(stream_uuid, 0, events, created_at_override: created_at)
+
+    :ok =
+      EventStore.append_to_stream(stream_uuid, :any_version, events2,
+        created_at_override: created_at2
+      )
+
+    [event1, event2] = EventStore.stream_all_forward() |> Enum.to_list()
+    {:ok, stream_info} = EventStore.stream_info(stream_uuid)
+
+    assert event1.created_at == created_at
+    assert stream_info.created_at == created_at
+    assert event2.created_at == created_at2
+  end
+
+  test "override created_at any_version" do
+    created_at = DateTime.utc_now() |> DateTime.add(-1, :day)
+    stream_uuid = UUID.uuid4()
+    events = EventFactory.create_events(1)
+
+    :ok =
+      EventStore.append_to_stream(stream_uuid, :any_version, events,
+        created_at_override: created_at
+      )
+
+    [recorded_event] = EventStore.stream_all_forward() |> Enum.to_list()
+    {:ok, stream_info} = EventStore.stream_info(stream_uuid)
+
+    assert recorded_event.created_at == created_at
+    assert stream_info.created_at == created_at
+  end
+
+  test "override created_at any_version existing stream" do
+    created_at = DateTime.utc_now() |> DateTime.add(-1, :day)
+    created_at2 = DateTime.utc_now() |> DateTime.add(-1, :hour)
+    stream_uuid = UUID.uuid4()
+    events = EventFactory.create_events(1)
+    events2 = EventFactory.create_events(1)
+
+    :ok =
+      EventStore.append_to_stream(stream_uuid, :any_version, events,
+        created_at_override: created_at
+      )
+
+    :ok =
+      EventStore.append_to_stream(stream_uuid, :any_version, events2,
+        created_at_override: created_at2
+      )
+
+    [event1, event2] = EventStore.stream_all_forward() |> Enum.to_list()
+    {:ok, stream_info} = EventStore.stream_info(stream_uuid)
+
+    assert event1.created_at == created_at
+    assert stream_info.created_at == created_at
+    assert event2.created_at == created_at2
+  end
+
   describe "transient subscription" do
     test "should notify subscribers after event persisted to stream" do
       stream_uuid = UUID.uuid4()