tecepe 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,8 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in tecepe.gemspec
+gemspec
+
+group :test do
+  gem 'rspec'
+end

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Ismael Celis
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,51 @@
+# Tecepe
+
+Launch small (evented) TCP JSON services on a given host:port
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'tecepe'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install tecepe
+
+## Usage
+
+```ruby
+Worker.listen("localhost", 5555) do
+  
+  on 'perms' do |json|
+    redis.smember("permissions:#{json['user_id']}", json['product_id']) do |r|
+      reply allowed: r
+    end
+  end
+
+end
+```
+
+In reality you would do some more computation than just proxying Redis!
+
+## Clients
+
+This is simple enough that it should be easy to write clients in different langs/stacks. See lib/tecepe/blocking_client.rb for a reference implementation.
+
+```ruby
+PEMISSIONS_SERVICE = Tecepe::BlockingClient.new('localhost', 5555)
+
+PEMISSIONS_SERVICE.call 'perms', user_id: current_user.id, @product.id # true/false
+```
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/examples/blocking_client.rb

@@ -0,0 +1,19 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'tecepe/blocking_client'
+
+socket = Tecepe::BlockingClient.new('localhost', 5555)
+
+delay = ARGV[0]
+
+m = <<-eos
+Hello this is a text
+
+with line changes
+And more
+eos
+
+20.times do |i|
+  response = socket.call 'perms', delay: delay.to_f , m: m        # Send request
+  puts "RESPONSE: #{response.class.name} - #{response}"
+end

data/examples/latency.rb

@@ -0,0 +1,13 @@
+$:.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'tecepe'
+
+Tecepe.listen("localhost", 5555) do
+  
+  on 'perms' do |json|
+    EventMachine::Timer.new(json['delay'].to_f) do
+      reply message: ">>>you sent: #{json}\n"
+    end
+  end
+
+end

data/lib/tecepe.rb

@@ -0,0 +1,32 @@
+require "tecepe/version"
+require "tecepe/connection"
+
+module Tecepe
+  def self.events
+    @events ||= {}
+  end
+  
+  def self.on(event, &handler)
+    events[event] = handler
+  end
+  
+  def self.dispatch(connection, event_name, payload)
+    if handler = events[event_name]
+      connection.instance_exec payload, &handler
+    else
+      connection.send_error "No handler for event #{event_name}"
+    end
+  end
+  
+  def self.listen(host, port, &block)
+    instance_eval &block
+    puts "[listen] #{host}:#{port}"
+    EventMachine::run {
+      EventMachine::start_server host, port, Tecepe::Connection
+    }
+  end
+  
+  def self.clear_handlers
+    @events = {}
+  end
+end

data/lib/tecepe/blocking_client.rb

@@ -0,0 +1,87 @@
+require 'socket'
+require 'json'
+require "timeout"
+
+# Simple (blocking) client with basic reconnect functionality
+# Some code taken from Redis client
+# https://github.com/redis/redis-rb
+#
+# EXAMPLE:
+# socket = Tecepe::BlockingClient.new('localhost', 5555)
+# 
+# 20.times do |i|
+#   response = socket.call 'perms', user_id: 3          # Send request
+#   puts "RESPONSE: #{response}"
+# end
+#
+module Tecepe
+  
+  class BlockingClient
+
+    def initialize(host, port, cid = Process.pid)
+      @host, @port, @cid = host, port, cid
+      @socket = nil
+      @reconnect = true
+      connect
+    end
+
+    def connected?
+      !! @socket
+    end
+
+    def call(event_name, payload)
+      json = JSON.generate(event: event_name, cid: @cid, payload: payload)
+      ensure_connected do
+        @socket.print "#{json}\n"
+        JSON.parse(@socket.gets)
+      end
+    end
+
+    private
+
+    def connect
+      with_timeout 1 do
+        @socket = TCPSocket.new(@host, @port)
+        # @socket.setsockopt Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1
+        puts "[open]"
+      end
+    end
+
+    def disconnect
+      return false unless connected?
+      @socket.close
+    rescue
+    ensure
+      puts "[close]"
+      @socket = nil
+    end
+
+    def with_timeout(seconds, &block)
+      Timeout.timeout(seconds, &block)
+    end
+
+    def ensure_connected(&block)
+      tries = 0
+
+      begin
+        connect unless connected?
+        tries += 1
+
+        yield
+      rescue Errno::ECONNRESET, Errno::EPIPE, Errno::ECONNABORTED, Errno::EBADF, Errno::EINVAL
+        disconnect
+
+        if tries < 2 && @reconnect
+          retry
+        else
+          raise Errno::ECONNRESET
+        end
+      rescue Exception
+        disconnect
+        raise
+      end
+    end
+
+  end
+  
+end

data/lib/tecepe/connection.rb

