stompserver 0.9.7 → 0.9.8

data/lib/frame_journal.rb

@@ -1,135 +0,0 @@
-# Simple Journal File(s) Manager
-# You select the directory, and it will collect the messages
-# The journal file format is:
-#
-# Status Byte: 0 - pending, 1 - processed
-# Frame Size: 4 byte long (network endian - yes I limit my messages to 4G)
-# Message 
-#
-# Repeat
-#
-# When the size of a journal file exceeds its limit
-
-require 'rubygems'
-require 'madeleine'
-require 'madeleine/automatic'
-
-class MadFrameJournal
-  include Madeleine::Automatic::Interceptor
-  attr_reader :frames
-  automatic_read_only :frames
-  attr_accessor :frame_index
-  automatic_read_only :frame_index
-  attr_accessor :system_id
-  automatic_read_only :system_id
-
-  def initialize
-    @frames = {}
-    @frame_index = 0
-    @system_id = nil
-  end
-
-  def add(msgid, frame)
-    @frames[msgid] = frame
-  end
-
-  def delete(msgid)
-    @frames.delete(msgid)
-  end
-
-  def clear
-    @frames.clear
-  end
-  
-  automatic_read_only :lookup
-  def lookup(msgid)
-    @frames[msgid]
-  end
-end
-
-class FrameJournal
-  def initialize(directory='frame-journal', snap_freq = 60 * 5)
-    @directory = directory
-    @mad = AutomaticSnapshotMadeleine.new(directory) do
-      MadFrameJournal.new
-    end
-    
-    # always snap on startup, in case we had an previous failure
-    @modified = true
-    Thread.new(@mad, snap_freq) do |mad, freq|
-      while true
-        sleep(freq)
-        mad.take_snapshot if @modified
-        @modified = false
-      end
-    end
-  end
-  
-  def []=(msgid, frame)
-    @modified = true
-    @mad.system.add(msgid, frame)
-  end
-  
-  def [](msgid)
-    @mad.system.lookup(msgid)
-  end
-  
-  def delete(msgid)
-    @modified = true
-    @mad.system.delete(msgid)
-  end
-  
-  def keys
-    @mad.system.frames.keys
-  end
-  
-  def clear
-    @modified = true
-    @mad.system.clear
-    @mad.take_snapshot
-  end
-  
-  def index
-    @mad.system.frame_index
-  end
-  
-  def next_index
-    @modified = true
-    @mad.system.frame_index += 1
-  end
-  
-  def system_id
-    unless name = @mad.system.system_id
-      # todo - grab default name from some place smarter...
-      @modified = true
-      @mad.system.system_id = 'cmastomp'
-      name = @mad.system.system_id
-    end 
-    name
-  end
-end
-
-if __FILE__ == $0
-  fj = FrameJournal.new('fj', 3)
-  until ARGV.empty?
-    case cmd = ARGV.shift
-    when "keys"
-      puts fj.keys.inspect
-    when "dump"
-      fj.keys.each do |key|
-        puts "#{key}: #{fj[key]}"
-      end
-    when "show"
-      key = ARGV.shift
-      puts "#{key}: #{fj[key]}"
-    when "add"
-      key = ARGV.shift
-      val = ARGV.shift
-      fj[key] = val
-    when "sleep"
-      sleep ARGV.shift.to_i
-    when "delete"
-      fj.delete(ARGV.shift)
-    end
-  end
-end

data/lib/queue_manager.rb

@@ -1,81 +0,0 @@
-# queue - persistent sent to a single subscriber
-# queue_monitor - looks, but does not remove from queue
-
-class QueueManager
-  Struct::new('QueueUser', :user, :ack)
-  
-  def initialize(journal)
-    # read journal information
-    @journal = journal
-    @queues = Hash.new { Array.new }
-    @pending = Hash.new { Array.new }
-    @messages = Hash.new { Array.new }
-    
-    # recover from previous run
-    msgids = @journal.keys.sort
-    msgids.each do |msgid|
-      sendmsg(@journal[msgid])
-    end
-  end  
-
-  def subscribe(dest, user, use_ack=false)
-    user = Struct::QueueUser.new(user, use_ack)
-    @queues[dest] += [user]
-    
-    # TODO handle this is some form of call back
-    # it is quite possible that this could be a lot
-    # of data and block things up.
-    send_backlog(@messages[dest], user)
-  end
-  
-  def send_backlog(queue, user)
-    until queue.empty?
-      current = queue.first
-      send_to_user(current, user)
-      queue.shift
-    end 
-  end
-  
-  def unsubscribe(topic, user)
-    @queues[topic].delete_if { |u| u.user == user } 
-  end
-  
-  def ack(user, frame)
-    pending_size = @pending[user].size
-    msgid = frame.headers['message-id']
-    @pending[user].delete_if { |pf| pf.headers['message-id'] == msgid }
-    raise "Message (#{msgid}) not found" if pending_size == @pending[user].size
-    @journal.delete(msgid)
-  end
-
-  def disconnect(user)
-    @queues.each do |dest, queue|
-      queue.delete_if { |qu| qu.user == user }
-    end
-    @pending[user].each do |frame|
-      sendmsg(frame)
-    end
-  end
-    
-  def send_to_user(frame, user)
-    if user.ack
-      @pending[user.user] += [frame]
-    else
-      @journal.delete(frame.headers['message-id'])
-    end 
-    user.user.send_data(frame.to_s)
-  end
-  
-  def sendmsg(frame)
-    frame.command = "MESSAGE"
-    dest = frame.headers['destination']
-    @journal[frame.headers['message-id']] = frame
-
-    if user = @queues[dest].shift
-      send_to_user(frame, user)
-      @queues[dest].push(user)
-    else
-      @messages[dest] += [frame]
-    end
-  end  
-end

data/test/test_frame_journal.rb

@@ -1,14 +0,0 @@
-require 'frame_journal'
-require 'test/unit' unless defined? $ZENTEST and $ZENTEST
-require 'tesly'
-
-# how would you test this? It is pretty simple code
-# based almost completely on Madeleine...
-class TestFrameJournal < Test::Unit::TestCase
-  def setup
-  end  
-  
-  def test_something
-    
-  end
-end