sqlite

src/store/numbers/peekNumber.js

@@ -1,7 +1,25 @@
-import { ENSURE_TYPE } from '../../providers/mssql/queries/numbers.js';
-import request from '../request.js';
+import {
+  ENSURE_TYPE as MSSQL_ENSURE_TYPE,
+  GET_FOR_UPDATE as MSSQL_GET_FOR_UPDATE,
+} from '../../providers/mssql/queries/numbers.js';
+import {
+  ENSURE_TYPE as SQLITE_ENSURE_TYPE,
+  GET_FOR_UPDATE as SQLITE_GET_FOR_UPDATE,
+} from '../../providers/sqlite3/queries/numbers.js';
+import makeRunner from '../runner.js';
+
+function formatNumber(prefix, padding, num) {
+  const p = prefix ?? '';
+  const nStr = String(num);
+  const padded = padding > 0 ? nStr.padStart(padding, '0') : nStr;
+  return `${p}${padded}`;
+}
+
+export default function peekNumber(provider) {
+  const isSqlite = provider?.dialect === 'sqlite3';
+  const ensureSql = isSqlite ? SQLITE_ENSURE_TYPE : MSSQL_ENSURE_TYPE;
+  const getSql = isSqlite ? SQLITE_GET_FOR_UPDATE : MSSQL_GET_FOR_UPDATE;
 
-export default function peekNumber(pool) {
   /**
    * Returns the next number for a company and type without consuming it.
    *
@@ -15,17 +33,22 @@ export default function peekNumber(pool) {
    * // => '1'
    */
   return async (companyId, type, tx) => {
-    await request(pool, tx)
-      .input('companyId', companyId)
-      .input('type', type)
-      .query(ENSURE_TYPE);
+    const doPeek = async (runner) => {
+      await runner.query(ensureSql, { companyId, type });
+      const res = await runner.query(getSql, { companyId, type });
+      const row = res.rows[0];
+      if (!row) {
+        throw new Error('Number type not found for company');
+      }
+      return formatNumber(row.prefix, row.padding, row.next);
+    };
 
-    const req = request(pool, tx);
-    req.input('company_id', companyId);
-    req.input('type', type);
-    req.output('value');
-
-    const result = await req.execute('peeknextnumberwithprefixandpadding');
-    return result.output.value;
+    if (tx) {
+      return doPeek(makeRunner(provider, tx));
+    }
+    if (typeof provider?.transaction === 'function') {
+      return provider.transaction(doPeek);
+    }
+    return doPeek(makeRunner(provider));
   };
 }