couchbase 3.9.0-aarch64-linux-musl

data/lib/couchbase/protostellar/generated/transactions/v1/transactions_services_pb.rb

@@ -0,0 +1,66 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: couchbase/transactions/v1/transactions.proto for package 'Couchbase.Protostellar.Generated.Transactions.V1'
+# Original file comments:
+# Transactions Service
+#
+# Provides server-managed distributed ACID transactions against a
+# Couchbase cluster. The transaction lifecycle is:
+#   1. BeginAttempt — start a new transaction attempt.
+#   2. Perform data operations (Get, Insert, Replace, Remove) within
+#      the transaction context.
+#   3. Commit or Rollback the transaction attempt.
+# Each attempt is identified by a (transaction_id, attempt_id) pair.
+
+require 'grpc'
+require 'couchbase/protostellar/generated/transactions/v1/transactions_pb'
+
+module Couchbase
+  module Protostellar
+    module Generated
+      module Transactions
+        module V1
+          module TransactionsService
+            # TransactionsService provides distributed ACID transactions.
+            # All data operations must reference a valid transaction_id and attempt_id
+            # obtained from TransactionBeginAttempt.
+            class Service
+
+              include ::GRPC::GenericService
+
+              self.marshal_class_method = :encode
+              self.unmarshal_class_method = :decode
+              self.service_name = 'couchbase.transactions.v1.TransactionsService'
+
+              # TransactionBeginAttempt starts a new transaction attempt. Returns
+              # a transaction_id and attempt_id to use for all subsequent operations.
+              # If transaction_id is provided, this begins a retry attempt for an
+              # existing transaction.
+              rpc :TransactionBeginAttempt, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionBeginAttemptRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionBeginAttemptResponse
+              # TransactionCommit commits all mutations made within the transaction
+              # attempt, making them visible to other operations.
+              rpc :TransactionCommit, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionCommitRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionCommitResponse
+              # TransactionRollback aborts the transaction attempt and discards
+              # all uncommitted mutations.
+              rpc :TransactionRollback, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionRollbackRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionRollbackResponse
+              # TransactionGet retrieves a document within the transaction context.
+              # The returned CAS must be used for subsequent Replace or Remove
+              # operations on this document within the same transaction.
+              rpc :TransactionGet, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionGetRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionGetResponse
+              # TransactionInsert creates a new document within the transaction.
+              # Fails if the document already exists.
+              rpc :TransactionInsert, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionInsertRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionInsertResponse
+              # TransactionReplace replaces an existing document within the transaction.
+              # Requires the CAS from a prior TransactionGet.
+              rpc :TransactionReplace, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionReplaceRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionReplaceResponse
+              # TransactionRemove removes a document within the transaction.
+              # Requires the CAS from a prior TransactionGet.
+              rpc :TransactionRemove, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionRemoveRequest, ::Couchbase::Protostellar::Generated::Transactions::V1::TransactionRemoveResponse
+            end
+
+            Stub = Service.rpc_stub_class
+          end
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/generated/view/v1/view_pb.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: couchbase/view/v1/view.proto
+
+require 'google/protobuf'
+
+
+descriptor_data = "\n\x1c\x63ouchbase/view/v1/view.proto\x12\x11\x63ouchbase.view.v1\"\xa0\t\n\x10ViewQueryRequest\x12\x13\n\x0b\x62ucket_name\x18\x08 \x01(\t\x12\x1c\n\x14\x64\x65sign_document_name\x18\x01 \x01(\t\x12\x11\n\tview_name\x18\x02 \x01(\t\x12\x11\n\x04skip\x18\x03 \x01(\rH\x00\x88\x01\x01\x12\x12\n\x05limit\x18\x04 \x01(\rH\x01\x88\x01\x01\x12R\n\x10scan_consistency\x18\x05 \x01(\x0e\x32\x33.couchbase.view.v1.ViewQueryRequest.ScanConsistencyH\x02\x88\x01\x01\x12\x13\n\x06reduce\x18\x06 \x01(\x08H\x03\x88\x01\x01\x12\x12\n\x05group\x18\x07 \x01(\x08H\x04\x88\x01\x01\x12\x18\n\x0bgroup_level\x18\t \x01(\rH\x05\x88\x01\x01\x12\x10\n\x03key\x18\n \x01(\x0cH\x06\x88\x01\x01\x12\x0c\n\x04keys\x18\x0b \x03(\x0c\x12\x16\n\tstart_key\x18\x0c \x01(\x0cH\x07\x88\x01\x01\x12\x14\n\x07\x65nd_key\x18\r \x01(\x0cH\x08\x88\x01\x01\x12\x1a\n\rinclusive_end\x18\x0e \x01(\x08H\t\x88\x01\x01\x12\x1d\n\x10start_key_doc_id\x18\x0f \x01(\tH\n\x88\x01\x01\x12\x1b\n\x0e\x65nd_key_doc_id\x18\x10 \x01(\tH\x0b\x88\x01\x01\x12\x44\n\x08on_error\x18\x11 \x01(\x0e\x32-.couchbase.view.v1.ViewQueryRequest.ErrorModeH\x0c\x88\x01\x01\x12\x12\n\x05\x64\x65\x62ug\x18\x12 \x01(\x08H\r\x88\x01\x01\x12\x45\n\tnamespace\x18\x13 \x01(\x0e\x32-.couchbase.view.v1.ViewQueryRequest.NamespaceH\x0e\x88\x01\x01\x12=\n\x05order\x18\x14 \x01(\x0e\x32).couchbase.view.v1.ViewQueryRequest.OrderH\x0f\x88\x01\x01\"y\n\x0fScanConsistency\x12 \n\x1cSCAN_CONSISTENCY_NOT_BOUNDED\x10\x00\x12!\n\x1dSCAN_CONSISTENCY_REQUEST_PLUS\x10\x01\x12!\n\x1dSCAN_CONSISTENCY_UPDATE_AFTER\x10\x02\"9\n\tErrorMode\x12\x17\n\x13\x45RROR_MODE_CONTINUE\x10\x00\x12\x13\n\x0f\x45RROR_MODE_STOP\x10\x01\"@\n\tNamespace\x12\x18\n\x14NAMESPACE_PRODUCTION\x10\x00\x12\x19\n\x15NAMESPACE_DEVELOPMENT\x10\x01\"2\n\x05Order\x12\x13\n\x0fORDER_ASCENDING\x10\x00\x12\x14\n\x10ORDER_DESCENDING\x10\x01\x42\x07\n\x05_skipB\x08\n\x06_limitB\x13\n\x11_scan_consistencyB\t\n\x07_reduceB\x08\n\x06_groupB\x0e\n\x0c_group_levelB\x06\n\x04_keyB\x0c\n\n_start_keyB\n\n\x08_end_keyB\x10\n\x0e_inclusive_endB\x13\n\x11_start_key_doc_idB\x11\n\x0f_end_key_doc_idB\x0b\n\t_on_errorB\x08\n\x06_debugB\x0c\n\n_namespaceB\x08\n\x06_order\"\xfe\x01\n\x11ViewQueryResponse\x12\x36\n\x04rows\x18\x01 \x03(\x0b\x32(.couchbase.view.v1.ViewQueryResponse.Row\x12\x45\n\tmeta_data\x18\x02 \x01(\x0b\x32-.couchbase.view.v1.ViewQueryResponse.MetaDataH\x00\x88\x01\x01\x1a-\n\x03Row\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0b\n\x03key\x18\x02 \x01(\x0c\x12\r\n\x05value\x18\x03 \x01(\x0c\x1a-\n\x08MetaData\x12\x12\n\ntotal_rows\x18\x01 \x01(\x04\x12\r\n\x05\x64\x65\x62ug\x18\x02 \x01(\x0c\x42\x0c\n\n_meta_data2i\n\x0bViewService\x12Z\n\tViewQuery\x12#.couchbase.view.v1.ViewQueryRequest\x1a$.couchbase.view.v1.ViewQueryResponse\"\x00\x30\x01\x42\xe6\x01\n)com.couchbase.client.protostellar.view.v1P\x01Z<github.com/couchbase/goprotostellar/genproto/view_v1;view_v1\xaa\x02\x1e\x43ouchbase.Protostellar.View.V1\xca\x02(Couchbase\\Protostellar\\Generated\\View\\V1\xea\x02,Couchbase::Protostellar::Generated::View::V1b\x06proto3"
+
+pool = ::Google::Protobuf::DescriptorPool.generated_pool
+pool.add_serialized_file(descriptor_data)
+
+module Couchbase
+  module Protostellar
+    module Generated
+      module View
+        module V1
+          ViewQueryRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryRequest").msgclass
+          ViewQueryRequest::ScanConsistency = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryRequest.ScanConsistency").enummodule
+          ViewQueryRequest::ErrorMode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryRequest.ErrorMode").enummodule
+          ViewQueryRequest::Namespace = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryRequest.Namespace").enummodule
+          ViewQueryRequest::Order = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryRequest.Order").enummodule
+          ViewQueryResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryResponse").msgclass
+          ViewQueryResponse::Row = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryResponse.Row").msgclass
+          ViewQueryResponse::MetaData = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("couchbase.view.v1.ViewQueryResponse.MetaData").msgclass
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/generated/view/v1/view_services_pb.rb

