added support for column count [ closes #26 ]

History.md

@@ -186,6 +186,7 @@
 ==================
 
   * Fixed issue with composite column meta column marshalling [ @devdazed ]
-  * Exposed setColumnValidator to the user [ @devdazed #
+  * Exposed setColumnValidator to the user [ @devdazed #3 ]
+  * Added support for column counts to Thrift API [ @devdazed #26 ]
 
 

lib/column_family.js

@@ -287,6 +287,31 @@ ColumnFamily.prototype.remove = function() {
   this.connection.execute('remove', marshalledKey, path, timestamp * 1000, consistency, args.callback);
 };
 
+/**
+ * Counts the number of columns in a row by it's key
+ * @param {String} key The key to get
+ * @param {Object} options Options for the get, can have start, end, max, consistencyLevel
+ *   <ul>
+ *     <li>
+ *       start: the from part of the column name, for composites pass an array. By default the
+ *       composite queries are inclusive, to make them exclusive pass an array of arrays where the
+ *       inner array is [ value, false ].
+ *     </li>
+ *       start: the end part of the column name, for composites pass an array. By default the
+ *       composite queries are inclusive, to make them exclusive pass an array of arrays where the
+ *       inner array is [ value, false ].
+ *     <li>reversed: {Boolean} to whether the range is reversed or not</li>
+ *     <li>max: the max amount of columns to return</li>
+ *     <li>columns: an {Array} of column names to get</li>
+ *     <li>consistencyLevel: the read consistency level</li>
+ *   </ul>
+ * @param {Function} callback The callback to invoke once the response has been received
+ */
+ColumnFamily.prototype.count = function(key, options, callback){
+  options.count = true;
+  this.get(key, options, callback);
+};
+
 /**
  * Get a row by its key
  * @param {String} key The key to get
@@ -329,10 +354,15 @@ ColumnFamily.prototype.get = function(key, options, callback){
       return;
     }
 
-    callback(null, Row.fromThrift(key, val, self));
+    if(options.count === true){
+      callback(null, val);
+    } else {
+      callback(null, Row.fromThrift(key, val, self));
+    }
   }
 
-  this.connection.execute('get_slice', marshalledKey, columnParent(self), predicate, consistency, onComplete);
+  var command = options.count === true ? 'get_count' : 'get_slice';
+  this.connection.execute(command, marshalledKey, columnParent(self), predicate, consistency, onComplete);
 };
 
 

test/thrift.js

@@ -249,6 +249,17 @@ module.exports = {
     });
   },
 
+  'test standard cf.get count':function(test, assert){
+    cf_standard.count(config.standard_row_key, config.standard_get_options, function(err, val){
+      assert.ifError(err);
+      assert.ok(typeof val === 'number');
+      assert.ok(val === 1);
+
+      test.finish();
+    });
+  },
+
+
   'test standard cf with composite column slice':function(test, assert){
     var values = [
         new Helenus.Column([1, new Date(1)], 'a'),