Add MariaDB queue table. Fix Bugs
This commit is contained in:
Vendored
+39
@@ -0,0 +1,39 @@
|
||||
import dbHandler from "../db-handler";
|
||||
import { quoteIdentifier } from "../../utils/sql-identifier";
|
||||
import { QUEUE_NOW_SQL, requireQueueTable } from "./queue-utils";
|
||||
/**
|
||||
* Extend the lease on a claimed job, for jobs that run longer than the
|
||||
* lease given to `claim`.
|
||||
*/
|
||||
export default async function QueueExtend({ table, id, lock_token, lease_ms, config, }) {
|
||||
try {
|
||||
const options = requireQueueTable(table);
|
||||
const leaseMs = Math.max(0, Math.floor(Number(lease_ms ?? options.lease_ms) || 0));
|
||||
const res = await dbHandler({
|
||||
query: `UPDATE ${quoteIdentifier(table)} SET queue_lease_expires_at = ${QUEUE_NOW_SQL} + ?, updated_at = ${QUEUE_NOW_SQL} WHERE id = ? AND queue_status = 'processing' AND queue_lock_token = ?`,
|
||||
values: [leaseMs, id, lock_token],
|
||||
config,
|
||||
});
|
||||
if (!res.success) {
|
||||
return { ...res, success: false };
|
||||
}
|
||||
if (!res.db_res?.affectedRows) {
|
||||
return {
|
||||
...res,
|
||||
success: false,
|
||||
msg: "Job is not claimed with this lock token",
|
||||
};
|
||||
}
|
||||
return await dbHandler({
|
||||
query: `SELECT * FROM ${quoteIdentifier(table)} WHERE id = ?`,
|
||||
values: [id],
|
||||
config,
|
||||
});
|
||||
}
|
||||
catch (error) {
|
||||
return {
|
||||
success: false,
|
||||
error: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user