@@ -0,0 +1,39 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: couchbase/view/v1/view.proto for package 'Couchbase.Protostellar.Generated.View.V1'
+# Original file comments:
+# View Service
+#
+# Provides MapReduce view query execution against a Couchbase cluster.
+# Views are defined in design documents and produce key-value results
+# via map and optional reduce functions. Results are server-streamed.
+
+require 'grpc'
+require 'couchbase/protostellar/generated/view/v1/view_pb'
+
+module Couchbase
+  module Protostellar
+    module Generated
+      module View
+        module V1
+          module ViewService
+            # ViewService provides MapReduce view query execution.
+            class Service
+
+              include ::GRPC::GenericService
+
+              self.marshal_class_method = :encode
+              self.unmarshal_class_method = :decode
+              self.service_name = 'couchbase.view.v1.ViewService'
+
+              # ViewQuery executes a MapReduce view query and streams back result rows
+              # and metadata. The final streamed response contains MetaData.
+              rpc :ViewQuery, ::Couchbase::Protostellar::Generated::View::V1::ViewQueryRequest, stream(::Couchbase::Protostellar::Generated::View::V1::ViewQueryResponse)
+            end
+
+            Stub = Service.rpc_stub_class
+          end
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/generated.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Couchbase
+  module Protostellar
+    # @api private
+    module Generated
+    end
+  end
+end

