diff options
author | Joshua Peek <josh@joshpeek.com> | 2009-08-04 11:03:57 -0500 |
---|---|---|
committer | Joshua Peek <josh@joshpeek.com> | 2009-08-04 11:03:57 -0500 |
commit | aad5a30bf25d8a3167afd685fc91c99f4f09cc57 (patch) | |
tree | f700750ee088e483537fa7ee992a109b834b1ddc /activerecord/lib/active_record | |
parent | 55d1d12c32a1b99f3f07d2346b49a63650ba2e9d (diff) | |
download | rails-aad5a30bf25d8a3167afd685fc91c99f4f09cc57.tar.gz rails-aad5a30bf25d8a3167afd685fc91c99f4f09cc57.tar.bz2 rails-aad5a30bf25d8a3167afd685fc91c99f4f09cc57.zip |
Add simple support for ActiveModel's StateMachine for ActiveRecord
Diffstat (limited to 'activerecord/lib/active_record')
-rw-r--r-- | activerecord/lib/active_record/state_machine.rb | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/activerecord/lib/active_record/state_machine.rb b/activerecord/lib/active_record/state_machine.rb new file mode 100644 index 0000000000..aebd03344a --- /dev/null +++ b/activerecord/lib/active_record/state_machine.rb @@ -0,0 +1,24 @@ +module ActiveRecord + module StateMachine #:nodoc: + extend ActiveSupport::Concern + include ActiveModel::StateMachine + + included do + before_validation :set_initial_state + validates_presence_of :state + end + + protected + def write_state(state_machine, state) + update_attributes! :state => state.to_s + end + + def read_state(state_machine) + self.state.to_sym + end + + def set_initial_state + self.state ||= self.class.state_machine.initial_state.to_s + end + end +end |