worker_node: specify user stream parameters in settings.yml

2 files changed, 6 insertions, 3 deletions

diff --git a/worker_node/lib/user_connection.rb b/worker_node/lib/user_connection.rb
index 2afcbe2..21717cd 100644
--- a/worker_node/lib/user_connection.rb
+++ b/worker_node/lib/user_connection.rb
@@ -2,7 +2,7 @@ class UserConnection
   def initialize(msg)
     @user_id = msg[:user_id]
     @account_id = msg[:id]
-    @client = UserStream::Client.new(msg.merge(params: { include_followings_activity: true }))
+    @client = UserStream::Client.new(msg.merge(params: Settings.user_stream_params, compression: Settings.user_stream_compression))
     setup_client
   end
 
diff --git a/worker_node/settings.yml.example b/worker_node/settings.yml.example
index 1ab9e29..b49d64d 100644
--- a/worker_node/settings.yml.example
+++ b/worker_node/settings.yml.example
@@ -1,5 +1,8 @@
-secret_key: secret key
+secret_key: secret_key
 collector_host: localhost
 collector_port: 42106
-log_level: debug
+log_level: info
 memcached: "127.0.0.1:11211"
+user_stream_compression: true
+user_stream_params:
+  replies: "all"