Skip to content

Commit c7d6752

Browse files
author
Kirill Lebedenko
committed
fix: prevent header toggle race condition causing DNR rules desync
1 parent b0a89a2 commit c7d6752

5 files changed

Lines changed: 57 additions & 13 deletions

File tree

src/entities/is-paused/model.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,10 @@ export const $isPaused = createStore<boolean>(false)
1919
.on(toggleIsPaused, state => !state)
2020
.on(loadIsPausedFromStorageFx.doneData, (_, isPaused) => Boolean(isPaused));
2121

22+
// Save isPaused only on explicit user action (toggleIsPaused), not on initial load from storage.
23+
// This prevents a save→onChanged→apply cycle when loading from storage.
2224
sample({
25+
clock: toggleIsPaused,
2326
source: $isPaused,
2427
target: saveIsPausedToStorageFx,
2528
});

src/entities/request-profile/model/selected-request-profile.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,13 @@ export const $selectedRequestProfile = createStore<string>('').on(
1414

1515
const loadSelectedProfileFromStorageFx = createEffect(loadSelectedProfileFromStorageApi);
1616

17-
sample({ source: $selectedRequestProfile, target: saveSelectedProfileToBrowserFx });
17+
// Save selected profile only on explicit user action, not on initial load from storage.
18+
// This prevents a save→onChanged→apply cycle when loading from storage.
19+
sample({
20+
clock: selectedRequestProfileIdChanged,
21+
source: $selectedRequestProfile,
22+
target: saveSelectedProfileToBrowserFx,
23+
});
24+
1825
sample({ clock: loadSelectedProfileFromStorage, target: loadSelectedProfileFromStorageFx });
1926
sample({ clock: loadSelectedProfileFromStorageFx.doneData, target: $selectedRequestProfile });

src/shared/utils/headersConfigMeta.ts

Lines changed: 29 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -29,25 +29,47 @@ function normalizeMeta(value: unknown): HeadersConfigMeta {
2929
return { seq: 0, updatedAt: 0 };
3030
}
3131

32-
export async function bumpHeadersConfigMeta(): Promise<HeadersConfigMeta> {
32+
/**
33+
* Queue to serialize storage writes and prevent race conditions.
34+
* Without this queue, rapid toggles can cause:
35+
* 1. Multiple concurrent reads of the same `seq` value
36+
* 2. Multiple writes with the same incremented `seq`
37+
* 3. Background script skipping updates because `isNewerMeta` returns false
38+
*/
39+
let writeQueue: Promise<void> = Promise.resolve();
40+
41+
async function bumpHeadersConfigMeta(): Promise<HeadersConfigMeta> {
3342
const current = await browser.storage.local.get([BrowserStorageKey.HeadersConfigMeta]);
3443
const prev = normalizeMeta(current[BrowserStorageKey.HeadersConfigMeta]);
3544

3645
// Ensure monotonicity even if system clock moves backwards or updates are very close.
3746
const now = Date.now();
3847
const nextUpdatedAt = Math.max(now, prev.updatedAt + 1);
3948

40-
// Best-effort monotonic seq. It can race across concurrent writers; updatedAt remains authoritative.
49+
// Monotonic seq - now guaranteed by queue serialization.
4150
const nextSeq = prev.seq + 1;
4251

4352
return { seq: nextSeq, updatedAt: nextUpdatedAt };
4453
}
4554

4655
export async function setWithBumpedHeadersConfigMeta(patch: Record<string, unknown>) {
47-
const meta = await bumpHeadersConfigMeta();
48-
await browser.storage.local.set({
49-
...patch,
50-
[BrowserStorageKey.HeadersConfigMeta]: meta,
56+
// Chain this write operation onto the queue to ensure serialization.
57+
// This prevents race conditions where concurrent calls read the same `seq`
58+
// and write the same incremented value.
59+
const result = writeQueue.then(async () => {
60+
const meta = await bumpHeadersConfigMeta();
61+
await browser.storage.local.set({
62+
...patch,
63+
[BrowserStorageKey.HeadersConfigMeta]: meta,
64+
});
65+
return meta;
5166
});
52-
return meta;
67+
68+
// Update queue to wait for this operation (ignore errors for queue chaining)
69+
writeQueue = result.then(
70+
() => {},
71+
() => {},
72+
);
73+
74+
return result;
5375
}

tests/e2e/fixtures.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,11 @@ export const test = base.extend<{
2222
},
2323
extensionId: async ({ context }, use) => {
2424
// Даем время расширению на инициализацию
25-
await new Promise(resolve => setTimeout(resolve, 3000));
25+
await new Promise(resolve => setTimeout(resolve, 5000));
2626

2727
let background;
2828
let attempts = 0;
29-
const maxAttempts = 10;
29+
const maxAttempts = 15;
3030

3131
// Пытаемся получить service worker с повторными попытками
3232
while (attempts < maxAttempts) {
@@ -37,13 +37,13 @@ export const test = base.extend<{
3737
}
3838

3939
try {
40-
background = await context.waitForEvent('serviceworker', { timeout: 3000 });
40+
background = await context.waitForEvent('serviceworker', { timeout: 5000 });
4141
break;
4242
} catch {
4343
attempts++;
4444
if (attempts < maxAttempts) {
4545
// Ждем немного перед следующей попыткой
46-
await new Promise(resolve => setTimeout(resolve, 500));
46+
await new Promise(resolve => setTimeout(resolve, 1000));
4747
}
4848
}
4949
}

tests/e2e/header-toggle-dnr.spec.ts

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ test.describe('Header Toggle DNR Rules', () => {
3636
async function waitForRulesCount(
3737
context: ReturnType<typeof test.extend>['context'],
3838
expectedCount: number,
39-
timeout = 5000,
39+
timeout = 10000,
4040
) {
4141
const startTime = Date.now();
4242
while (Date.now() - startTime < timeout) {
@@ -88,6 +88,9 @@ test.describe('Header Toggle DNR Rules', () => {
8888
await headerNameField.fill('X-Toggle-Test');
8989
await headerValueField.fill('toggle-value');
9090

91+
// Даём время на сохранение и применение DNR правил
92+
await page.waitForTimeout(1000);
93+
9194
// Шаг 3: Проверяем, что DNR правило создано
9295
await waitForRulesCount(context, initialRulesCount + 1);
9396
const rulesAfterAdd = await getDynamicRules(context);
@@ -144,6 +147,9 @@ test.describe('Header Toggle DNR Rules', () => {
144147
await headerNameField.fill('X-Multi-Toggle');
145148
await headerValueField.fill('multi-toggle-value');
146149

150+
// Даём время на сохранение и применение DNR правил
151+
await page.waitForTimeout(1000);
152+
147153
// Ждём создания правила
148154
await waitForRulesCount(context, initialRulesCount + 1);
149155

@@ -198,6 +204,9 @@ test.describe('Header Toggle DNR Rules', () => {
198204
await headerNameField.fill('X-Rapid-Toggle');
199205
await headerValueField.fill('rapid-value');
200206

207+
// Даём время на сохранение и применение DNR правил
208+
await page.waitForTimeout(1000);
209+
201210
// Ждём создания правила
202211
await waitForRulesCount(context, initialRulesCount + 1);
203212

@@ -259,6 +268,9 @@ test.describe('Header Toggle DNR Rules', () => {
259268
await headerNameField.fill('X-Value-Change');
260269
await headerValueField.fill('initial-value');
261270

271+
// Даём время на сохранение и применение DNR правил
272+
await page.waitForTimeout(1000);
273+
262274
// Ждём создания правила
263275
await waitForRulesCount(context, initialRulesCount + 1);
264276

0 commit comments

Comments
 (0)