feat(link-addCredits): make addCredits public and promisified

noodlefrenzy · Jun 11, 2015 · 4d1899a · 4d1899a
1 parent 4781f1d
commit 4d1899a
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/lib/link.js b/lib/link.js
@@ -140,10 +140,11 @@ Link.prototype._flowReceived = function(flowFrame) {
   }
 };
 
-Link.prototype._addCredits = function(credits, flowOptions) {
+Link.prototype.addCredits = function(credits, flowOptions) {
   if (this.role === constants.linkRole.sender) {
     throw new exceptions.InvalidStateError('Cannot add link credits as a sender');
   }
+
   var opts = flowOptions || {};
   this.linkCredit += credits;
   this.totalCredits += credits;
@@ -160,6 +161,17 @@ Link.prototype._addCredits = function(credits, flowOptions) {
   var flow = new FlowFrame(opts);
   flow.channel = this.session.channel;
   this.session.connection.sendFrame(flow);
+
+  var self = this;
+  return new Promise(function(resolve, reject) {
+    self.once(Link.CreditChange, function() {
+      resolve();
+    });
+
+    self.once(Link.ErrorReceived, function(error) {
+      reject(error);
+    });
+  });
 };
 
 Link.prototype._checkCredit = function() {