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

data/lib/couchbase/management/scope_search_index_manager.rb

@@ -0,0 +1,198 @@
+#  Copyright 2024. 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 Management
+    class ScopeSearchIndexManager
+      alias inspect to_s
+
+      # @param [Couchbase::Backend] backend
+      # @param [String] bucket_name
+      # @param [String] scope_name
+      def initialize(backend, bucket_name, scope_name)
+        @backend = backend
+        @bucket_name = bucket_name
+        @scope_name = scope_name
+      end
+
+      # Fetches an index from the server if it exists
+      #
+      # @param [String] index_name name of the index
+      # @param [GetIndexOptions] options
+      #
+      # @return [SearchIndex]
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def get_index(index_name, options = GetIndexOptions.new)
+        res = @backend.search_index_get(@bucket_name, @scope_name, index_name, options.timeout)
+        SearchIndexManager.extract_search_index(res)
+      end
+
+      # Fetches all indexes from the server
+      #
+      # @param [GetAllIndexesOptions] options
+      #
+      # @return [Array<SearchIndex>]
+      def get_all_indexes(options = GetAllIndexesOptions.new)
+        res = @backend.search_index_get_all(@bucket_name, @scope_name, options.timeout)
+        res[:indexes].map { |idx| SearchIndexManager.extract_search_index(idx) }
+      end
+
+      # Creates or updates the index
+      #
+      # @param [SearchIndex] index_definition the index definition
+      # @param [UpsertIndexOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError] if name, type or source_type is empty
+      def upsert_index(index_definition, options = UpsertIndexOptions.new)
+        @backend.search_index_upsert(@bucket_name, @scope_name, index_definition.to_backend, options.timeout)
+      end
+
+      # Drops the index
+      #
+      # @param [String] index_name name of the index
+      # @param [DropIndexOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def drop_index(index_name, options = DropIndexOptions.new)
+        @backend.search_index_drop(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Retrieves the number of documents that have been indexed for an index
+      #
+      # @param [String] index_name name of the index
+      # @param [GetIndexedDocumentsCountOptions] options
+      #
+      # @return [Integer]
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def get_indexed_documents_count(index_name, options = GetIndexedDocumentsCountOptions.new)
+        res = @backend.search_index_get_documents_count(@bucket_name, @scope_name, index_name, options.timeout)
+        res[:count]
+      end
+
+      # Pauses updates and maintenance for the index
+      #
+      # @param [String] index_name name of the index
+      # @param [PauseIngestOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def pause_ingest(index_name, options = PauseIngestOptions.new)
+        @backend.search_index_pause_ingest(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Resumes updates and maintenance for an index
+      #
+      # @param [String] index_name name of the index
+      # @param [ResumeIngestOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def resume_ingest(index_name, options = ResumeIngestOptions.new)
+        @backend.search_index_resume_ingest(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Allows querying against the index
+      #
+      # @param [String] index_name name of the index
+      # @param [AllowQueryingOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def allow_querying(index_name, options = AllowQueryingOptions.new)
+        @backend.search_index_allow_querying(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Disallows querying against the index
+      #
+      # @param [String] index_name name of the index
+      # @param [DisallowQueryingOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def disallow_querying(index_name, options = DisallowQueryingOptions.new)
+        @backend.search_index_disallow_querying(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Freeze the assignment of index partitions to nodes
+      #
+      # @param [String] index_name name of the index
+      # @param [FreezePlanOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def freeze_plan(index_name, options = FreezePlanOptions.new)
+        @backend.search_index_freeze_plan(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Unfreeze the assignment of index partitions to nodes
+      #
+      # @param [String] index_name name of the index
+      # @param [UnfreezePlanOptions] options
+      #
+      # @return void
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def unfreeze_plan(index_name, options = UnfreezePlanOptions.new)
+        @backend.search_index_unfreeze_plan(@bucket_name, @scope_name, index_name, options.timeout)
+      end
+
+      # Allows to see how a document is analyzed against a specific index
+      #
+      # @param [String] index_name name of the index
+      # @param [Hash] document the document to be analyzed
+      #
+      # @return [Array<Hash>]
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def analyze_document(index_name, document, options = AnalyzeDocumentOptions.new)
+        res = @backend.search_index_analyze_document(@bucket_name, @scope_name, index_name, JSON.generate(document), options.timeout)
+        JSON.parse(res[:analysis])
+      end
+
+      GetIndexOptions = SearchIndexManager::GetIndexOptions
+      GetAllIndexesOptions = SearchIndexManager::GetAllIndexesOptions
+      UpsertIndexOptions = SearchIndexManager::UpsertIndexOptions
+      DropIndexOptions = SearchIndexManager::DropIndexOptions
+      GetIndexedDocumentsCountOptions = SearchIndexManager::GetIndexedDocumentsCountOptions
+      PauseIngestOptions = SearchIndexManager::PauseIngestOptions
+      ResumeIngestOptions = SearchIndexManager::ResumeIngestOptions
+      AllowQueryingOptions = SearchIndexManager::AllowQueryingOptions
+      DisallowQueryingOptions = SearchIndexManager::DisallowQueryingOptions
+      FreezePlanOptions = SearchIndexManager::FreezePlanOptions
+      UnfreezePlanOptions = SearchIndexManager::UnfreezePlanOptions
+      AnalyzeDocumentOptions = SearchIndexManager::AnalyzeDocumentOptions
+    end
+  end
+end

data/lib/couchbase/management/search_index_manager.rb

@@ -1,4 +1,4 @@
-#    Copyright 2020 Couchbase, Inc.
+#  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.
@@ -17,7 +17,7 @@ require "couchbase/errors"
 module Couchbase
   module Management
     class SearchIndexManager
-      alias_method :inspect, :to_s
+      alias inspect to_s
 
       # @param [Couchbase::Backend] backend
       def initialize(backend)
@@ -34,8 +34,8 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def get_index(index_name, options = GetIndexOptions.new)
-        res = @backend.search_index_get(index_name, options.timeout)
-        extract_search_index(res)
+        res = @backend.search_index_get(nil, nil, index_name, options.timeout)
+        self.class.extract_search_index(res)
       end
 
       # Fetches all indexes from the server
@@ -44,8 +44,8 @@ module Couchbase
       #
       # @return [Array<SearchIndex>]
       def get_all_indexes(options = GetAllIndexesOptions.new)
-        res = @backend.search_index_get_all(options.timeout)
-        res[:indexes].map { |idx| extract_search_index(idx) }
+        res = @backend.search_index_get_all(nil, nil, options.timeout)
+        res[:indexes].map { |idx| self.class.extract_search_index(idx) }
       end
 
       # Creates or updates the index
@@ -58,17 +58,20 @@ module Couchbase
       # @raise [ArgumentError] if name, type or source_type is empty
       def upsert_index(index_definition, options = UpsertIndexOptions.new)
         @backend.search_index_upsert(
-            {
-                name: index_definition.name,
-                type: index_definition.type,
-                uuid: index_definition.uuid,
-                params: (JSON.generate(index_definition.params) if index_definition.params),
-                source_name: index_definition.source_name,
-                source_type: index_definition.source_type,
-                source_uuid: index_definition.source_uuid,
-                source_params: (JSON.generate(index_definition.source_params) if index_definition.source_params),
-                plan_params: (JSON.generate(index_definition.plan_params) if index_definition.plan_params),
-            }, options.timeout)
+          nil,
+          nil,
+          {
+            name: index_definition.name,
+            type: index_definition.type,
+            uuid: index_definition.uuid,
+            params: (JSON.generate(index_definition.params) if index_definition.params),
+            source_name: index_definition.source_name,
+            source_type: index_definition.source_type,
+            source_uuid: index_definition.source_uuid,
+            source_params: (JSON.generate(index_definition.source_params) if index_definition.source_params),
+            plan_params: (JSON.generate(index_definition.plan_params) if index_definition.plan_params),
+          }, options.timeout
+        )
       end
 
       # Drops the index
@@ -81,7 +84,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def drop_index(index_name, options = DropIndexOptions.new)
-        @backend.search_index_drop(index_name, options.timeout)
+        @backend.search_index_drop(nil, nil, index_name, options.timeout)
       end
 
       # Retrieves the number of documents that have been indexed for an index
@@ -94,10 +97,41 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def get_indexed_documents_count(index_name, options = GetIndexedDocumentsCountOptions.new)
-        res = @backend.search_index_get_documents_count(index_name, options.timeout)
+        res = @backend.search_index_get_documents_count(nil, nil, index_name, options.timeout)
         res[:count]
       end
 
+      # Retrieves metrics, timings and counters for a given index
+      #
+      # @!macro uncommitted
+      #
+      # @param [String] index_name name of the index
+      # @param [GetIndexStatsOptions] options
+      #
+      # @return [Integer]
+      #
+      # @raise [ArgumentError]
+      # @raise [Error::IndexNotFound]
+      def get_index_stats(index_name, options = GetIndexStatsOptions.new)
+        res = @backend.search_index_get_stats(index_name, options.timeout)
+        JSON.parse(res)
+      end
+
+      # Retrieves statistics on search service. Information is provided on documents, partition indexes, mutations,
+      # compactions, queries, and more.
+      #
+      # @!macro uncommitted
+      #
+      # @param [GetIndexStatsOptions] options
+      #
+      # @return [Integer]
+      #
+      # @raise [ArgumentError]
+      def get_stats(options = GetIndexStatsOptions.new)
+        res = @backend.search_get_stats(options.timeout)
+        JSON.parse(res)
+      end
+
       # Pauses updates and maintenance for the index
       #
       # @param [String] index_name name of the index
@@ -108,7 +142,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def pause_ingest(index_name, options = PauseIngestOptions.new)
-        @backend.search_index_pause_ingest(index_name, options.timeout)
+        @backend.search_index_pause_ingest(nil, nil, index_name, options.timeout)
       end
 
       # Resumes updates and maintenance for an index
@@ -121,7 +155,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def resume_ingest(index_name, options = ResumeIngestOptions.new)
-        @backend.search_index_resume_ingest(index_name, options.timeout)
+        @backend.search_index_resume_ingest(nil, nil, index_name, options.timeout)
       end
 
       # Allows querying against the index
@@ -134,7 +168,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def allow_querying(index_name, options = AllowQueryingOptions.new)
-        @backend.search_index_allow_querying(index_name, options.timeout)
+        @backend.search_index_allow_querying(nil, nil, index_name, options.timeout)
       end
 
       # Disallows querying against the index
@@ -147,7 +181,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def disallow_querying(index_name, options = DisallowQueryingOptions.new)
-        @backend.search_index_disallow_querying(index_name, options.timeout)
+        @backend.search_index_disallow_querying(nil, nil, index_name, options.timeout)
       end
 
       # Freeze the assignment of index partitions to nodes
@@ -160,7 +194,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def freeze_plan(index_name, options = FreezePlanOptions.new)
-        @backend.search_index_freeze_plan(index_name, options.timeout)
+        @backend.search_index_freeze_plan(nil, nil, index_name, options.timeout)
       end
 
       # Unfreeze the assignment of index partitions to nodes
@@ -173,7 +207,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def unfreeze_plan(index_name, options = UnfreezePlanOptions.new)
-        @backend.search_index_unfreeze_plan(index_name, options.timeout)
+        @backend.search_index_unfreeze_plan(nil, nil, index_name, options.timeout)
       end
 
       # Allows to see how a document is analyzed against a specific index
@@ -186,7 +220,7 @@ module Couchbase
       # @raise [ArgumentError]
       # @raise [Error::IndexNotFound]
       def analyze_document(index_name, document, options = AnalyzeDocumentOptions.new)
-        res = @backend.search_index_analyze_document(index_name, JSON.generate(document), options.timeout)
+        res = @backend.search_index_analyze_document(nil, nil, index_name, JSON.generate(document), options.timeout)
         JSON.parse(res[:analysis])
       end
 
@@ -234,7 +268,17 @@ module Couchbase
         # @return [Integer] the time in milliseconds allowed for the operation to complete
         attr_accessor :timeout
 
-        # @yieldparam [GetIndexedDocumentOptions] self
+        # @yieldparam [GetIndexedDocumentCountOptions] self
+        def initialize
+          yield self if block_given?
+        end
+      end
+
+      class GetIndexStatsOptions
+        # @return [Integer] the time in milliseconds allowed for the operation to complete
+        attr_accessor :timeout
+
+        # @yieldparam [GetStatsOptions] self
         def initialize
           yield self if block_given?
         end
@@ -310,9 +354,8 @@ module Couchbase
         end
       end
 
-      private
-
-      def extract_search_index(resp)
+      # @api private
+      def self.extract_search_index(resp)
         SearchIndex.new do |index|
           index.name = resp[:name]
           index.type = resp[:type]
@@ -357,8 +400,25 @@ module Couchbase
 
       # @yieldparam [SearchIndex] self
       def initialize
+        @type = "fulltext-index"
+        @source_type = "couchbase"
         yield self if block_given?
       end
+
+      # @api private
+      def to_backend
+        {
+          name: name,
+          type: type,
+          uuid: uuid,
+          params: (JSON.generate(params) if params),
+          source_name: source_name,
+          source_type: source_type,
+          source_uuid: source_uuid,
+          source_params: (JSON.generate(source_params) if source_params),
+          plan_params: (JSON.generate(plan_params) if plan_params),
+        }
+      end
     end
   end
 end