1
0
Fork 0
mirror of https://github.com/aasm/aasm synced 2023-03-27 23:22:41 -04:00

add redis persistence

This commit is contained in:
Jack Thorne 2014-12-02 16:33:39 -08:00
parent 95b61a3e63
commit 49fc631405
6 changed files with 192 additions and 0 deletions

View file

@ -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 => "../"

View file

@ -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 => "../"

View file

@ -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 => "../"

View file

@ -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

View 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

View 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