message.rb in deepstream-0.2.1

- old
+ new

@@ -1,20 +1,21 @@
 require 'json'
 require 'deepstream/constants'
 
 module Deepstream
   class Message
-    attr_reader :topic, :action, :data
+    attr_reader :action, :data, :topic, :sending_deadline
 
     def self.parse(*args)
       args.first.is_a?(self) ? args.first : new(*args)
     end
 
-    def initialize(*args)
+    def initialize(*args, timeout: nil)
       if args.one?
         args = args.first.delete(MESSAGE_SEPARATOR).split(MESSAGE_PART_SEPARATOR)
       end
+      @sending_deadline = Time.now + timeout if timeout
       @topic, @action = args.take(2).map(&:to_sym)
       @data = args.drop(2)
     end
 
     def to_s
@@ -27,8 +28,12 @@
       "#{self.class.name}: #{@topic} #{@action} #{@data}"
     end
 
     def needs_authentication?
       ![TOPIC::CONNECTION, TOPIC::AUTH].include?(@topic)
+    end
+
+    def expired?
+      @sending_deadline && @sending_deadline < Time.now
     end
   end
 end