Merge 2dccc5d into 75aebb6

lib/sandthorn.rb

@@ -2,6 +2,7 @@
 require "sandthorn/errors"
 require "sandthorn/aggregate_root"
 require "sandthorn/event_stores"
+require "sandthorn/snapshot_store"
 require 'yaml'
 require 'securerandom'
 
@@ -10,6 +11,7 @@ class << self
     extend Forwardable
 
     def_delegators :configuration, :event_stores
+    def_delegators :configuration, :snapshot_store
 
     def default_event_store
       event_stores.default_store
@@ -39,12 +41,20 @@ def all aggregate_type
       event_store_for(aggregate_type).all(aggregate_type)
     end
 
-    def find aggregate_id, aggregate_type
-      event_store_for(aggregate_type).find(aggregate_id, aggregate_type)
+    def find aggregate_id, aggregate_type, after_aggregate_version = 0
+      event_store_for(aggregate_type).find(aggregate_id, aggregate_type, after_aggregate_version)
     end
 
-    def save_snapshot(aggregate)
-      raise "Not Implemented"
+    def save_snapshot aggregate
+      if aggregate.aggregate_events.count > 0
+        raise Errors::SnapshotError,
+          "Can't take snapshot on object with unsaved events"
+      end
+      snapshot_store.save aggregate.aggregate_id, aggregate
+    end
+
+    def find_snapshot aggregate_id
+      return snapshot_store.find aggregate_id
     end
 
     def find_event_store(name)
@@ -84,6 +94,16 @@ def map_types= data
         @event_stores.map_types data
       end
 
+      def snapshot_store
+        @snapshot_store ||= SnapshotStore.new
+      end
+
+      def snapshot_types= aggregate_types
+        aggregate_types.each do |aggregate_type|
+          aggregate_type.snapshot(true)
+        end
+      end
+
       alias_method :event_stores=, :event_store=
     end
   end

lib/sandthorn/aggregate_root_base.rb

@@ -29,6 +29,8 @@ def save
           @aggregate_originating_version = @aggregate_current_event_version
         end
 
+        Sandthorn.save_snapshot self if self.class.snapshot
+
         self
       end
 
@@ -70,9 +72,17 @@ def event_store(event_store = nil)
           end
         end
 
+        def snapshot(value = nil)
+          if value
+            @snapshot = value
+          else
+            @snapshot
+          end
+        end
+
         def all
           Sandthorn.all(self).map { |events|
-            aggregate_build events
+            aggregate_build events, nil
           }
         end
 
@@ -83,13 +93,15 @@ def find id
 
         def aggregate_find aggregate_id
           begin
-            events = Sandthorn.find(aggregate_id, self)
-            unless events && !events.empty?
+            aggregate_from_snapshot = Sandthorn.find_snapshot(aggregate_id) if self.snapshot
+            current_aggregate_version = aggregate_from_snapshot.nil? ? 0 : aggregate_from_snapshot.aggregate_current_event_version
+            events = Sandthorn.find(aggregate_id, self, current_aggregate_version)
+            if aggregate_from_snapshot.nil? && events.empty?
               raise Errors::AggregateNotFound
             end
 
-            return aggregate_build events
-          rescue Exception
+            return aggregate_build events, aggregate_from_snapshot
+          rescue Exception => error
             raise Errors::AggregateNotFound
           end
 
@@ -111,8 +123,8 @@ def new *args, &block
 
         end
 
-        def aggregate_build events
-          aggregate = create_new_empty_aggregate
+        def aggregate_build events, aggregate_from_snapshot = nil
+          aggregate = aggregate_from_snapshot || create_new_empty_aggregate
 
           if events.any?
             current_aggregate_version = events.last[:aggregate_version]

lib/sandthorn/snapshot_store.rb

@@ -0,0 +1,17 @@
+module Sandthorn
+  class SnapshotStore
+  	def initialize
+      @store = Hash.new
+    end
+
+    attr_reader :store
+
+    def save key, value
+      @store[key] = value
+    end
+
+    def find key
+      @store[key]
+    end
+  end
+end

spec/aggregate_root_spec.rb

@@ -51,6 +51,17 @@ def no_state_change_only_empty_event
       end
     end
 
+    describe "::snapshot" do
+      let(:klass) { Class.new { include Sandthorn::AggregateRoot } }
+      it "is available as a class method" do
+        expect(klass).to respond_to(:snapshot)
+      end
+      it "sets the snapshot to true and returns it" do
+        klass.snapshot(true)
+        expect(klass.snapshot).to eq(true)
+      end
+    end
+
     describe "when get all aggregates from DirtyClass" do
 
       before(:each) do

spec/snapshot_spec.rb

@@ -0,0 +1,68 @@
+require 'spec_helper'
+
+module Sandthorn
+  module Snapshot
+    class KlassOne
+      include Sandthorn::AggregateRoot
+      snapshot true
+    end
+
+    class KlassTwo
+      include Sandthorn::AggregateRoot
+    end
+
+    class KlassThree
+      include Sandthorn::AggregateRoot
+    end
+
+    describe "::snapshot" do
+      before do
+        Sandthorn.configure do |c|
+          c.snapshot_types = [KlassTwo]
+        end
+      end
+      it "snapshot should be enabled on KlassOne and KlassTwo but not KlassThree" do
+        expect(KlassOne.snapshot).to be_truthy
+        expect(KlassTwo.snapshot).to be_truthy
+        expect(KlassThree.snapshot).not_to be_truthy
+      end
+    end
+
+    describe "find snapshot on snapshot enabled aggregate" do
+      let(:klass) { KlassOne.new.save }
+
+      it "should find on snapshot enabled Class" do
+        copy = KlassOne.find klass.aggregate_id
+        expect(copy.aggregate_version).to eql(klass.aggregate_version)
+      end
+
+      it "should get saved snapshot" do
+        copy = Sandthorn.find_snapshot klass.aggregate_id
+        expect(copy.aggregate_version).to eql(klass.aggregate_version)
+      end
+
+    end
+
+    describe "save and find snapshot on snapshot disabled aggregate" do
+      let(:klass) { KlassThree.new.save }
+
+      it "should not find snapshot" do
+        snapshot = Sandthorn.find_snapshot klass.aggregate_id
+        expect(snapshot).to be_nil
+      end
+
+      it "should save and get saved snapshot" do
+        Sandthorn.save_snapshot klass
+        snapshot = Sandthorn.find_snapshot klass.aggregate_id
+        expect(snapshot).not_to be_nil
+
+        #Check by key on the snapshot_store hash
+        expect(Sandthorn.snapshot_store.store.has_key?(klass.aggregate_id)).to be_truthy
+
+      end
+
+    end
+
+  end
+end
+

spec/spec_helper.rb

@@ -45,17 +45,13 @@ def class_including(mod)
 def url
   "sqlite://spec/db/sequel_driver.sqlite3"
 end
+
 def sqlite_store_setup
 
   SandthornDriverSequel.migrate_db url: url
-
-  driver = SandthornDriverSequel.driver_from_url(url: url) do |conf|
-    conf.event_serializer       = Proc.new { |data| YAML::dump(data) }
-    conf.event_deserializer     = Proc.new { |data| YAML::load(data) }
-  end
 
   Sandthorn.configure do |c|
-    c.event_store = driver
+    c.event_store = SandthornDriverSequel.driver_from_url(url: url)
   end
 
 end