1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
module StompServer
class DBMQueue < Queue
def initialize *args
super
# Please don't use dbm files for storing large frames, it's problematic at best and uses large amounts of memory.
# sdbm croaks on marshalled data that contains certain characters, so we don't use it at all
@dbm = false
if RUBY_PLATFORM =~/linux|bsd/
types = ['bdb','dbm','gdbm']
else
types = ['bdb','gdbm']
end
types.each do |dbtype|
begin
require dbtype
@dbm = dbtype
puts "#{@dbm} loaded"
break
rescue LoadError => e
end
end
raise "No DBM library found. Tried bdb,dbm,gdbm" unless @dbm
@db = Hash.new
@queues.keys.each {|q| _open_queue(q)}
end
def dbmopen(dbname)
if @dbm == 'bdb'
BDB::Hash.new(dbname, nil, "a")
elsif @dbm == 'dbm'
DBM.open(dbname)
elsif @dbm == 'gdbm'
GDBM.open(dbname)
end
end
def _open_queue(dest)
queue_name = dest.gsub('/', '_')
dbname = @directory + '/' + queue_name
@db[dest] = Hash.new
@db[dest][:dbh] = dbmopen(dbname)
@db[dest][:dbname] = dbname
end
def _close_queue(dest)
@db[dest][:dbh].close
dbname = @db[dest][:dbname]
File.delete(dbname) if File.exist?(dbname)
File.delete("#{dbname}.db") if File.exist?("#{dbname}.db")
File.delete("#{dbname}.pag") if File.exist?("#{dbname}.pag")
File.delete("#{dbname}.dir") if File.exist?("#{dbname}.dir")
end
def _writeframe(dest,frame,msgid)
@db[dest][:dbh][msgid] = Marshal::dump(frame)
end
def _readframe(dest,msgid)
frame_image = @db[dest][:dbh][msgid]
Marshal::load(frame_image)
end
end
end
|
