riak-client 2.4.1 → 2.5.0

data/lib/riak/crdt/hyper_log_log.rb

@@ -0,0 +1,97 @@
+require 'riak/util/string'
+
+module Riak
+  module Crdt
+
+    # A distributed set containing strings, using the Riak 2 Data Types feature and Hyper Log Log algorithm
+    class HyperLogLog < Base
+      include Util::String
+
+      # Create a HLL instance. The bucket type is determined by the first of
+      # these sources:
+      #
+      # 1. The `bucket_type` String argument
+      # 2. A {BucketTyped::Bucket} as the `bucket` argument
+      # 3. The `Crdt::Base::DEFAULT_BUCKET_TYPES[:hll]` entry
+      #
+      # @param bucket [Bucket] the {Riak::Bucket} for this set
+      # @param [String, nil] key The name of the set. A nil key makes
+      #        Riak assign a key.
+      # @param [String] bucket_type The bucket type for this HLL datatype
+      # @param options [Hash]
+      def initialize(bucket, key, bucket_type = nil, options = {})
+        super(bucket, key, bucket_type || :hll, options)
+      end
+
+      # Gets the current HLL value from Riak
+      #
+      # @return [Integer]
+      def value
+        reload if dirty?
+        @value
+      end
+      alias :cardinality :value
+
+      def batch
+        batcher = BatchHyperLogLog.new self
+
+        yield batcher
+
+        operate batcher.operations
+      end
+
+      # Add a {String} to the {Riak::Crdt::HyperLogLog}
+      #
+      # @param [String] element the element to add to the set
+      # @param [Hash] options
+      def add(element, options = {})
+        operate operation(:add, element), options
+      end
+
+      def pretty_print(pp)
+        super pp do
+          pp.comma_breakable
+          pp.pp to_a
+        end
+      end
+
+      private
+      def vivify(value)
+        @value = value
+      end
+
+      def operation(direction, element)
+        Operation::Update.new.tap do |op|
+          op.type = :hll
+          op.value = { direction => element }
+        end
+      end
+
+      class BatchHyperLogLog
+        def initialize(base)
+          @base = base
+          @adds = ::Set.new
+        end
+
+        def add(element)
+          @adds.add element
+        end
+
+        def to_a
+          @adds.to_a
+        end
+
+        def value
+          @adds
+        end
+
+        def operations
+          Operation::Update.new.tap do |op|
+            op.type = :hll
+            op.value = {add: @adds.to_a}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/riak/crdt/operation.rb

@@ -1,19 +1,19 @@
-module Riak
-  module Crdt
-    # @api private
-    module Operation
-      class Update
-        attr_accessor :parent
-        attr_accessor :name
-        attr_accessor :type
-        attr_accessor :value
-      end
-
-      class Delete
-        attr_accessor :parent
-        attr_accessor :name
-        attr_accessor :type
-      end
-    end
-  end
-end
+module Riak
+  module Crdt
+    # @api private
+    module Operation
+      class Update
+        attr_accessor :parent
+        attr_accessor :name
+        attr_accessor :type
+        attr_accessor :value
+      end
+
+      class Delete
+        attr_accessor :parent
+        attr_accessor :name
+        attr_accessor :type
+      end
+    end
+  end
+end

data/lib/riak/encoding.rb

@@ -1,6 +1,6 @@
-if defined? Encoding
-  Encoding.default_internal = "UTF-8" if Encoding.default_internal.nil? ||
-                                        !Encoding.default_internal.ascii_compatible?
-else
-  $KCODE = "U"
-end
+if defined? Encoding
+  Encoding.default_internal = "UTF-8" if Encoding.default_internal.nil? ||
+                                        !Encoding.default_internal.ascii_compatible?
+else
+  $KCODE = "U"
+end

data/lib/riak/errors/backend_creation.rb

@@ -1,9 +1,9 @@
-require 'riak/errors/base'
-
-module Riak
-  class BackendCreationError < Error
-    def initialize(backend)
-      super t('protobuffs_configuration', backend: backend)
-    end
-  end
-end
+require 'riak/errors/base'
+
+module Riak
+  class BackendCreationError < Error
+    def initialize(backend)
+      super t('protobuffs_configuration', backend: backend)
+    end
+  end
+end

