consume the YAML interchange more cleanly

lib/rude_q.rb

@@ -11,8 +11,18 @@ module RudeQ
 
   def self.included(mod) # :nodoc:
     mod.extend(ClassMethods)
+    mod.send(:include, InstanceMethods)
   end
-
+
+  module InstanceMethods
+    def data # :nodoc:
+      YAML.load(self[:data])
+    end
+    def data=(value) # :nodoc:
+      self[:data] = YAML.dump(value)
+    end
+  end
+
   module ClassMethods
     # Cleanup old processed items
     #
@@ -36,8 +46,7 @@ def cleanup!(expiry=1.hour)
     def set(queue_name, data)
       queue_name = sanitize_queue_name(queue_name)
 
-      yaml_data = YAML.dump(data)
-      self.create!(:queue_name => queue_name, :data => yaml_data)
+      self.create!(:queue_name => queue_name, :data => data)
       return nil # in line with Starling
     end
 
@@ -58,7 +67,7 @@ def get(queue_name)
       fetch_with_lock(qname) do |record|
         if record
           processed!(record)
-          return YAML.load(record.data)
+          return record.data
         else
           return nil # Starling waits indefinitely for a corresponding queue item
         end
@@ -137,6 +146,12 @@ def queue_options
       @queue_options ||= {:processed => :set_flag, :lock => :pessimistic}
     end
 
+    def data # :nodoc:
+      YAML.load(self[:data])
+    end
+    def data=(value) # :nodoc:
+      self[:data] = YAML.dump(value)
+    end
     private
 
     def sanitize_queue_name(queue_name) # :nodoc:

spec/rude_q_spec.rb

@@ -91,7 +91,7 @@ def create_some_noise
 
   describe ".set" do
     it "should delegate to :create!" do
-      ProcessQueue.should_receive(:create!).with(:queue_name => 'abcde', :data => YAML.dump(:magical_planet))
+      ProcessQueue.should_receive(:create!).with(:queue_name => 'abcde', :data => :magical_planet)
       ProcessQueue.set('abcde', :magical_planet)
     end
     it "should return nil" do
@@ -107,15 +107,15 @@ def create_some_noise
       # confirm the object is in the db
       record = ProcessQueue.find(:first, :order => "id DESC")
       record.queue_name.should == 'abcde'
-      record.data.should == YAML.dump(:this_will_remain_unprocessed)
+      record.data.should == :this_will_remain_unprocessed
       record.processed?.should == false
       record.token.should == nil
 
       lambda {ProcessQueue.get('abcde')}.should raise_error(RuntimeError)
 
       record.reload
       record.queue_name.should == 'abcde'
-      record.data.should == YAML.dump(:this_will_remain_unprocessed)
+      record.data.should == :this_will_remain_unprocessed
       record.processed?.should == false
       record.token.should == nil
     end