RCBC-510: Add BucketSettings#num_vbuckets

Author: DemetrisChr

ext/rcb_buckets.cxx

@@ -232,6 +232,17 @@ cb_generate_bucket_settings(VALUE bucket,
     }
   }
 
+  if (VALUE num_vbuckets = rb_hash_aref(bucket, rb_id2sym(rb_intern("num_vbuckets")));
+      !NIL_P(num_vbuckets)) {
+    if (TYPE(num_vbuckets) == T_FIXNUM) {
+      entry.num_vbuckets = FIX2UINT(num_vbuckets);
+    } else {
+      throw ruby_exception(
+        rb_eArgError,
+        rb_sprintf("num vbuckets must be an Integer, given %+" PRIsVALUE, num_vbuckets));
+    }
+  }
+
   if (is_create) {
     if (VALUE conflict_resolution_type =
           rb_hash_aref(bucket, rb_id2sym(rb_intern("conflict_resolution_type")));
@@ -530,6 +541,9 @@ cb_extract_bucket_settings(const core::management::cluster::bucket_settings& ent
     rb_hash_aset(
       bucket, rb_id2sym(rb_intern("history_retention_duration")), ULONG2NUM(val.value()));
   }
+  if (const auto& val = entry.num_vbuckets; val.has_value()) {
+    rb_hash_aset(bucket, rb_id2sym(rb_intern("num_vbuckets")), USHORT2NUM(val.value()));
+  }
 
   VALUE capabilities = rb_ary_new_capa(static_cast<long>(entry.capabilities.size()));
   for (const auto& capa : entry.capabilities) {

lib/couchbase/management/bucket_manager.rb

@@ -221,6 +221,7 @@ def create_bucket(settings, options = Options::Bucket::CreateBucket.new)
             history_retention_collection_default: settings.history_retention_collection_default,
             history_retention_duration: settings.history_retention_duration,
             history_retention_bytes: settings.history_retention_bytes,
+            num_vbuckets: settings.num_vbuckets,
           }, options.to_backend
         )
       end
@@ -251,6 +252,7 @@ def update_bucket(settings, options = Options::Bucket::UpdateBucket.new)
             history_retention_collection_default: settings.history_retention_collection_default,
             history_retention_bytes: settings.history_retention_bytes,
             history_retention_duration: settings.history_retention_duration,
+            num_vbuckets: settings.num_vbuckets,
           }, options.to_backend
         )
       end
@@ -345,6 +347,7 @@ def extract_bucket_settings(entry)
           bucket.history_retention_collection_default = entry[:history_retention_collection_default]
           bucket.history_retention_bytes = entry[:history_retention_bytes]
           bucket.history_retention_duration = entry[:history_retention_duration]
+          bucket.num_vbuckets = entry[:num_vbuckets]
         end
       end
     end
@@ -405,7 +408,7 @@ class BucketSettings
       # @return [nil, :none, :majority, :majority_and_persist_to_active, :persist_to_majority] the minimum durability level
       attr_accessor :minimum_durability_level
 
-      # @return [Boolean, nil] whether to enable history retention on collections  by default
+      # @return [Boolean, nil] whether to enable history retention on collections by default
       attr_accessor :history_retention_collection_default
 
       # @return [Integer, nil] the maximum size, in bytes, of the change history that is written to disk for all
@@ -416,6 +419,9 @@ class BucketSettings
       # collections in this bucket
       attr_accessor :history_retention_duration
 
+      # @return [Integer, nil] the number of vBuckets the bucket should have. If not set, the server default will be used
+      attr_accessor :num_vbuckets
+
       # @api private
       # @return [Boolean] false if status of the bucket is not healthy
       def healthy?