feat: sqlite bun compat patch, fix race condition

db/sqlite.js

@@ -31,34 +31,29 @@ function getAll(name) {
   })
 }
 
-function setNum(name, num) {
-  return new Promise((resolve, reject) => {
-    db.exec(`INSERT INTO tb_count(\`name\`, \`num\`)
-            VALUES($name, $num)
-            ON CONFLICT(name) DO
-            UPDATE SET \`num\` = $num;`
-      ,
-      { $name: name, $num: num }
-    )
-
-    resolve()
-  })
+async function setNum(name, num) {
+  const stmt = db.prepare(
+    `INSERT INTO tb_count(name, num)
+          VALUES(?, ?)
+          ON CONFLICT(name) DO
+          UPDATE SET num = excluded.num;`
+  );
+  stmt.run(name, num);
 }
 
-function setNumMulti(counters) {
-  return new Promise((resolve, reject) => {
-    const stmt = db.prepare(`INSERT INTO tb_count(\`name\`, \`num\`)
-    VALUES($name, $num)
-    ON CONFLICT(name) DO
-    UPDATE SET \`num\` = $num;`)
-
-    const setMany = db.transaction((counters) => {
-      for (const counter of counters) stmt.run(counter)
-    })
+async function setNumMulti(counters) {
+  const stmt = db.prepare(
+    `INSERT INTO tb_count(name, num)
+          VALUES(?, ?)
+          ON CONFLICT(name) DO
+          UPDATE SET num = excluded.num;`
+  );
 
-    setMany(counters)
-    resolve()
-  })
+  db.transaction((counters) => {
+    counters.forEach((counter) => {
+      stmt.run(counter.name, counter.num);
+    });
+  })(counters);
 }
 
 module.exports = {

index.js

@@ -145,16 +145,13 @@ async function getCountByName(name, num) {
   if (num > 0) { return { name, num } };
 
   try {
-    if (!(name in __cache_counter)) {
-      const counter = (await db.getNum(name)) || defaultCount;
-      __cache_counter[name] = counter.num + 1;
-    } else {
-      __cache_counter[name]++;
-    }
+    // Try to first get counter for this name from cache, then from db, then use default value
+    const count = (__cache_counter[name] || (db.getNum(name) || defaultCount).num) + 1
+    __cache_counter[name] = count;
 
     pushDB();
 
-    return { name, num: __cache_counter[name] };
+    return { name, num: count };
   } catch (error) {
     logger.error("get count by name is error: ", error);
     return defaultCount;