data/lib/couchbase/protostellar/management/bucket_manager.rb

@@ -0,0 +1,67 @@
+#  Copyright 2023. 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.
+
+# frozen_string_literal: true
+
+require "couchbase/management/bucket_manager"
+require "couchbase/options"
+
+require_relative "../request_generator/admin/bucket"
+require_relative "../response_converter/admin/bucket"
+
+module Couchbase
+  module Protostellar
+    module Management
+      class BucketManager
+        def initialize(client)
+          @client = client
+          @request_generator = RequestGenerator::Admin::Bucket.new
+        end
+
+        def create_bucket(settings, options = Couchbase::Management::Options::Bucket::CreateBucket.new)
+          req = @request_generator.create_bucket_request(settings, options)
+          @client.send_request(req)
+        end
+
+        def update_bucket(settings, options = Couchbase::Management::Options::Bucket::UpdateBucket.new)
+          req = @request_generator.update_bucket_request(settings, options)
+          @client.send_request(req)
+        end
+
+        def drop_bucket(settings, options = Couchbase::Management::Options::Bucket::DropBucket.new)
+          req = @request_generator.delete_bucket_request(settings, options)
+          @client.send_request(req)
+        end
+
+        def get_all_buckets(options = Couchbase::Management::Options::Bucket::GetAllBuckets.new)
+          req = @request_generator.list_buckets_request(options)
+          resp = @client.send_request(req)
+          ResponseConverter::Admin::Bucket.to_bucket_settings_array(resp)
+        end
+
+        def get_bucket(bucket_name, options = Couchbase::Management::Options::Bucket::GetBucket.new)
+          buckets = get_all_buckets(options)
+          bucket = buckets.find { |b| b.name == bucket_name }
+          raise Couchbase::Error::BucketNotFound, "Could not find bucket '#{bucket_name}'" if bucket.nil?
+
+          bucket
+        end
+
+        def flush_bucket(_bucket_name, _options = Couchbase::Management::Options::Bucket::FlushBucket.new)
+          raise Couchbase::Error::FeatureNotAvailable, "The #{Protostellar::NAME} protocol does not support flush_bucket yet"
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/management/collection_manager.rb