@@ -0,0 +1,71 @@
+require 'eventmachine'
+require 'json'
+
+module Tecepe
+  
+  module Connection
+    
+    HEARTBEAT = 5.freeze
+    
+    def post_init
+      @heartbeat = setup_heartbeat
+      @cid = nil
+      log :conn, signature
+      super
+    end
+
+    def unbind
+      log :bye
+      @heartbeat.cancel
+      super
+    end
+    
+    def receive_data data
+      log :data, data
+      (@buffer ||= BufferedTokenizer.new).extract(data).each do |line|
+        receive_line(line)
+      end
+    end
+    
+    def receive_line data
+       begin
+         json = JSON.parse(data)
+         @cid = json['cid']
+         log :rcvd, json['payload']
+         Tecepe.dispatch self, json['event'], json['payload']
+       rescue JSON::ParserError => e
+         send_error e.message
+       rescue Encoding::UndefinedConversionError => e
+         send_error e.message
+       end
+    end
+
+    def reply(payload = {}, status = 1)
+      json = JSON.generate(status: status, payload: payload)
+      log :repl, json
+      if error?
+        log :error
+      else
+        send_data "#{json}\n"
+      end
+    end
+
+    def send_error(msg)
+      reply({message: msg}, -1)
+    end
+
+    private
+
+    def setup_heartbeat
+      EventMachine::PeriodicTimer.new(HEARTBEAT) do
+        log :heartbeat
+        send_data ''
+      end
+    end
+    
+    def log(key, msg = '')
+      puts "#{Process.pid} [#{key} #{@cid}] #{msg}"
+    end
+  end
+  
+end

data/lib/tecepe/version.rb

@@ -0,0 +1,3 @@
+module Tecepe
+  VERSION = "0.0.1"
+end

data/spec/connection_spec.rb

@@ -0,0 +1,69 @@
+require File.dirname(__FILE__) + '/spec_helper'
+
+describe Tecepe::Connection do
+  
+  def message(args)
+    "#{JSON.generate(args)}\n"
+  end
+  
+  describe "#receive_data" do
+    
+    before do
+      @test_klass = Class.new do
+        include Tecepe::Connection
+        
+        attr_reader :replies
+        
+        def send_data raw_data
+          (@replies ||= []) << JSON.parse(raw_data)
+        end
+        
+        def error?
+          false
+        end
+      end
+      
+      @connection = @test_klass.new
+    end
+    
+    after do
+      Tecepe.clear_handlers
+    end
+    
+    it "triggers event callbacks" do
+      Tecepe.on 'test1' do |json|
+        json['name'].should == 'ismael'
+      end
+      @connection.receive_data message(event: 'test1', payload: {name: 'ismael'})
+    end
+    
+    it "replies in event callbacks" do
+      Tecepe.on 'test1' do |json|
+        reply message: "event received!", by: json['name']
+      end
+      @connection.receive_data message(event: 'test1', payload: {name: 'ismael'})
+      @connection.replies.last['status'].should == 1
+      @connection.replies.last['payload']['message'].should == 'event received!'
+      @connection.replies.last['payload']['by'].should == 'ismael'
+    end
+    
+    it 'waits for line breaks to parse messages' do
+      Tecepe.on 'test1' do |json|
+        reply message: "event received!", pong: json['message']
+      end
+      @connection.receive_data '{"event":"test1"'
+      @connection.receive_data ',"payload":{"message":"hola!"}'
+      @connection.receive_data "}\n"
+      
+      @connection.replies.last['payload']['message'].should == 'event received!'
+      @connection.replies.last['status'].should == 1
+      @connection.replies.last['payload']['pong'].should == 'hola!'
+    end
+    
+    it 'replies with error if no handler found' do
+      @connection.receive_data message(event: 'foo', payload: {name: 'ismael'})
+      @connection.replies.last['status'].should == -1
+      @connection.replies.last['payload']['message'].should == 'No handler for event foo'
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,30 @@
+require 'rubygems'
+require 'bundler'
+Bundler.setup
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'tecepe'
+
+
+require 'rspec'
+
+module Support
+  
+  # Forks process and launches a service
+  def fork_service(&block)
+    begin
+      pid = fork do
+        trap('TERM') { exit }
+        sleep 1
+        yield
+      end
+    ensure
+      if pid
+        Process.kill("TERM", pid)
+        Process.wait(pid)
+      end
+    end
+  end
+end

data/tecepe.gemspec

@@ -0,0 +1,19 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/tecepe/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Ismael Celis"]
+  gem.email         = ["ismaelct@gmail.com"]
+  gem.description   = %q{Tiny evented TCP server for JSON services}
+  gem.summary       = %q{Tiny evented TCP server for JSON services}
+  gem.homepage      = ""
+  
+  gem.add_dependency 'eventmachine'
+  gem.add_dependency 'json'
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  gem.name          = "tecepe"
+  gem.require_paths = ["lib"]
+  gem.version       = Tecepe::VERSION
+end

metadata

@@ -0,0 +1,90 @@
+--- !ruby/object:Gem::Specification 
+name: tecepe
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Ismael Celis
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-06-10 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: json
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+description: Tiny evented TCP server for JSON services
+email: 
+- ismaelct@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- examples/blocking_client.rb
+- examples/latency.rb
+- lib/tecepe.rb
+- lib/tecepe/blocking_client.rb
+- lib/tecepe/connection.rb
+- lib/tecepe/version.rb
+- spec/connection_spec.rb
+- spec/spec_helper.rb
+- tecepe.gemspec
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: Tiny evented TCP server for JSON services
+test_files: 
+- spec/connection_spec.rb
+- spec/spec_helper.rb