couchbase 3.0.0.alpha.5-x86_64-linux → 3.5.1-x86_64-linux

data/lib/couchbase/config_profiles.rb

@@ -0,0 +1,55 @@
+#  Copyright 2020-2022 Couchbase, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+module Couchbase
+  module ConfigProfiles
+    class Profiles
+      attr_reader :profiles
+
+      def initialize
+        @profiles = {}
+        register_profile("wan_development", DevelopmentProfile.new)
+      end
+
+      def register_profile(name, profile)
+        @profiles[name] = profile
+      end
+
+      def apply(profile_name, options)
+        raise ArgumentError, "#{profile_name} is not a registered profile" unless @profiles.key?(profile_name)
+
+        @profiles[profile_name].apply(options)
+      end
+    end
+
+    class Profile
+      def apply(options); end
+    end
+
+    class DevelopmentProfile < Profile
+      def apply(options)
+        options.key_value_timeout = 20_000
+        # TODO: Add `options.key_value_durable_timeout = 20_000` when key_value_durable_timeout is added to Options::Cluster
+        options.connect_timeout = 20_000
+        options.view_timeout = 120_000
+        options.query_timeout = 120_000
+        options.analytics_timeout = 120_000
+        options.search_timeout = 120_000
+        options.management_timeout = 120_000
+      end
+    end
+
+    KNOWN_PROFILES = Profiles.new
+  end
+end

data/lib/couchbase/configuration.rb

@@ -0,0 +1,56 @@
+#  Copyright 2020-2021 Couchbase, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require "yaml"
+require "erb"
+
+module Couchbase
+  class Configuration
+    attr_accessor :connection_string
+    attr_accessor :username
+    attr_accessor :password
+
+    def initialize
+      @connection_string = "couchbase://localhost"
+    end
+
+    def load!(path, environment = default_environment_name)
+      settings = load_yaml(path, environment)
+      load_configuration(settings)
+    end
+
+    private
+
+    def load_configuration(settings)
+      @connection_string = settings[:connection_string] || settings["connection_string"]
+      @username = settings[:username] || settings["username"]
+      @password = settings[:password] || settings["password"]
+    end
+
+    def load_yaml(path, environment)
+      file_content = ERB.new(File.read(path)).result
+      YAML.safe_load(file_content, aliases: :default)[environment]
+    end
+
+    def default_environment_name
+      if defined?(::Rails)
+        ::Rails.env
+      elsif defined?(::Sinatra)
+        ::Sinatra::Base.environment.to_s
+      else
+        ENV.fetch("RACK_ENV", nil) || ENV.fetch("COUCHBASE_ENV", nil) or raise ::Couchbase::Error::NoEnvironment
+      end
+    end
+  end
+end

data/lib/couchbase/datastructures/couchbase_list.rb

@@ -0,0 +1,160 @@
+#  Copyright 2020-2021 Couchbase, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require "couchbase/collection"
+require "couchbase/errors"
+require "couchbase/options"
+
+module Couchbase
+  module Datastructures
+    # A {CouchbaseList} is implements +Enumerable+ interface and backed by {Collection} document (more specifically
+    # a JSON array).
+    #
+    # Note that as such, a {CouchbaseList} is restricted to the types that JSON array can contain.
+    class CouchbaseList
+      include Enumerable
+
+      # Create a new List, backed by the document identified by +id+ in +collection+.
+      #
+      # @param [String] id the id of the document to back the list.
+      # @param [Collection] collection the Couchbase collection through which to interact with the document.
+      # @param [Options::CouchbaseList] options customization of the datastructure
+      def initialize(id, collection, options = Options::CouchbaseList.new)
+        @id = id
+        @collection = collection
+        @options = options
+        @cas = 0
+      end
+
+      # Calls the given block once for each element in the list, passing that element as a parameter.
+      #
+      # @yieldparam [Object] item
+      #
+      # @return [CouchbaseList, Enumerable]
+      def each
+        if block_given?
+          begin
+            result = @collection.get(@id, @options.get_options)
+            current = result.content
+            @cas = result.cas
+          rescue Error::DocumentNotFound
+            current = []
+            @cas = 0
+          end
+          current.each do |entry|
+            yield entry
+          end
+          self
+        else
+          enum_for(:each)
+        end
+      end
+
+      # @return [Integer] returns the number of elements in the list.
+      def length
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.count(""),
+                                       ], @options.lookup_in_options)
+        result.content(0)
+      rescue Error::DocumentNotFound
+        0
+      end
+
+      alias size length
+
+      # @return [Boolean] returns true if list is empty
+      def empty?
+        size.zero?
+      end
+
+      # Appends the given object(s) on to the end of this error. This expression returns the array itself, so several
+      # appends may be chained together.
+      #
+      # @param [Object...] obj object(s) to append
+      # @return [CouchbaseList]
+      def push(*obj)
+        @collection.mutate_in(@id, [
+                                MutateInSpec.array_append("", obj),
+                              ], @options.mutate_in_options)
+        self
+      end
+
+      alias append push
+
+      # Prepends objects to the front of the list, moving other elements upwards
+      #
+      # @param [Object...] obj object(s) to prepend
+      # @return [CouchbaseList]
+      def unshift(*obj)
+        @collection.mutate_in(@id, [
+                                MutateInSpec.array_prepend("", obj),
+                              ], @options.mutate_in_options)
+        self
+      end
+
+      alias prepend unshift
+
+      # Inserts the given values before the element with the given +index+.
+      #
+      # @param [Integer] index
+      # @param [Object...] obj object(s) to insert
+      # @return [CouchbaseList]
+      def insert(index, *obj)
+        @collection.mutate_in(@id, [
+                                MutateInSpec.array_insert("[#{index.to_i}]", obj),
+                              ])
+        self
+      end
+
+      # Returns the element at +index+. A negative index counts from the end. Returns +nil+ if the index is out of range.
+      #
+      # @param [Integer] index
+      # @return [Object, nil]
+      def at(index)
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.get("[#{index.to_i}]"),
+                                       ], @options.lookup_in_options)
+        result.exists?(0) ? result.content(0) : nil
+      rescue Error::DocumentNotFound
+        nil
+      end
+
+      alias [] at
+
+      # Deletes the element at the specified +index+, returning that element, or nil
+      #
+      # @param [Integer] index
+      # @return [CouchbaseList]
+      def delete_at(index)
+        @collection.mutate_in(@id, [
+                                MutateInSpec.remove("[#{index.to_i}]"),
+                              ])
+        self
+      rescue Error::DocumentNotFound
+        self
+      end
+
+      # Removes all elements from the list
+      def clear
+        @collection.remove(@id, @options.remove_options)
+        nil
+      rescue Error::DocumentNotFound
+        nil
+      end
+    end
+
+    # @api private
+    CouchbaseListOptions = ::Couchbase::Options::CouchbaseList
+  end
+end

