Class: Faye::Redis::MessageQueue

Inherits:
Object
  • Object
show all
Defined in:
lib/faye/redis/message_queue.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(connection, options = {}) ⇒ MessageQueue

Returns a new instance of MessageQueue.



9
10
11
12
 
# File 'lib/faye/redis/message_queue.rb', line 9

def initialize(connection, options = {})
  @connection = connection
  @options = options
end

Instance Attribute Details

#connection ⇒ Object (readonly)

Returns the value of attribute connection.



7
8
9
 
# File 'lib/faye/redis/message_queue.rb', line 7

def connection
  @connection
end

#options ⇒ Object (readonly)

Returns the value of attribute options.



7
8
9
 
# File 'lib/faye/redis/message_queue.rb', line 7

def options
  @options
end

Instance Method Details

#clear(client_id, &callback) ⇒ Object

Clear a client’s message queue



140
141
142
143
144
145
146
147
148
149
 
# File 'lib/faye/redis/message_queue.rb', line 140

def clear(client_id, &callback)
  @connection.with_redis do |redis|
    redis.del(queue_key(client_id))
  end

  EventMachine.next_tick { callback.call(true) } if callback
rescue => e
  log_error("Failed to clear queue for client #{client_id}: #{e.message}")
  EventMachine.next_tick { callback.call(false) } if callback
end

#dequeue_all(client_id, &callback) ⇒ Object

Dequeue all messages for a client



50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
 
# File 'lib/faye/redis/message_queue.rb', line 50

def dequeue_all(client_id, &callback)
  # Get all message IDs from queue
  message_ids = @connection.with_redis do |redis|
    redis.lrange(queue_key(client_id), 0, -1)
  end

  # Fetch all messages using pipeline
  messages = []
  unless message_ids.empty?
    @connection.with_redis do |redis|
      redis.pipelined do |pipeline|
        message_ids.each do |message_id|
          pipeline.hgetall(message_key(message_id))
        end
      end.each do |data|
        next if data.nil? || data.empty?

        # Parse JSON values
        parsed_data = data.transform_values do |v|
          begin
            JSON.parse(v)
          rescue JSON::ParserError
            v
          end
        end

        # Convert to Faye message format
        messages << {
          'channel' => parsed_data['channel'],
          'data' => parsed_data['data'],
          'clientId' => parsed_data['client_id'],
          'id' => parsed_data['id']
        }
      end
    end
  end

  # Delete queue and all message data using pipeline
  unless message_ids.empty?
    @connection.with_redis do |redis|
      redis.pipelined do |pipeline|
        pipeline.del(queue_key(client_id))
        message_ids.each do |message_id|
          pipeline.del(message_key(message_id))
        end
      end
    end
  end

  EventMachine.next_tick { callback.call(messages) } if callback
  messages
rescue => e
  log_error("Failed to dequeue messages for client #{client_id}: #{e.message}")
  EventMachine.next_tick { callback.call([]) } if callback
  []
end

#enqueue(client_id, message, &callback) ⇒ Object

Enqueue a message for a client



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
 
# File 'lib/faye/redis/message_queue.rb', line 15

def enqueue(client_id, message, &callback)
  message_id = generate_message_id
  timestamp = Time.now.to_i

  message_data = {
    id: message_id,
    channel: message['channel'],
    data: message['data'],
    client_id: message['clientId'],
    timestamp: timestamp
  }

  @connection.with_redis do |redis|
    redis.multi do |multi|
      # Store message data
      multi.hset(message_key(message_id), message_data.transform_keys(&:to_s).transform_values { |v| v.to_json })

      # Add message to client's queue
      multi.rpush(queue_key(client_id), message_id)

      # Set TTL on message
      multi.expire(message_key(message_id), message_ttl)

      # Set TTL on queue
      multi.expire(queue_key(client_id), message_ttl)
    end
  end

  EventMachine.next_tick { callback.call(true) } if callback
rescue => e
  log_error("Failed to enqueue message for client #{client_id}: #{e.message}")
  EventMachine.next_tick { callback.call(false) } if callback
end

#peek(client_id, limit = 10, &callback) ⇒ Object

Peek at messages without removing them



108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
 
# File 'lib/faye/redis/message_queue.rb', line 108

def peek(client_id, limit = 10, &callback)
  message_ids = @connection.with_redis do |redis|
    redis.lrange(queue_key(client_id), 0, limit - 1)
  end

  messages = message_ids.map do |message_id|
    fetch_message(message_id)
  end.compact

  EventMachine.next_tick { callback.call(messages) } if callback
  messages
rescue => e
  log_error("Failed to peek messages for client #{client_id}: #{e.message}")
  EventMachine.next_tick { callback.call([]) } if callback
  []
end

#size(client_id, &callback) ⇒ Object

Get queue size for a client



126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/faye/redis/message_queue.rb', line 126

def size(client_id, &callback)
  queue_size = @connection.with_redis do |redis|
    redis.llen(queue_key(client_id))
  end

  EventMachine.next_tick { callback.call(queue_size) } if callback
  queue_size
rescue => e
  log_error("Failed to get queue size for client #{client_id}: #{e.message}")
  EventMachine.next_tick { callback.call(0) } if callback
  0
end