lib/riak/stamp.rb in riak-client-2.4.1 vs lib/riak/stamp.rb in riak-client-2.5.0

- old
+ new

@@ -1,77 +1,77 @@ -require 'riak/client' -require 'riak/util/translation' -require 'thread' - -module Riak - # Implements a client-side form of monotonically-increasing k-sorted - # unique identifiers. These are useful for key generation if your - # data is time-sequential and needs to be sorted by key, perhaps in - # Riak Search. Inspired by Twitter's Snowflake project. - class Stamp - attr_reader :client - - CLIENT_ID_MASK = (1 << 10) - 1 - SEQUENCE_MASK = (1 << 12) - 1 - TIMESTAMP_MASK = (1 << 41) - 1 - SEQUENCE_SHIFT = 10 - TIMESTAMP_SHIFT = 22 - - # @param [Client] client a {Riak::Client} which will be used for - # the "worker ID" component of the stamp. - # @see Client#stamp - def initialize(client) - @client = client - @mutex = Mutex.new - @timestamp = time_gen - @sequence = 0 - end - - # Generates a k-sorted unique ID for use as a key or other - # disambiguation purposes. - def next - @mutex.synchronize do - now = time_gen - if @timestamp == now - @sequence = (@sequence + 1) & SEQUENCE_MASK - now = wait_for_next_ms(@timestamp) if @sequence == 0 - else - @sequence = 0 - end - - raise BackwardsClockError.new(@timestamp - now) if now < @timestamp - - @timestamp = now - @timestamp << TIMESTAMP_SHIFT | @sequence << SEQUENCE_SHIFT | client_id - end - end - - private - def client_id - case id = @client.client_id - when Integer - id & CLIENT_ID_MASK - else - id.hash & CLIENT_ID_MASK - end - end - - def time_gen - (Time.now.to_f * 1000).floor & TIMESTAMP_MASK - end - - def wait_for_next_ms(start) - now = time_gen - now = time_gen while now <= start - now - end - end - - # Raised when calling {Stamp#next} and NTP or some other external - # event has moved the system clock backwards. - class BackwardsClockError < StandardError - include Util::Translation - def initialize(delay) - super t('backwards_clock', :delay => delay) - end - end -end +require 'riak/client' +require 'riak/util/translation' +require 'thread' + +module Riak + # Implements a client-side form of monotonically-increasing k-sorted + # unique identifiers. These are useful for key generation if your + # data is time-sequential and needs to be sorted by key, perhaps in + # Riak Search. Inspired by Twitter's Snowflake project. + class Stamp + attr_reader :client + + CLIENT_ID_MASK = (1 << 10) - 1 + SEQUENCE_MASK = (1 << 12) - 1 + TIMESTAMP_MASK = (1 << 41) - 1 + SEQUENCE_SHIFT = 10 + TIMESTAMP_SHIFT = 22 + + # @param [Client] client a {Riak::Client} which will be used for + # the "worker ID" component of the stamp. + # @see Client#stamp + def initialize(client) + @client = client + @mutex = Mutex.new + @timestamp = time_gen + @sequence = 0 + end + + # Generates a k-sorted unique ID for use as a key or other + # disambiguation purposes. + def next + @mutex.synchronize do + now = time_gen + if @timestamp == now + @sequence = (@sequence + 1) & SEQUENCE_MASK + now = wait_for_next_ms(@timestamp) if @sequence == 0 + else + @sequence = 0 + end + + raise BackwardsClockError.new(@timestamp - now) if now < @timestamp + + @timestamp = now + @timestamp << TIMESTAMP_SHIFT | @sequence << SEQUENCE_SHIFT | client_id + end + end + + private + def client_id + case id = @client.client_id + when Integer + id & CLIENT_ID_MASK + else + id.hash & CLIENT_ID_MASK + end + end + + def time_gen + (Time.now.to_f * 1000).floor & TIMESTAMP_MASK + end + + def wait_for_next_ms(start) + now = time_gen + now = time_gen while now <= start + now + end + end + + # Raised when calling {Stamp#next} and NTP or some other external + # event has moved the system clock backwards. + class BackwardsClockError < StandardError + include Util::Translation + def initialize(delay) + super t('backwards_clock', :delay => delay) + end + end +end