@@ -0,0 +1,94 @@
+#  Copyright 2023. 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.
+
+# frozen_string_literal: true
+
+require "couchbase/management/collection_manager"
+
+require_relative "../request_generator/admin/collection"
+require_relative "../response_converter/admin/collection"
+
+module Couchbase
+  module Protostellar
+    module Management
+      class CollectionManager
+        def initialize(client:, bucket_name:)
+          @client = client
+          @bucket_name = bucket_name
+          @request_generator = RequestGenerator::Admin::Collection.new(bucket_name: @bucket_name)
+        end
+
+        def get_all_scopes(options = Couchbase::Management::Options::Collection::GetAllScopes.new)
+          req = @request_generator.list_collections_request(options)
+          resp = @client.send_request(req)
+          ResponseConverter::Admin::Collection.to_scope_spec_array(resp)
+        end
+
+        def create_scope(scope_name, options = Couchbase::Management::Options::Collection::CreateScope.new)
+          req = @request_generator.create_scope_request(scope_name, options)
+          @client.send_request(req)
+        end
+
+        def drop_scope(scope_name, options = Couchbase::Management::Options::Collection::DropScope.new)
+          req = @request_generator.delete_scope_request(scope_name, options)
+          @client.send_request(req)
+        end
+
+        def create_collection(*args)
+          req =
+            if args[0].is_a?(Couchbase::Management::CollectionSpec)
+              collection = args[0]
+              options = args[1] || Couchbase::Management::Options::Collection::CreateCollection::DEFAULT
+              settings = Couchbase::Management::CreateCollectionSettings.new(max_expiry: collection.max_expiry, history: collection.history)
+
+              warn "Calling create_collection with a CollectionSpec object has been deprecated, supply scope name, " \
+                   "collection name and optionally a CreateCollectionSettings instance"
+
+              @request_generator.create_collection_request(collection.scope_name, collection.name, settings, options)
+            else
+              scope_name = args[0]
+              collection_name = args[1]
+              settings = args[2] || Couchbase::Management::CreateCollectionSettings::DEFAULT
+              options = args[3] || Couchbase::Management::Options::Collection::CreateCollection::DEFAULT
+              @request_generator.create_collection_request(scope_name, collection_name, settings, options)
+            end
+          @client.send_request(req)
+        end
+
+        def drop_collection(*args)
+          req =
+            if args[0].is_a?(Couchbase::Management::CollectionSpec)
+              collection = args[0]
+              options = args[1] || Couchbase::Management::Options::Collection::CreateCollection::DEFAULT
+
+              warn "Calling drop_collection with a CollectionSpec object has been deprecated, supply scope name and collection name"
+
+              @request_generator.delete_collection_request(collection.scope_name, collection.name, options)
+            else
+              scope_name = args[0]
+              collection_name = args[1]
+              options = args[2] || Couchbase::Management::Options::Collection::CreateCollection::DEFAULT
+              @request_generator.delete_collection_request(scope_name, collection_name, options)
+            end
+          @client.send_request(req)
+        end
+
+        def update_collection(_scope_name, _collection_name, _settings = UpdateCollectionSettings::DEFAULT,
+                              _options = Couchbase::Management::Options::Collection::UpdateCollection::DEFAULT)
+          raise Couchbase::Error::FeatureNotAvailable, "The #{Protostellar::NAME} protocol does not support update_collection"
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/management/collection_query_index_manager.rb

