Reorder callbacks

Tinder · Jul 11, 2018 · 563de6e · 563de6e
1 parent 83a8028
commit 563de6e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/com/tinder/StateMachine.kt b/src/main/kotlin/com/tinder/StateMachine.kt
@@ -12,14 +12,15 @@ class StateMachine<STATE : Any, EVENT : Any, SIDE_EFFECT : Any> private construc
         get() = stateRef.get()
 
     fun transition(event: EVENT): Transition<STATE, EVENT, SIDE_EFFECT> {
-        val transition = synchronized(stateRef) {
+        val transition = synchronized(this) {
             val fromState = stateRef.get()
             val transition = fromState.getTransition(event)
             if (transition is Transition.Valid) {
                 stateRef.set(transition.toState)
             }
             transition
         }
+        transition.notifyOnTransition()
         if (transition is Transition.Valid) {
             with(transition) {
                 with(fromState) {
@@ -30,7 +31,6 @@ class StateMachine<STATE : Any, EVENT : Any, SIDE_EFFECT : Any> private construc
                 }
             }
         }
-        transition.notifyOnTransition()
         return transition
     }