data/lib/couchbase/datastructures/couchbase_map.rb

@@ -0,0 +1,194 @@
+#  Copyright 2020-2021 Couchbase, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require "couchbase/collection"
+require "couchbase/options"
+require "couchbase/errors"
+
+module Couchbase
+  module Datastructures
+    # A {CouchbaseMap} is implements +Enumerable+ interface and backed by {Collection} document (more specifically
+    # a JSON array).
+    #
+    # Note that as such, a {CouchbaseMap} is restricted to the types that JSON array can contain.
+    class CouchbaseMap
+      include Enumerable
+
+      # Create a new Map, backed by the document identified by +id+ in +collection+.
+      #
+      # @param [String] id the id of the document to back the map.
+      # @param [Collection] collection the Couchbase collection through which to interact with the document.
+      # @param [Options::CouchbaseMap] options customization of the datastructure
+      def initialize(id, collection, options = Options::CouchbaseMap.new)
+        @id = id
+        @collection = collection
+        @options = options
+        @cas = 0
+      end
+
+      # Calls the given block once for each element in the map, passing that element as a parameter.
+      #
+      # @yieldparam [Object] item
+      #
+      # @return [CouchbaseMap, Enumerable]
+      def each
+        if block_given?
+          begin
+            result = @collection.get(@id, @options.get_options)
+            current = result.content
+            @cas = result.cas
+          rescue Error::DocumentNotFound
+            current = []
+            @cas = 0
+          end
+          current.each do |key, value|
+            yield key, value
+          end
+          self
+        else
+          enum_for(:each)
+        end
+      end
+
+      # @return [Integer] returns the number of elements in the map.
+      def length
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.count(""),
+                                       ], @options.lookup_in_options)
+        result.content(0)
+      rescue Error::DocumentNotFound
+        0
+      end
+
+      alias size length
+
+      # @return [Boolean] returns true if map is empty
+      def empty?
+        size.zero?
+      end
+
+      # Removes all elements from the map
+      def clear
+        @collection.remove(@id, @options.remove_options)
+        nil
+      rescue Error::DocumentNotFound
+        nil
+      end
+
+      # Returns a value from the map for the given key.
+      #
+      # If the key cannot be found, there are several options:
+      #
+      # * with no other arguments, it will raise a KeyError exception
+      # * if +default+ is given, then that will be returned
+      # * if the optional code +block+ is specified, then that will be run and its result returned
+      #
+      # @overload fetch(key)
+      #   Gets the value, associated with the key, or raise KeyError if key could not be found
+      #   @param key [String] key
+      #
+      # @overload fetch(key, default)
+      #   Gets the value, associated with the key, or return +default+ if key could not be found
+      #   @param key [String] key
+      #
+      # @overload fetch(key, &block)
+      #   Gets the value, associated with the key, or invoke specified +block+ and propagate its return value
+      #   if key could not be found
+      #   @param key [String] key
+      #   @yieldreturn [Object] the default value to return in case the key could not be found
+      #
+      # @return [Object]
+      def fetch(key, *rest)
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.get(key),
+                                       ], @options.lookup_in_options)
+        result.content(0)
+      rescue Error::DocumentNotFound, Error::PathNotFound
+        return yield if block_given?
+        return rest.first unless rest.empty?
+
+        raise KeyError, "key not found: #{key}"
+      end
+
+      # Returns a value from the map for the given key.
+      #
+      # If the key cannot be found, +nil+ will be returned.
+      #
+      # @param [String] key
+      #
+      # @return [Object, nil] value, associated with the key or +nil+
+      def [](key)
+        fetch(key, nil)
+      end
+
+      # Associate the value given by +value+ with the key given by +key+.
+      #
+      # @param [String] key
+      # @param [Object] value
+      #
+      # @return [void]
+      def []=(key, value)
+        @collection.mutate_in(@id, [
+                                MutateInSpec.upsert(key, value),
+                              ], @options.mutate_in_options)
+      end
+
+      # Deletes the key-value pair from the map.
+      #
+      # @param [String] key
+      #
+      # @return void
+      def delete(key)
+        @collection.mutate_in(@id, [
+                                MutateInSpec.remove(key),
+                              ])
+      rescue Error::DocumentNotFound, Error::PathNotFound
+        nil
+      end
+
+      # Returns +true+ if the given key is present
+      #
+      # @param [String] key
+      # @return [Boolean]
+      def key?(key)
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.exists(key),
+                                       ], @options.lookup_in_options)
+        result.exists?(0)
+      rescue Error::DocumentNotFound, Error::PathNotFound
+        false
+      end
+
+      alias member? key?
+      alias include? key?
+
+      # Returns a new array populated with the keys from the map.
+      #
+      # @return [Array]
+      def keys
+        map { |key, _value| key }
+      end
+
+      # Returns a new array populated with the values from the map.
+      #
+      # @return [Array]
+      def values
+        map { |_key, value| value }
+      end
+    end
+
+    # @api private
+    CouchbaseMapOptions = ::Couchbase::Options::CouchbaseMap
+  end
+end

