Fix atomicity for real

Author: mxstbr

src/index.js

@@ -30,7 +30,7 @@ module.exports = async function (req, res) {
     const currentViews = db.has(pathname) ? db.get(pathname).views.length : 0
     // Add a view and send the total views back to the client
     if (shouldIncrement) {
-      pushView(pathname, { time: Date.now() })
+      await pushView(pathname, { time: Date.now() })
     }
     if (req.method === 'GET') {
       send(res, 200, { views: shouldIncrement ? currentViews + 1 : currentViews })

src/utils.js

@@ -4,10 +4,10 @@ const db = require('./db')
 // db messing up.
 const pushView = async (key, view) => {
   const locks = {}
-  push()
+  await push()
 
-  function push() {
-    if (locks[key]) return setImmediate(push)
+  async function push() {
+    if (locks[key]) return setImmediate(async () => { await push() })
     locks[key] = true
 
     let views
@@ -18,11 +18,8 @@ const pushView = async (key, view) => {
     }
 
     try {
-      db.put(key, { views: views.concat([view]) })
-        .then(() => {
-          delete locks[key]
-        })
-        .catch(err => { throw err })
+      await db.put(key, { views: views.concat([view]) })
+      delete locks[key]
     } catch (err) {
       delete locks[key]
       throw err