mirror of
https://github.com/tdurieux/anonymous_github.git
synced 2026-09-12 13:48:58 +02:00
fix: ignore stale removal jobs after repository restoration
This commit is contained in:
+25
-3
@@ -144,18 +144,40 @@ export async function addRemovalJob(
|
||||
/**
|
||||
* Requeue removals whose database status survived but whose BullMQ job did
|
||||
* not, for example after a crash between updating MongoDB and queueing Redis.
|
||||
* Repository removal is idempotent, so replaying a terminal job is safe.
|
||||
* Failed jobs are replayed only while their removal request is still current.
|
||||
*/
|
||||
export async function recoverStuckRemoving() {
|
||||
if (!removeQueue) return;
|
||||
try {
|
||||
// A failed job is durable proof that removal was requested. Retry it even
|
||||
// if the final failure handler already changed the repository to ERROR.
|
||||
// Claim the pending removal before queueing it. A restored repository, or
|
||||
// an error from a later operation, must not revive an old deletion request.
|
||||
const failedJobs = await removeQueue.getJobs(["failed"]);
|
||||
for (const job of failedJobs) {
|
||||
const repoId = job.data?.repoId;
|
||||
if (!repoId) continue;
|
||||
try {
|
||||
const pending = await AnonymizedRepositoryModel.findOneAndUpdate(
|
||||
{
|
||||
repoId,
|
||||
$or: [
|
||||
{ status: RepositoryStatus.REMOVING },
|
||||
...(job.timestamp
|
||||
? [{
|
||||
status: RepositoryStatus.ERROR,
|
||||
$or: [
|
||||
{ anonymizeDate: { $lte: new Date(job.timestamp) } },
|
||||
{ anonymizeDate: { $exists: false } },
|
||||
],
|
||||
}]
|
||||
: []),
|
||||
],
|
||||
},
|
||||
{ $set: { status: RepositoryStatus.REMOVING } }
|
||||
).collation({ locale: "en", strength: 2 }).exec();
|
||||
if (!pending) {
|
||||
await job.remove();
|
||||
continue;
|
||||
}
|
||||
await addRemovalJob(repoId);
|
||||
logger.info("requeued failed removal", { repoId });
|
||||
} catch (e) {
|
||||
|
||||
@@ -24,6 +24,93 @@ const {
|
||||
} = require("../src/server/schedule");
|
||||
const Repository = require("../src/core/Repository").default;
|
||||
const AnonymizedRepositoryModel = require("../src/core/model/anonymizedRepositories/anonymizedRepositories.model").default;
|
||||
const queueModule = require("../src/queue");
|
||||
const routeUtils = require("../src/server/routes/route-utils");
|
||||
|
||||
const UserModel = require("../src/core/model/users/users.model").default;
|
||||
|
||||
describe("removal recovery", function () {
|
||||
let originals;
|
||||
beforeEach(function () {
|
||||
originals = {
|
||||
distinct: UserModel.distinct,
|
||||
queue: queueModule.removeQueue,
|
||||
find: AnonymizedRepositoryModel.find,
|
||||
findOneAndUpdate: AnonymizedRepositoryModel.findOneAndUpdate,
|
||||
getRepo: routeUtils.getRepo,
|
||||
getUser: routeUtils.getUser,
|
||||
handleError: routeUtils.handleError,
|
||||
};
|
||||
UserModel.distinct = () => ({ exec: async () => [] });
|
||||
});
|
||||
afterEach(function () {
|
||||
UserModel.distinct = originals.distinct;
|
||||
queueModule.removeQueue = originals.queue;
|
||||
AnonymizedRepositoryModel.find = originals.find;
|
||||
AnonymizedRepositoryModel.findOneAndUpdate = originals.findOneAndUpdate;
|
||||
routeUtils.getRepo = originals.getRepo;
|
||||
routeUtils.getUser = originals.getUser;
|
||||
routeUtils.handleError = originals.handleError;
|
||||
});
|
||||
|
||||
for (const status of ["ready", "preparing", "removed", "error"]) {
|
||||
it(`discards an old failed removal after restoration to ${status}`, async function () {
|
||||
let discarded = false;
|
||||
let queued = false;
|
||||
const job = {
|
||||
data: { repoId: "repo-1" }, timestamp: 1000,
|
||||
remove: async () => { discarded = true; },
|
||||
};
|
||||
queueModule.removeQueue = {
|
||||
getJobs: async () => [job],
|
||||
getJob: async () => undefined,
|
||||
add: async () => { queued = true; },
|
||||
};
|
||||
AnonymizedRepositoryModel.findOneAndUpdate = (filter) => ({
|
||||
collation() { return this; },
|
||||
exec: async () => {
|
||||
// A restored snapshot was created after the failed deletion request.
|
||||
const eligible = require("sift").default(filter)({
|
||||
repoId: "repo-1", status, anonymizeDate: new Date(2000),
|
||||
});
|
||||
return eligible ? { repoId: "repo-1" } : null;
|
||||
},
|
||||
});
|
||||
AnonymizedRepositoryModel.find = () => ({ lean: async () => [] });
|
||||
await queueModule.recoverStuckRemoving();
|
||||
expect(discarded).to.equal(true);
|
||||
expect(queued).to.equal(false);
|
||||
});
|
||||
}
|
||||
|
||||
for (const status of ["error", "removing"]) {
|
||||
it(`retries a failed removal still in ${status}`, async function () {
|
||||
const repo = {
|
||||
repoId: "repo-1", status, anonymizeDate: new Date(500),
|
||||
statusDate: new Date(3000),
|
||||
};
|
||||
let added;
|
||||
const job = { data: { repoId: repo.repoId }, timestamp: 1000, finishedOn: 2000 };
|
||||
queueModule.removeQueue = {
|
||||
getJobs: async () => [job],
|
||||
getJob: async () => undefined,
|
||||
add: async (...args) => { added = args; },
|
||||
};
|
||||
AnonymizedRepositoryModel.findOneAndUpdate = (filter, update) => ({
|
||||
collation() { return this; },
|
||||
exec: async () => {
|
||||
if (!require("sift").default(filter)(repo)) return null;
|
||||
Object.assign(repo, update.$set);
|
||||
return repo;
|
||||
},
|
||||
});
|
||||
AnonymizedRepositoryModel.find = () => ({ lean: async () => [] });
|
||||
await queueModule.recoverStuckRemoving();
|
||||
expect(repo.status).to.equal("removing");
|
||||
expect(added[1]).to.deep.equal({ repoId: repo.repoId });
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
describe("conference edits", function () {
|
||||
const form = {
|
||||
|
||||
Reference in New Issue
Block a user