@@ -0,0 +1,124 @@
+#  Copyright 2023. 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.
+
+# frozen_string_literal: true
+
+require "couchbase/management/query_index_manager"
+
+require_relative "../request_generator/admin/query"
+require_relative "../response_converter/admin/query"
+
+module Couchbase
+  module Protostellar
+    module Management
+      class CollectionQueryIndexManager
+        def initialize(client:, bucket_name:, scope_name:, collection_name:)
+          @client = client
+          @bucket_name = bucket_name
+          @scope_name = scope_name
+          @collection_name = collection_name
+          @request_generator = RequestGenerator::Admin::Query.new(
+            bucket_name: @bucket_name, scope_name: @scope_name, collection_name: @collection_name,
+          )
+        end
+
+        def get_all_indexes(options = Couchbase::Management::Options::Query::GetAllIndexes.new)
+          validate_options(options)
+
+          req = @request_generator.get_all_indexes_request(options)
+          resp = @client.send_request(req)
+          ResponseConverter::Admin::Query.to_query_index_array(resp)
+        end
+
+        def create_index(index_name, fields, options = Couchbase::Management::Options::Query::CreateIndex.new)
+          validate_options(options)
+
+          req = @request_generator.create_index_request(index_name, fields, options)
+          @client.send_request(req)
+        end
+
+        def create_primary_index(options = Couchbase::Management::Options::Query::CreatePrimaryIndex.new)
+          validate_options(options)
+
+          req = @request_generator.create_primary_index_request(options)
+          @client.send_request(req)
+        end
+
+        def drop_index(index_name, options = Couchbase::Management::Options::Query::DropIndex.new)
+          validate_options(options)
+
+          req = @request_generator.drop_index_request(index_name, options)
+          @client.send_request(req)
+        end
+
+        def drop_primary_index(options = Couchbase::Management::Options::Query::DropPrimaryIndex.new)
+          validate_options(options)
+
+          req = @request_generator.drop_primary_index_request(options)
+          @client.send_request(req)
+        end
+
+        def build_deferred_indexes(options = Couchbase::Management::Options::Query::BuildDeferredIndexes.new)
+          validate_options(options)
+
+          req = @request_generator.build_deferred_indexes_request(options)
+          @client.send_request(req)
+        end
+
+        def watch_indexes(index_names, timeout, options = Couchbase::Management::Options::Query::WatchIndexes.new)
+          validate_options(options)
+
+          # TODO: Will this be implemented in the gateway instead?
+
+          index_names.append("#primary") if options.watch_primary
+
+          interval_millis = 50
+          deadline = Time.now + (Utils::Time.extract_duration(timeout) * 0.001)
+
+          while Time.now <= deadline
+            get_all_opts = Couchbase::Management::Options::Query::GetAllIndexes.new(timeout: ((deadline - Time.now) * 1000).round)
+            indexes = get_all_indexes(get_all_opts).select { |idx| index_names.include? idx.name }
+            indexes_not_found = index_names - indexes.map(&:name)
+            unless indexes_not_found.empty?
+              raise Couchbase::Error::IndexNotFound,
+                    "Failed to find the indexes: #{indexes_not_found.join(', ')}"
+            end
+
+            all_online = indexes.all? { |idx| idx.state == :online }
+            return if all_online
+
+            sleep(interval_millis / 1000)
+            interval_millis += 500
+            interval_millis = 1000 if interval_millis > 1000
+          end
+          raise Couchbase::Error::UnambiguousTimeout, "Failed to find all indexes online within the allotted time"
+        end
+
+        private
+
+        def validate_options(options)
+          unless options.scope_name.nil?
+            raise Couchbase::Error::InvalidArgument,
+                  "Scope name cannot be set in the options when using the Query Index manager at the collection level"
+          end
+
+          return if options.collection_name.nil?
+
+          raise Couchbase::Error::InvalidArgument,
+                "Collection name cannot be set in the options when using the Query Index manager at the collection level"
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/management/query_index_manager.rb

