AggregateRoot + Protobuf

Maybe fixes: #578 :)
RailsEventStore · May 1, 2019 · 83a9f76 · 83a9f76
1 parent 26976c7
commit 83a9f76
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/aggregate_root/lib/aggregate_root.rb b/aggregate_root/lib/aggregate_root.rb
@@ -12,7 +12,7 @@ def on(*event_klasses, &block)
         handler_name = "on_#{name}"
         define_method(handler_name, &block)
         @on_methods ||= {}
-        @on_methods[event_klass]=handler_name
+        @on_methods[name]=handler_name
         private(handler_name)
       end
     end

diff --git a/aggregate_root/lib/aggregate_root/default_apply_strategy.rb b/aggregate_root/lib/aggregate_root/default_apply_strategy.rb
@@ -19,15 +19,15 @@ def call(aggregate, event)
     private
 
     def handler_name(event)
-      on_methods.fetch(event.class) { handler_name_by_class(event.class) }
+      on_methods.fetch(event.type) { handler_name_by_type(event.type) }
     end
 
-    def handler_name_by_class(event_class)
-      "apply_#{Transform.to_snake_case(class_name(event_class))}"
+    def handler_name_by_type(event_type)
+      "apply_#{Transform.to_snake_case(event_type(event_type))}"
     end
 
-    def class_name(event_class)
-      event_class.name.split("::").last
+    def event_type(event_type)
+      event_type.split("::").last
     end
 
     private