diff options
Diffstat (limited to 'lib/killjoy/cassandra/non_blocking_writes_consumer.rb')
| -rw-r--r-- | lib/killjoy/cassandra/non_blocking_writes_consumer.rb | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/lib/killjoy/cassandra/non_blocking_writes_consumer.rb b/lib/killjoy/cassandra/non_blocking_writes_consumer.rb index d62ae89..23aa24a 100644 --- a/lib/killjoy/cassandra/non_blocking_writes_consumer.rb +++ b/lib/killjoy/cassandra/non_blocking_writes_consumer.rb @@ -3,6 +3,13 @@ require "killjoy/consumer" module Killjoy module Cassandra class NonBlockingWritesConsumer < Killjoy::Consumer + attr_reader :writers + + def initialize(writers, shard) + @writers = writers + super(shard) + end + def work(message) writes = writers.map do |writer| writer.write(message.to_hash, async: true) |