@@ -0,0 +1,112 @@
+#  Copyright 2023. 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.
+
+# frozen_string_literal: true
+
+require "couchbase/management/query_index_manager"
+
+require_relative "../request_generator/admin/query"
+require_relative "../response_converter/admin/query"
+
+module Couchbase
+  module Protostellar
+    module Management
+      class QueryIndexManager
+        def initialize(client:)
+          @client = client
+          @request_generator = RequestGenerator::Admin::Query.new
+        end
+
+        def get_all_indexes(bucket_name, options = Couchbase::Management::Options::Query::GetAllIndexes.new)
+          validate_options(options)
+
+          req = @request_generator.get_all_indexes_request(options, bucket_name)
+          resp = @client.send_request(req)
+          ResponseConverter::Admin::Query.to_query_index_array(resp)
+        end
+
+        def create_index(bucket_name, index_name, fields, options = Couchbase::Management::Options::Query::CreateIndex.new)
+          validate_options(options)
+
+          req = @request_generator.create_index_request(index_name, fields, options, bucket_name)
+          @client.send_request(req)
+        end
+
+        def create_primary_index(bucket_name, options = Couchbase::Management::Options::Query::CreatePrimaryIndex.new)
+          validate_options(options)
+
+          req = @request_generator.create_primary_index_request(options, bucket_name)
+          @client.send_request(req)
+        end
+
+        def drop_index(bucket_name, index_name, options = Couchbase::Management::Options::Query::DropIndex.new)
+          validate_options(options)
+
+          req = @request_generator.drop_index_request(index_name, options, bucket_name)
+          @client.send_request(req)
+        end
+
+        def drop_primary_index(bucket_name, options = Couchbase::Management::Options::Query::DropPrimaryIndex.new)
+          validate_options(options)
+
+          req = @request_generator.drop_primary_index_request(options, bucket_name)
+          @client.send_request(req)
+        end
+
+        def build_deferred_indexes(bucket_name, options = Couchbase::Management::Options::Query::BuildDeferredIndexes.new)
+          validate_options(options)
+
+          req = @request_generator.build_deferred_indexes_request(options, bucket_name)
+          @client.send_request(req)
+        end
+
+        def watch_indexes(bucket_name, index_names, timeout, options = Couchbase::Management::Options::Query::WatchIndexes.new)
+          validate_options(options)
+
+          # TODO: Will this be implemented in the gateway instead?
+
+          index_names.append("#primary") if options.watch_primary
+
+          interval_millis = 50
+          deadline = Time.now + (Utils::Time.extract_duration(timeout) * 0.001)
+          while Time.now <= deadline
+            get_all_opts = Couchbase::Management::Options::Query::GetAllIndexes.new(timeout: ((deadline - Time.now) * 1000).round)
+            indexes = get_all_indexes(bucket_name, get_all_opts).select { |idx| index_names.include? idx.name }
+            indexes_not_found = index_names - indexes.map(&:name)
+            unless indexes_not_found.empty?
+              raise Couchbase::Error::IndexNotFound,
+                    "Failed to find the indexes: #{indexes_not_found.join(', ')}"
+            end
+
+            all_online = indexes.all? { |idx| idx.state == :online }
+            return if all_online
+
+            sleep(interval_millis / 1000)
+            interval_millis += 500
+            interval_millis = 1000 if interval_millis > 1000
+          end
+          raise Couchbase::Error::UnambiguousTimeout, "Failed to find all indexes online within the allotted time"
+        end
+
+        private
+
+        def validate_options(options)
+          return if options.scope_name.nil? && options.collection_name.nil?
+
+          warn "The attributes 'scope_name' and 'collection_name' have been deprecated. Use 'collection.query_indexes' instead"
+        end
+      end
+    end
+  end
+end

data/lib/couchbase/protostellar/management.rb

@@ -0,0 +1,24 @@
+#  Copyright 2023. 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.
+
+# frozen_string_literal: true
+
+require_relative "management/collection_manager"
+
+module Couchbase
+  module Protostellar
+    module Management
+    end
+  end
+end

data/lib/couchbase/protostellar/request.rb

@@ -0,0 +1,79 @@
+#  Copyright 2023. 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.
+
+# frozen_string_literal: true
+
+require "couchbase/utils/time"
+require_relative "retry/strategies/best_effort"
+require_relative "retry/strategies/fail_fast"
+
+module Couchbase
+  module Protostellar
+    # @api private
+    class Request
+      attr_reader :service
+      attr_reader :rpc
+      attr_reader :timeout
+      attr_reader :proto_request
+      attr_reader :idempotent
+      attr_reader :retry_attempts
+      attr_reader :retry_reasons
+      attr_reader :retry_strategy
+      attr_accessor :context
+
+      def initialize(
+        service:,
+        rpc:,
+        proto_request:,
+        timeout:,
+        retry_strategy: Retry::Strategies::BestEffort::DEFAULT,
+        idempotent: false
+      )
+        @service = service
+        @rpc = rpc
+        @timeout = timeout
+        @deadline = nil
+        @proto_request = proto_request
+        @idempotent = idempotent
+        @retry_attempts = 0
+        @retry_reasons = Set.new
+        @retry_strategy = retry_strategy
+        @context = {}
+      end
+
+      def deadline
+        if @deadline.nil?
+          timeout_secs = 0.001 * Utils::Time.extract_duration(@timeout)
+          @deadline = Time.now + timeout_secs
+        end
+        @deadline
+      end
+
+      def add_retry_attempt(reason)
+        @retry_reasons.add(reason)
+        @retry_attempts += 1
+      end
+
+      def error_context
+        @context.merge({retry_reasons: @retry_reasons.to_a, retry_attempts: @retry_attempts})
+      end
+
+      def set_timeout_from_defaults(timeouts)
+        return unless @timeout.nil?
+
+        @timeout = timeouts.timeout_for_request(self)
+      end
+    end
+  end
+end