Fix error

2 files changed, 5 insertions, 2 deletions

diff --git a/client/worker.rb b/client/worker.rb
index f83b090..7de9805 100644
--- a/client/worker.rb
+++ b/client/worker.rb
@@ -165,7 +165,7 @@ class Worker
             client.on_unauthorized do
               # revoked?
               $logger.warn("Unauthorized(##{client.user_id})")
-              send_chunk("UNAUTHORISED #{client.row_id}&#{client.user_id}")
+              send_chunk("UNAUTHORIZED #{client.row_id}&#{client.user_id}")
               client.stop_stream
               @clients.delete(client)
             end
diff --git a/lib/receiver/worker.rb b/lib/receiver/worker.rb
index 63f9f24..7c2c11e 100644
--- a/lib/receiver/worker.rb
+++ b/lib/receiver/worker.rb
@@ -94,7 +94,7 @@ class Receiver::Worker < DaemonSpawn::Base
     end
 
     def send_account_all
-      Account.where("user_id % ? = ?", $worker_count, @worker_number).each do |account|
+      Account.where("id % ? = ?", $worker_count, @worker_number).each do |account|
         puts "Sent #{account.id}/#{account.user_id}"
         send_account(account)
       end
@@ -134,6 +134,9 @@ class Receiver::Worker < DaemonSpawn::Base
             send_chunk("ERROR Invalid Secret Token")
             close_connection_after_writing
           end
+        when "UNAUTHORIZED"
+          $logger.warn("Unauthorized: #{arg.last}")
+          # unregister
         when "QUIT"
           $logger.info("Quit: #{@worker_number}")
           send_chunk("BYE")