change wait to return function

wei3hua2 · wei3hua2 · commit 8305547a82ab · 2018-08-03T10:32:44.000+08:00
diff --git a/src/index.ts b/src/index.ts
@@ -149,18 +149,27 @@ export default class RPSBasic {
  * @summary wait :: Number → a → a
  * 
 */
-  // @rpsAction({verbName:'wait'})
-  // wait (ctx:RpsContext,opts:{}, period:number, response?:any) : Promise<any>{
-  //   response = response ? response : ctx.$RESULT;
-
-  //   return new Promise(function (resolve,reject) {
-      
-  //     setTimeout(async function () {
-  //       resolve(response);
-  //     }, period*1000);
-
-  //   });
-  // }
+  @rpsAction({verbName:'wait'})
+  async wait (ctx:RpsContext,opts:Object, period:number, response?:any) : Promise<any>{
+    let isFn = opts['function'];
+    response = response ? response : ctx.$RESULT;
+
+    function sleep(res) {
+      let time = period * 1000;
+      Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, time);
+
+      return res;
+    }
+
+    if(isFn) return sleep;
+    else return sleep(response);
+
+    // return new Promise(function (resolve,reject) {
+      // setTimeout(async function () {
+      //   resolve(response);
+      // }, period*1000);
+    // });
+  }
 
 /**
  * @function stringify
diff --git a/test/index.ts b/test/index.ts
@@ -20,6 +20,15 @@ m.describe('Basic', () => {
     // expect(context.variables['val']).to.be.equals('hello');
     expect(context.variables['$val']).to.be.equals('hello');
 
+    console.log('hello');
+    let fn = await basic.wait(context,{function:true},1);
+    let output = fn('world');
+    console.log(output);
+
+    console.log(await basic.wait(context,{},1,'welcome'));
+    console.log(await basic.wait(context,{},1,undefined));
+    console.log(await basic.wait(context,{},1));
+
   }).timeout(0);
 
 