data/lib/riak/errors/connection_error.rb

@@ -1,50 +1,50 @@
-require 'riak/errors/base'
-
-module Riak
-  class ConnectionError < Error
-  end
-
-  class TlsError < ConnectionError
-    class SslVersionConfigurationError < TlsError
-      def initialize
-        super t('ssl.version_configuration_error')
-      end
-    end
-
-    class CertHostMismatchError < TlsError
-      def initialize
-        super t('ssl.cert_host_mismatch')
-      end
-    end
-
-    class CertNotValidError < TlsError
-      def initialize
-        super t('ssl.cert_not_valid')
-      end
-    end
-
-    class CertRevokedError < TlsError
-      def initialize
-        super t('ssl.cert_revoked')
-      end
-    end
-
-    class ReadDataError < TlsError
-      def initialize(actual, candidate)
-        super t('ssl.read_data_error', actual: actual, candidate: candidate)
-      end
-    end
-
-    class UnknownKeyTypeError < TlsError
-      def initialize
-        super t('ssl.unknown_key_type')
-      end
-    end
-  end
-
-  class UserConfigurationError < ConnectionError
-    def initialize
-      super t('pbc.user_not_username')
-    end
-  end
-end
+require 'riak/errors/base'
+
+module Riak
+  class ConnectionError < Error
+  end
+
+  class TlsError < ConnectionError
+    class SslVersionConfigurationError < TlsError
+      def initialize
+        super t('ssl.version_configuration_error')
+      end
+    end
+
+    class CertHostMismatchError < TlsError
+      def initialize
+        super t('ssl.cert_host_mismatch')
+      end
+    end
+
+    class CertNotValidError < TlsError
+      def initialize
+        super t('ssl.cert_not_valid')
+      end
+    end
+
+    class CertRevokedError < TlsError
+      def initialize
+        super t('ssl.cert_revoked')
+      end
+    end
+
+    class ReadDataError < TlsError
+      def initialize(actual, candidate)
+        super t('ssl.read_data_error', actual: actual, candidate: candidate)
+      end
+    end
+
+    class UnknownKeyTypeError < TlsError
+      def initialize
+        super t('ssl.unknown_key_type')
+      end
+    end
+  end
+
+  class UserConfigurationError < ConnectionError
+    def initialize
+      super t('pbc.user_not_username')
+    end
+  end
+end

data/lib/riak/errors/protobuffs_error.rb

@@ -1,11 +1,11 @@
-require 'riak/errors/base'
-module Riak
-  class ProtobuffsError < Error
-  end
-
-  class ProtobuffsFailedHeader < ProtobuffsError
-    def initialize
-      super t('pbc.failed_header')
-    end
-  end
-end
+require 'riak/errors/base'
+module Riak
+  class ProtobuffsError < Error
+  end
+
+  class ProtobuffsFailedHeader < ProtobuffsError
+    def initialize
+      super t('pbc.failed_header')
+    end
+  end
+end

data/lib/riak/i18n.rb

@@ -1,7 +1,7 @@
-require 'i18n'
-
-I18n.config.enforce_available_locales = true unless I18n.config.enforce_available_locales == false
-
-Dir.glob(File.expand_path("../locale/*.yml", __FILE__)).each do |locale_file|
-  I18n.load_path << locale_file
-end
+require 'i18n'
+
+I18n.config.enforce_available_locales = true unless I18n.config.enforce_available_locales == false
+
+Dir.glob(File.expand_path("../locale/*.yml", __FILE__)).each do |locale_file|
+  I18n.load_path << locale_file
+end

data/lib/riak/instrumentation.rb

@@ -1,6 +1,6 @@
-begin
-  require 'instrumentable'
-  require 'riak/client/instrumentation'
-rescue LoadError => e
-  # Go quietly into the night...(?)
-end
+begin
+  require 'instrumentable'
+  require 'riak/client/instrumentation'
+rescue LoadError => e
+  # Go quietly into the night...(?)
+end

data/lib/riak/json.rb

