Action Cable take#1
This commit is contained in:
commit
db568553d1
|
@ -0,0 +1,4 @@
|
||||||
|
source 'http://rubygems.org'
|
||||||
|
gemspec
|
||||||
|
|
||||||
|
gem 'cramp', github: "lifo/cramp"
|
|
@ -0,0 +1,3 @@
|
||||||
|
# ActionCable
|
||||||
|
|
||||||
|
Action Cable is a framework for realtime communication over websockets.
|
|
@ -0,0 +1,19 @@
|
||||||
|
Gem::Specification.new do |s|
|
||||||
|
s.platform = Gem::Platform::RUBY
|
||||||
|
s.name = 'action_cable'
|
||||||
|
s.version = '0.0.1'
|
||||||
|
s.summary = 'Framework for websockets.'
|
||||||
|
s.description = 'Action Cable is a framework for realtime communication over websockets.'
|
||||||
|
|
||||||
|
s.author = ['Pratik Naik']
|
||||||
|
s.email = ['pratiknaik@gmail.com']
|
||||||
|
s.homepage = 'http://basecamp.com'
|
||||||
|
|
||||||
|
s.add_dependency('activesupport', '~> 4.2.0')
|
||||||
|
s.add_dependency('cramp', '~> 0.15.4')
|
||||||
|
|
||||||
|
s.files = Dir['README', 'lib/**/*']
|
||||||
|
s.has_rdoc = false
|
||||||
|
|
||||||
|
s.require_path = 'lib'
|
||||||
|
end
|
|
@ -0,0 +1,12 @@
|
||||||
|
require 'cramp'
|
||||||
|
require 'active_support'
|
||||||
|
require 'active_support/json'
|
||||||
|
require 'active_support/concern'
|
||||||
|
require 'active_support/core_ext/hash/indifferent_access'
|
||||||
|
|
||||||
|
module ActionCable
|
||||||
|
VERSION = '0.0.1'
|
||||||
|
|
||||||
|
autoload :Channel, 'action_cable/channel'
|
||||||
|
autoload :Server, 'action_cable/server'
|
||||||
|
end
|
|
@ -0,0 +1,6 @@
|
||||||
|
module ActionCable
|
||||||
|
module Channel
|
||||||
|
autoload :Callbacks, 'action_cable/channel/callbacks'
|
||||||
|
autoload :Base, 'action_cable/channel/base'
|
||||||
|
end
|
||||||
|
end
|
|
@ -0,0 +1,64 @@
|
||||||
|
module ActionCable
|
||||||
|
module Channel
|
||||||
|
|
||||||
|
class Base
|
||||||
|
include Callbacks
|
||||||
|
|
||||||
|
on_subscribe :start_periodic_timers
|
||||||
|
on_unsubscribe :stop_periodic_timers
|
||||||
|
|
||||||
|
attr_reader :params
|
||||||
|
|
||||||
|
class << self
|
||||||
|
def matches?(identifier)
|
||||||
|
raise "Please implement #{name}#matches? method"
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def initialize(connection, channel_identifier, params = {})
|
||||||
|
@connection = connection
|
||||||
|
@channel_identifier = channel_identifier
|
||||||
|
@_active_periodic_timers = []
|
||||||
|
@params = params
|
||||||
|
|
||||||
|
setup
|
||||||
|
end
|
||||||
|
|
||||||
|
def receive(data)
|
||||||
|
raise "Not implemented"
|
||||||
|
end
|
||||||
|
|
||||||
|
def subscribe
|
||||||
|
self.class.on_subscribe_callbacks.each do |callback|
|
||||||
|
EM.next_tick { send(callback) }
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def unsubscribe
|
||||||
|
self.class.on_unsubscribe.each do |callback|
|
||||||
|
EM.next_tick { send(callback) }
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
protected
|
||||||
|
def setup
|
||||||
|
# Override in subclasses
|
||||||
|
end
|
||||||
|
|
||||||
|
def publish(data)
|
||||||
|
@connection.publish(data.merge(identifier: @channel_identifier).to_json)
|
||||||
|
end
|
||||||
|
|
||||||
|
def start_periodic_timers
|
||||||
|
self.class.periodic_timers.each do |method, options|
|
||||||
|
@_active_periodic_timers << EventMachine::PeriodicTimer.new(options[:every]) { send(method) }
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def stop_periodic_timers
|
||||||
|
@_active_periodic_timers.each {|t| t.cancel }
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
end
|
||||||
|
end
|
|
@ -0,0 +1,32 @@
|
||||||
|
module ActionCable
|
||||||
|
module Channel
|
||||||
|
|
||||||
|
module Callbacks
|
||||||
|
extend ActiveSupport::Concern
|
||||||
|
|
||||||
|
included do
|
||||||
|
class_attribute :on_subscribe_callbacks, :on_unsubscribe_callbacks, :periodic_timers, :instance_reader => false
|
||||||
|
|
||||||
|
self.on_subscribe_callbacks = []
|
||||||
|
self.on_unsubscribe_callbacks = []
|
||||||
|
self.periodic_timers = []
|
||||||
|
end
|
||||||
|
|
||||||
|
module ClassMethods
|
||||||
|
def on_subscribe(*methods)
|
||||||
|
self.on_subscribe_callbacks += methods
|
||||||
|
end
|
||||||
|
|
||||||
|
def on_unsubscribe(*methods)
|
||||||
|
self.on_unsubscribe_callbacks += methods
|
||||||
|
end
|
||||||
|
|
||||||
|
def periodic_timer(method, every:)
|
||||||
|
self.periodic_timers += [ [ method, every: every ] ]
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
end
|
||||||
|
|
||||||
|
end
|
||||||
|
end
|
|
@ -0,0 +1,73 @@
|
||||||
|
require 'set'
|
||||||
|
|
||||||
|
module ActionCable
|
||||||
|
class Server < Cramp::Websocket
|
||||||
|
on_start :initialize_subscriptions
|
||||||
|
on_data :received_data
|
||||||
|
on_finish :cleanup_subscriptions
|
||||||
|
|
||||||
|
class_attribute :registered_channels
|
||||||
|
self.registered_channels = Set.new
|
||||||
|
|
||||||
|
class << self
|
||||||
|
def register_channels(*channel_classes)
|
||||||
|
registered_channels.merge(channel_classes)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def initialize_subscriptions
|
||||||
|
@subscriptions = {}
|
||||||
|
end
|
||||||
|
|
||||||
|
def received_data(data)
|
||||||
|
data = ActiveSupport::JSON.decode data
|
||||||
|
|
||||||
|
case data['action']
|
||||||
|
when 'subscribe'
|
||||||
|
subscribe_channel(data)
|
||||||
|
when 'unsubscribe'
|
||||||
|
unsubscribe_channel(data)
|
||||||
|
when 'message'
|
||||||
|
process_message(data)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def cleanup_subscriptions
|
||||||
|
@subscriptions.each do |id, channel|
|
||||||
|
channel.unsubscribe
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def publish(data)
|
||||||
|
render data
|
||||||
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
def subscribe_channel(data)
|
||||||
|
id_key = data['identifier']
|
||||||
|
id_options = ActiveSupport::JSON.decode(id_key).with_indifferent_access
|
||||||
|
|
||||||
|
if subscription = registered_channels.detect { |channel_klass| channel_klass.matches?(id_options) }
|
||||||
|
@subscriptions[id_key] = subscription.new(self, id_key, id_options)
|
||||||
|
@subscriptions[id_key].subscribe
|
||||||
|
else
|
||||||
|
# No channel found
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def process_message(message)
|
||||||
|
id_key = message['identifier']
|
||||||
|
|
||||||
|
if @subscriptions[id_key]
|
||||||
|
@subscriptions[id_key].receive(ActiveSupport::JSON.decode message['data'])
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def unsubscribe_channel(data)
|
||||||
|
id_key = data['identifier']
|
||||||
|
@subscriptions[id_key].unsubscribe
|
||||||
|
@subscriptions.delete(id_key)
|
||||||
|
end
|
||||||
|
|
||||||
|
end
|
||||||
|
end
|
Loading…
Reference in New Issue