f

lib/channel.js

@@ -3,6 +3,7 @@
 const is = require('is-type-of');
 const crypto = require('crypto');
 const Base = require('tcp-base');
+const promisify = require('util').promisify;
 const RemotingCommand = require('./protocol/command/remoting_command');
 
 class Channel extends Base {
@@ -24,6 +25,7 @@ class Channel extends Base {
       headerLength: 4,
       needHeartbeat: false,
     }, options));
+    this.sendPromise = promisify(this.send);
   }
 
   get accessKey() {
@@ -98,9 +100,9 @@ class Channel extends Base {
    * @param {Number} timeout - response timeout
    * @return {Object} response
    */
-  async invoke(command, timeout) {
+  invoke(command, timeout) {
     this.beforeRequest(command);
-    return await this.sendThunk({
+    return this.sendPromise({
       id: command.opaque,
       data: command.encode(),
       timeout,
@@ -110,11 +112,11 @@ class Channel extends Base {
   /**
    * invoke rocketmq api without need response
    * @param {RemotingCommand} command - remoting command
-   * @return {void}
+   * @return {Promise} Promise
    */
-  async invokeOneway(command) {
+  invokeOneway(command) {
     this.beforeRequest(command);
-    await this.sendThunk({
+    return this.sendPromise({
       id: command.opaque,
       data: command.encode(),
       oneway: true,