Change flag after call

Author: tshemsedinov

JavaScript/6-wrap-once.js

@@ -6,8 +6,9 @@ const once = (fn) => {
   let finished = false;
   return (...args) => {
     if (finished) return;
+    const res = fn(...args);
     finished = true;
-    fn(...args);
+    return res;
   };
 };
 

JavaScript/7-wrap-count.js

@@ -5,8 +5,10 @@
 const limit = (count, fn) => {
   let counter = 0;
   return (...args) => {
-    if (counter++ === count) return;
-    fn(...args);
+    if (counter === count) return;
+    const res = fn(...args);
+    counter++;
+    return res;
   };
 };