feat: listener add async function support

index.js

@@ -39,12 +39,15 @@ class Base extends EventEmitter {
   }
 
   _wrapListener(eventName, listener) {
-    if (is.generatorFunction(listener)) {
-      assert(eventName !== 'error', '[sdk-base] `error` event should not have a generator listener.');
+    if (is.generatorFunction(listener) || is.asyncFunction(listener)) {
+      assert(eventName !== 'error', '[sdk-base] `error` event should not have a generator/async listener.');
 
       const newListener = (...args) => {
         co(function* () {
-          yield listener(...args);
+          const ret = yield listener(...args);
+          if (is.promise(ret)) {
+            yield ret;
+          }
         }).catch(err => {
           err.name = 'EventListenerProcessError';
           this.emit('error', err);

package.json

@@ -1,6 +1,6 @@
 {
   "name": "sdk-base",
-  "version": "3.4.0",
+  "version": "3.4.1",
   "description": "a base class for sdk with default error handler",
   "main": "index.js",
   "scripts": {

test/index.test.js

@@ -242,6 +242,49 @@ describe('sdk-base', () => {
       client.emit('event_code', 1, 2);
     });
 
+    it('should add async listener', done => {
+      done = pedding(done, 8);
+      const client = new SomeServiceClient();
+
+      client.addListener('event_code_async', async function(a, b) {
+        console.log('event_code_async in addListener');
+        assert(a === 1);
+        assert(b === 2);
+        done();
+      });
+
+      client.on('event_code_async', async function(a, b) {
+        console.log('event_code_async in on');
+        assert(a === 1);
+        assert(b === 2);
+        done();
+      });
+
+      client.once('event_code_async', async function(a, b) {
+        console.log('event_code_async in once');
+        assert(a === 1);
+        assert(b === 2);
+        done();
+      });
+
+      client.prependListener('event_code_async', async function(a, b) {
+        console.log('event_code_async in prependListener');
+        assert(a === 1);
+        assert(b === 2);
+        done();
+      });
+
+      client.prependOnceListener('event_code_async', async function(a, b) {
+        console.log('event_code_async in prependOnceListener');
+        assert(a === 1);
+        assert(b === 2);
+        done();
+      });
+
+      client.emit('event_code_async', 1, 2);
+      client.emit('event_code_async', 1, 2);
+    });
+
     it('should catch generator exception and emit on error event', done => {
       done = pedding(done, 2);
       const client = new SomeServiceClient();