@@ -1,52 +1,52 @@
-require 'multi_json'
-if MultiJson.respond_to?(:adapter)
-  MultiJson.adapter
-else
-  MultiJson.engine # Force loading of an engine
-end
-require 'riak/core_ext/json'
-
-module Riak
-  class << self
-    # Options that will be passed to the JSON parser and encoder.
-    # Defaults to `{:max_nesting => 20}`
-    attr_accessor :json_options
-  end
-  self.json_options = {:max_nesting => 20}
-
-  # JSON module for internal use inside riak-client
-  module JSON
-    class << self
-      if MultiJson.respond_to?(:dump) # MultiJson 1.2 or later
-        # Parse a JSON string
-        # @param [String] str a JSON payload
-        # @return [Array,Hash] a Ruby object decoded from the JSON payload
-        def parse(str)
-          MultiJson.load(str, Riak.json_options)
-        end
-
-        # Generate a JSON string
-        # @param [Array, Hash] obj an object to JSON-encode
-        # @return [String] a JSON payload
-        def encode(obj)
-          MultiJson.dump(obj)
-        end
-      else
-        # Parse a JSON string
-        # @param [String] str a JSON payload
-        # @return [Array,Hash] a Ruby object decoded from the JSON payload
-        def parse(str)
-          MultiJson.decode(str, Riak.json_options)
-        end
-
-        # Generate a JSON string
-        # @param [Array, Hash] obj an object to JSON-encode
-        # @return [String] a JSON payload
-        def encode(obj)
-          MultiJson.encode(obj)
-        end
-      end
-      alias :dump :encode
-    end
-  end
-end
+require 'multi_json'
+if MultiJson.respond_to?(:adapter)
+  MultiJson.adapter
+else
+  MultiJson.engine # Force loading of an engine
+end
+require 'riak/core_ext/json'
+
+module Riak
+  class << self
+    # Options that will be passed to the JSON parser and encoder.
+    # Defaults to `{:max_nesting => 20}`
+    attr_accessor :json_options
+  end
+  self.json_options = {:max_nesting => 20}
+
+  # JSON module for internal use inside riak-client
+  module JSON
+    class << self
+      if MultiJson.respond_to?(:dump) # MultiJson 1.2 or later
+        # Parse a JSON string
+        # @param [String] str a JSON payload
+        # @return [Array,Hash] a Ruby object decoded from the JSON payload
+        def parse(str)
+          MultiJson.load(str, Riak.json_options)
+        end
+
+        # Generate a JSON string
+        # @param [Array, Hash] obj an object to JSON-encode
+        # @return [String] a JSON payload
+        def encode(obj)
+          MultiJson.dump(obj)
+        end
+      else
+        # Parse a JSON string
+        # @param [String] str a JSON payload
+        # @return [Array,Hash] a Ruby object decoded from the JSON payload
+        def parse(str)
+          MultiJson.decode(str, Riak.json_options)
+        end
+
+        # Generate a JSON string
+        # @param [Array, Hash] obj an object to JSON-encode
+        # @return [String] a JSON payload
+        def encode(obj)
+          MultiJson.encode(obj)
+        end
+      end
+      alias :dump :encode
+    end
+  end
+end

data/lib/riak/list_buckets.rb

@@ -1,28 +1,28 @@
-module Riak
-  class ListBuckets
-    def initialize(client, options, block)
-      @client = client
-      @block = block
-      @options = options
-      perform_request
-    end
-
-    def perform_request
-      @client.backend do |be|
-        be.list_buckets @options, &wrapped_block
-      end
-    end
-
-    private
-
-    def wrapped_block
-      proc do |bucket_names|
-        next if bucket_names.nil?
-        bucket_names.each do |bucket_name|
-          bucket = @client.bucket bucket_name
-          @block.call bucket
-        end
-      end
-    end
-  end
-end
+module Riak
+  class ListBuckets
+    def initialize(client, options, block)
+      @client = client
+      @block = block
+      @options = options
+      perform_request
+    end
+
+    def perform_request
+      @client.backend do |be|
+        be.list_buckets @options, &wrapped_block
+      end
+    end
+
+    private
+
+    def wrapped_block
+      proc do |bucket_names|
+        next if bucket_names.nil?
+        bucket_names.each do |bucket_name|
+          bucket = @client.bucket bucket_name
+          @block.call bucket
+        end
+      end
+    end
+  end
+end