mirror of
https://github.com/aasm/aasm
synced 2023-03-27 23:22:41 -04:00
add redis persistence
This commit is contained in:
parent
95b61a3e63
commit
49fc631405
6 changed files with 192 additions and 0 deletions
|
@ -9,5 +9,6 @@ gem "activerecord-jdbcsqlite3-adapter", :platforms => :jruby
|
|||
gem "rails", "3.2.21"
|
||||
gem 'mongoid' if Gem::Version.create(RUBY_VERSION.dup) >= Gem::Version.create('1.9.3')
|
||||
gem 'sequel'
|
||||
gem "redis-objects"
|
||||
|
||||
gemspec :path => "../"
|
||||
|
|
|
@ -12,5 +12,6 @@ gem "rails", "4.0.12"
|
|||
# gem 'mongoid' if Gem::Version.create(RUBY_VERSION.dup) >= Gem::Version.create('1.9.3')
|
||||
|
||||
gem 'sequel'
|
||||
gem "redis-objects"
|
||||
|
||||
gemspec :path => "../"
|
||||
|
|
|
@ -12,5 +12,6 @@ gem "rails", "4.1.8"
|
|||
# gem 'mongoid' if Gem::Version.create(RUBY_VERSION.dup) >= Gem::Version.create('1.9.3')
|
||||
|
||||
gem 'sequel'
|
||||
gem "redis-objects"
|
||||
|
||||
gemspec :path => "../"
|
||||
|
|
|
@ -12,6 +12,8 @@ module AASM
|
|||
include_persistence base, :mongoid
|
||||
elsif hierarchy.include?("Sequel::Model")
|
||||
include_persistence base, :sequel
|
||||
elsif hierarchy.include?("Redis::Objects")
|
||||
include_persistence base, :redis
|
||||
else
|
||||
include_persistence base, :plain
|
||||
end
|
||||
|
|
110
lib/aasm/persistence/redis_persistence.rb
Normal file
110
lib/aasm/persistence/redis_persistence.rb
Normal file
|
@ -0,0 +1,110 @@
|
|||
require_relative "base"
|
||||
|
||||
module AASM
|
||||
module Persistence
|
||||
module RedisPersistence
|
||||
|
||||
def self.included(base)
|
||||
base.send(:include, AASM::Persistence::Base)
|
||||
base.send(:include, AASM::Persistence::RedisPersistence::InstanceMethods)
|
||||
end
|
||||
|
||||
module InstanceMethods
|
||||
# Add the inital value to intiializer
|
||||
#
|
||||
# redis-objects removed the key from redis when set to nil
|
||||
def initialize(*args)
|
||||
super
|
||||
#require "pry"; binding.pry
|
||||
state = send(self.class.aasm.attribute_name)
|
||||
state.value = aasm.determine_state_name(self.class.aasm.initial_state)
|
||||
end
|
||||
# Returns the value of the aasm.attribute_name - called from <tt>aasm.current_state</tt>
|
||||
#
|
||||
# If it's a new record, and the aasm state column is blank it returns the initial state
|
||||
#
|
||||
# class Foo
|
||||
# include Redis::Objects
|
||||
# include AASM
|
||||
# aasm :column => :status do
|
||||
# state :opened
|
||||
# state :closed
|
||||
# end
|
||||
# end
|
||||
#
|
||||
# foo = Foo.new
|
||||
# foo.current_state # => :opened
|
||||
# foo.close
|
||||
# foo.current_state # => :closed
|
||||
#
|
||||
# foo = Foo[1]
|
||||
# foo.current_state # => :opened
|
||||
# foo.aasm_state = nil
|
||||
# foo.current_state # => nil
|
||||
#
|
||||
# NOTE: intended to be called from an event
|
||||
#
|
||||
# This allows for nil aasm states - be sure to add validation to your model
|
||||
def aasm_read_state
|
||||
state = send(self.class.aasm.attribute_name)
|
||||
|
||||
if state.value.nil?
|
||||
nil
|
||||
else
|
||||
state.value.to_sym
|
||||
end
|
||||
end
|
||||
|
||||
# Ensures that if the aasm_state column is nil and the record is new
|
||||
# that the initial state gets populated before validation on create
|
||||
#
|
||||
# foo = Foo.new
|
||||
# foo.aasm_state # => nil
|
||||
# foo.valid?
|
||||
# foo.aasm_state # => "open" (where :open is the initial state)
|
||||
#
|
||||
#
|
||||
# foo = Foo.find(:first)
|
||||
# foo.aasm_state # => 1
|
||||
# foo.aasm_state = nil
|
||||
# foo.valid?
|
||||
# foo.aasm_state # => nil
|
||||
#
|
||||
def aasm_ensure_initial_state
|
||||
aasm.enter_initial_state if
|
||||
send(self.class.aasm.attribute_name).to_s.strip.empty?
|
||||
end
|
||||
|
||||
# Writes <tt>state</tt> to the state column and persists it to the database
|
||||
#
|
||||
# foo = Foo[1]
|
||||
# foo.aasm.current_state # => :opened
|
||||
# foo.close!
|
||||
# foo.aasm.current_state # => :closed
|
||||
# Foo[1].aasm.current_state # => :closed
|
||||
#
|
||||
# NOTE: intended to be called from an event
|
||||
def aasm_write_state(state)
|
||||
aasm_column = self.class.aasm.attribute_name
|
||||
self.send("#{aasm_column}=", state)
|
||||
end
|
||||
|
||||
# Writes <tt>state</tt> to the state column, but does not persist it to the database
|
||||
#
|
||||
# foo = Foo[1]
|
||||
# foo.aasm.current_state # => :opened
|
||||
# foo.close
|
||||
# foo.aasm.current_state # => :closed
|
||||
# Foo[1].aasm.current_state # => :opened
|
||||
# foo.save
|
||||
# foo.aasm.current_state # => :closed
|
||||
# Foo[1].aasm.current_state # => :closed
|
||||
#
|
||||
# NOTE: intended to be called from an event
|
||||
def aasm_write_state_without_persistence(state)
|
||||
send("#{self.class.aasm.attribute_name}=", state)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
77
spec/unit/persistence/redis_persistence_spec.rb
Normal file
77
spec/unit/persistence/redis_persistence_spec.rb
Normal file
|
@ -0,0 +1,77 @@
|
|||
|
||||
describe 'redis' do
|
||||
begin
|
||||
require 'redis-objects'
|
||||
require 'logger'
|
||||
require 'spec_helper'
|
||||
|
||||
before(:all) do
|
||||
Redis.current = Redis.new(host: '127.0.0.1', port: 6379)
|
||||
|
||||
@model = Class.new do
|
||||
attr_accessor :default
|
||||
|
||||
include Redis::Objects
|
||||
include AASM
|
||||
|
||||
value :status
|
||||
|
||||
def id
|
||||
1
|
||||
end
|
||||
|
||||
aasm column: :status
|
||||
aasm do
|
||||
state :alpha, initial: true
|
||||
state :beta
|
||||
state :gamma
|
||||
event :release do
|
||||
transitions from: [:alpha, :beta, :gamma], to: :beta
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
describe "instance methods" do
|
||||
let(:model) {@model.new}
|
||||
|
||||
it "should respond to aasm persistence methods" do
|
||||
expect(model).to respond_to(:aasm_read_state)
|
||||
expect(model).to respond_to(:aasm_write_state)
|
||||
expect(model).to respond_to(:aasm_write_state_without_persistence)
|
||||
end
|
||||
|
||||
it "should return the initial state when new and the aasm field is nil" do
|
||||
expect(model.aasm.current_state).to eq(:alpha)
|
||||
end
|
||||
|
||||
it "should return the aasm column when new and the aasm field is not nil" do
|
||||
model.status = "beta"
|
||||
expect(model.aasm.current_state).to eq(:beta)
|
||||
end
|
||||
|
||||
it "should allow a nil state" do
|
||||
model.status = nil
|
||||
expect(model.aasm.current_state).to be_nil
|
||||
end
|
||||
end
|
||||
|
||||
describe 'subclasses' do
|
||||
it "should have the same states as its parent class" do
|
||||
expect(Class.new(@model).aasm.states).to eq(@model.aasm.states)
|
||||
end
|
||||
|
||||
it "should have the same events as its parent class" do
|
||||
expect(Class.new(@model).aasm.events).to eq(@model.aasm.events)
|
||||
end
|
||||
|
||||
it "should have the same column as its parent even for the new dsl" do
|
||||
expect(@model.aasm.attribute_name).to eq(:status)
|
||||
expect(Class.new(@model).aasm.attribute_name).to eq(:status)
|
||||
end
|
||||
end
|
||||
|
||||
rescue LoadError
|
||||
puts "Not running Sequel specs because sequel gem is not installed!!!"
|
||||
end
|
||||
end
|
Loading…
Reference in a new issue