Custom last ID

.gitignore

@@ -6,4 +6,5 @@
 /doc/
 /pkg/
 /spec/reports/
+/spec/examples.txt
 /tmp/

.ruby-version

@@ -0,0 +1 @@
+2.5.0

.travis.yml

@@ -1,6 +1,5 @@
 language: ruby
 rvm:
-  - 2.1.9
-  - 2.2.5
-  - 2.3.1
+  - 2.4.1
+  - 2.5.0
 before_install: gem update bundler

CHANGELOG.md

@@ -0,0 +1,9 @@
+# Changelog
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/)
+and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.html).
+
+## [Unreleased]
+### Added
+- Ability to specify `last_id` of a channel

Gemfile

@@ -2,3 +2,5 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in message_bus-client.gemspec
 gemspec
+
+gem "pry-byebug"

Rakefile

@@ -11,9 +11,18 @@ task :server do
   end
 
   at_exit do
-    $stderr.puts "Killing pid #{pid}"
-    Process.kill('KILL', pid)
-    Process.wait(pid)
+    if RbConfig::CONFIG["host_os"] =~ /mswin|mingw/
+      $stderr.puts "Killing pid #{pid}"
+      Process.kill('KILL', pid)
+      Process.wait(pid)
+    else
+      child_pid = `pgrep -P #{pid}`.to_i
+      pids = [child_pid, pid]
+      pids.each do |pid|
+        $stderr.puts "Killing pid #{pid}"
+        Process.kill('KILL', pid)
+      end
+    end
   end
 
   sleep 3

lib/message_bus/client.rb

@@ -1,5 +1,6 @@
 require 'excon'
 require 'json'
+require "securerandom"
 
 require 'message_bus/client/version'
 require 'message_bus/client/configuration'

lib/message_bus/client/message_handler.rb

@@ -1,6 +1,6 @@
 module MessageBus::Client::MessageHandler
   SubscribedChannel = Struct.new(:callbacks, :last_id) do
-    def initialize(last_id = -1)
+    def initialize(last_id)
       self.callbacks = []
       self.last_id = last_id
     end
@@ -20,14 +20,16 @@ def initialize(base_url)
 
     @pending_messages = []
     @subscribed_channels = {}
-    @subscribed_channels.default_proc = proc do |hash, key|
-      hash[key] = SubscribedChannel.new
-    end
     @payload = String.new
   end
 
-  def subscribe(channel, &callback)
-    @subscribed_channels[channel].callbacks << callback
+  def subscribe(channel, last_id=-1, &callback)
+    subscribed_channel = @subscribed_channels[channel]
+    if subscribed_channel.nil?
+      subscribed_channel = @subscribed_channels[channel] =
+        SubscribedChannel.new(last_id)
+    end
+    subscribed_channel.callbacks << callback
   end
 
   def unsubscribe

spec/message_bus/client_spec.rb

@@ -37,7 +37,7 @@ def write_message(message, user = 'message_bus-client')
       end
     end
 
-    it 'receives messages' do
+    it 'receives new messages by default (last_id of -1)' do
       subject.start
 
       text = "Hello World! #{Random.rand}"
@@ -72,7 +72,7 @@ def write_message(message, user = 'message_bus-client')
       subject.stop
     end
 
-    it 'receives messages' do
+    it 'receives new messages by default (last_id of -1)' do
       subject.start
 
       text = "Hello World! #{Random.rand}"

spec/spec_helper.rb

@@ -1,6 +1,6 @@
 $LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
-require 'coverage_helper'
 require 'message_bus/client'
+require "pry-byebug"
 
 # This file was generated by the `rspec --init` command. Conventionally, all
 # specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.