-
Notifications
You must be signed in to change notification settings - Fork 51
/
redis.ru
70 lines (65 loc) · 2.45 KB
/
redis.ru
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
69
70
# This example implements a Redis pub/sub engine according to the Iodine::PubSub::Engine specifications.
#
# Run this applications on two ports, in two terminals to see the synchronization is action:
#
# IODINE_REDIS_URL=redis://localhost:6379/0 iodine -t 1 -p 3000 redis.ru
# IODINE_REDIS_URL=redis://localhost:6379/0 iodine -t 1 -p 3030 redis.ru
#
# Or:
#
# iodine -t 1 -p 3000 redis.ru -redis redis://localhost:6379/0
# iodine -t 1 -p 3030 redis.ru -redis redis://localhost:6379/0
#
require 'iodine'
# initialize the Redis engine for each Iodine process.
if Iodine::DEFAULT_HTTP_ARGS[:redis_]
puts "* Redis support automatically detected."
else
puts "* No Redis, it's okay, pub/sub will support the process cluster."
end
# A simple router - Checks for Websocket Upgrade and answers HTTP.
module MyHTTPRouter
# This is the HTTP response object according to the Rack specification.
HTTP_RESPONSE = [200, { 'Content-Type' => 'text/html',
'Content-Length' => '32' },
['Please connect using websockets.']]
WS_RESPONSE = [0, {}, []]
# this is function will be called by the Rack server (iodine) for every request.
def self.call env
# check if this is an upgrade request.
if(env['rack.upgrade?'.freeze])
puts "SSE connections will not be able te send data, just listen." if(env['rack.upgrade?'.freeze] == :sse)
env['rack.upgrade'.freeze] = WS_RedisPubSub.new(env['PATH_INFO'] && env['PATH_INFO'].length > 1 ? env['PATH_INFO'][1..-1] : "guest")
return WS_RESPONSE
end
# simply return the RESPONSE object, no matter what request was received.
HTTP_RESPONSE
end
end
# A simple Websocket Callback Object.
class WS_RedisPubSub
def initialize name
@name = name
end
# seng a message to new clients.
def on_open client
client.subscribe "chat"
# let everyone know we arrived
client.publish "chat", "#{@name} entered the chat."
end
# send a message, letting the client know the server is suggunt down.
def on_shutdown client
client.write "Server shutting down. Goodbye."
end
# perform the echo
def on_message client, data
client.publish "chat", "#{@name}: #{data}"
end
def on_close client
# let everyone know we left
client.publish "chat", "#{@name} left the chat."
# we don't need to unsubscribe, subscriptions are cleared automatically once the connection is closed.
end
end
# this function call links our HelloWorld application with Rack
run MyHTTPRouter