data/lib/couchbase/datastructures/couchbase_queue.rb

@@ -0,0 +1,134 @@
+#  Copyright 2020-2021 Couchbase, Inc.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+
+require "couchbase/collection"
+require "couchbase/options"
+require "couchbase/errors"
+
+module Couchbase
+  module Datastructures
+    # A {CouchbaseQueue} is implements FIFO queue with +Enumerable+ interface and backed by {Collection} document
+    # (more specifically a JSON array).
+    #
+    # Note: sets are restricted to containing primitive types only due to server-side comparison restrictions.
+    class CouchbaseQueue
+      include Enumerable
+
+      # Create a new List, backed by the document identified by +id+ in +collection+.
+      #
+      # @param [String] id the id of the document to back the queue.
+      # @param [Collection] collection the collection through which to interact with the document.
+      # @param [Options::CouchbaseList] options customization of the datastructure
+      def initialize(id, collection, options = Options::CouchbaseQueue.new)
+        @id = id
+        @collection = collection
+        @options = options
+        @cas = 0
+      end
+
+      # Calls the given block once for each element in the queue, passing that element as a parameter.
+      #
+      # @yieldparam [Object] item
+      #
+      # @return [CouchbaseQueue, Enumerable]
+      def each
+        if block_given?
+          begin
+            result = @collection.get(@id, @options.get_options)
+            current = result.content
+            @cas = result.cas
+          rescue Error::DocumentNotFound
+            current = []
+            @cas = 0
+          end
+          current.each do |entry|
+            yield entry
+          end
+          self
+        else
+          enum_for(:each)
+        end
+      end
+
+      # @return [Integer] returns the number of elements in the queue.
+      def length
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.count(""),
+                                       ], @options.lookup_in_options)
+        result.content(0)
+      rescue Error::DocumentNotFound
+        0
+      end
+
+      alias size length
+
+      # @return [Boolean] returns true if queue is empty
+      def empty?
+        size.zero?
+      end
+
+      # Removes all elements from the queue
+      def clear
+        @collection.remove(@id, @options.remove_options)
+        nil
+      rescue Error::DocumentNotFound
+        nil
+      end
+
+      # Adds the given value to the queue
+      #
+      # @param [Object] obj
+      # @return [CouchbaseQueue]
+      def push(obj)
+        begin
+          @collection.mutate_in(@id, [
+                                  MutateInSpec.array_prepend("", [obj]),
+                                ], @options.mutate_in_options)
+        rescue Error::PathExists
+          # ignore
+        end
+        self
+      end
+
+      alias enq push
+      alias << push
+
+      # Retrieves object from the queue
+      #
+      # @return [Object, nil] queue entry or nil
+      def pop
+        result = @collection.lookup_in(@id, [
+                                         LookupInSpec.get("[-1]"),
+                                       ], @options.lookup_in_options)
+        obj = result.exists?(0) ? result.content(0) : nil
+        options = Collection::MutateInOptions.new
+        options.cas = result.cas
+        @collection.mutate_in(@id, [
+                                MutateInSpec.remove("[-1]"),
+                              ], options)
+        obj
+      rescue Error::CasMismatch
+        retry
+      rescue Error::DocumentNotFound, Error::PathNotFound
+        nil
+      end
+
+      alias deq pop
+      alias shift pop
+    end
+
+    # @api private
+    CouchbaseQueueOptions = ::Couchbase::Options::CouchbaseQueue
+  end
+end