Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: update cached setting immediately at the time of updating the db #32742

Merged
merged 30 commits into from
Jul 20, 2024
Merged
Show file tree
Hide file tree
Changes from 23 commits
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
c4841a2
fix: use correct setting for cache when not overwritten
debdutdeb Jul 8, 2024
2f33e7c
fix registry:add and add some unit tests for it
debdutdeb Jul 12, 2024
51619fc
move tests to non-ee folder
debdutdeb Jul 13, 2024
b3a00e1
fix import path
debdutdeb Jul 13, 2024
d587438
some comment
debdutdeb Jul 13, 2024
f6ff46a
...
debdutdeb Jul 13, 2024
53c2f30
fix test names
debdutdeb Jul 13, 2024
593aea2
reset file first
debdutdeb Jul 13, 2024
6c1bb5b
update
debdutdeb Jul 15, 2024
6a345bb
trigger ci
debdutdeb Jul 15, 2024
f7e06cd
Merge branch 'develop' into fix-incorrect-cache-set
casalsgh Jul 15, 2024
5e28127
Merge branch 'develop' into fix-incorrect-cache-set
casalsgh Jul 16, 2024
d2e5f8a
remove my tests
debdutdeb Jul 16, 2024
f4e41dc
fix existing tests
debdutdeb Jul 16, 2024
902a861
new tests
debdutdeb Jul 16, 2024
49eabbb
Merge branch 'develop' into fix-incorrect-cache-set
debdutdeb Jul 16, 2024
0c6f8ca
did this right before
debdutdeb Jul 16, 2024
028a579
Merge branch 'develop' into fix-incorrect-cache-set
debdutdeb Jul 17, 2024
1e06664
...
debdutdeb Jul 17, 2024
5e7e611
Update apps/meteor/app/settings/server/SettingsRegistry.ts
debdutdeb Jul 17, 2024
03f74e0
Update apps/meteor/app/settings/server/SettingsRegistry.ts
debdutdeb Jul 17, 2024
644e689
Update apps/meteor/app/settings/server/SettingsRegistry.ts
debdutdeb Jul 17, 2024
bd04040
Update apps/meteor/app/settings/server/SettingsRegistry.ts
debdutdeb Jul 17, 2024
39ab4c2
Update apps/meteor/tests/unit/app/settings/server/functions/compareSe…
debdutdeb Jul 17, 2024
d465424
revert global copy
debdutdeb Jul 17, 2024
c12b7a2
lint hi
debdutdeb Jul 17, 2024
8e09b4d
Merge branch 'develop' into fix-incorrect-cache-set
debdutdeb Jul 19, 2024
0eb1fa5
remove comment
debdutdeb Jul 19, 2024
88ae469
add the other changeset
debdutdeb Jul 19, 2024
8931608
fix: incorrect cache set review (#32848)
ggazzo Jul 19, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
66 changes: 41 additions & 25 deletions apps/meteor/app/settings/server/SettingsRegistry.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ import type { ISetting, ISettingGroup, Optional, SettingValue } from '@rocket.ch
import { isSettingEnterprise } from '@rocket.chat/core-typings';
import { Emitter } from '@rocket.chat/emitter';
import type { ISettingsModel } from '@rocket.chat/model-typings';
import { type ModifyResult } from 'mongodb';
import { isEqual } from 'underscore';

import { SystemLogger } from '../../../server/lib/logger/system';
Expand Down Expand Up @@ -73,7 +74,7 @@ const compareSettingsIgnoringKeys =
.filter((key) => !keys.includes(key as keyof ISetting))
.every((key) => isEqual(a[key as keyof ISetting], b[key as keyof ISetting]));

const compareSettings = compareSettingsIgnoringKeys([
export const compareSettings = compareSettingsIgnoringKeys([
'value',
'ts',
'createdAt',
Expand Down Expand Up @@ -138,53 +139,64 @@ export class SettingsRegistry {

const settingFromCodeOverwritten = overwriteSetting(settingFromCode);

const settingOverwrittenDefault = overrideSetting(settingFromCode);

const settingStored = this.store.getSetting(_id);

const settingStoredOverwritten = settingStored && overwriteSetting(settingStored);

const isOverwritten = settingFromCode !== settingFromCodeOverwritten || (settingStored && settingStored !== settingStoredOverwritten);

const updatedSettingAfterApplyingOverwrite = isOverwritten ? settingFromCodeOverwritten : settingOverwrittenDefault;

try {
validateSetting(settingFromCode._id, settingFromCode.type, settingFromCode.value);
} catch (e) {
IS_DEVELOPMENT && SystemLogger.error(`Invalid setting code ${_id}: ${(e as Error).message}`);
}

const { _id: _, ...settingProps } = settingFromCodeOverwritten;
const isOverwritten = settingFromCode !== settingFromCodeOverwritten || (settingStored && settingStored !== settingStoredOverwritten);

if (settingStored && !compareSettings(settingStored, settingFromCodeOverwritten)) {
const { value: _value, ...settingOverwrittenProps } = settingFromCodeOverwritten;

const { value: _value, ...settingOverwrittenProps } = settingFromCodeOverwritten; // settingStoredOverwrite not used since we need the updated props, that is in code not db

const overwrittenKeys = Object.keys(settingFromCodeOverwritten);
const removedKeys = Object.keys(settingStored).filter((key) => !['_updatedAt'].includes(key) && !overwrittenKeys.includes(key));

const updatedProps = (() => {
return {
...settingOverwrittenProps,
...(settingStoredOverwritten &&
settingStored.value !== settingStoredOverwritten.value && { value: settingStoredOverwritten.value }),
};
})();
const updatedProps = {
...settingOverwrittenProps,
...(settingStoredOverwritten &&
settingStored.value !== settingStoredOverwritten.value && { value: settingStoredOverwritten.value }),
};

await this.saveUpdatedSetting(_id, updatedProps, removedKeys);
if (!('value' in updatedProps)) {
// cache can only update if value changes
await this.saveUpdatedSetting(_id, updatedProps, removedKeys);
return;
}

this.store.set(updatedSettingAfterApplyingOverwrite);
const { value: settingForCache } = await this.saveUpdatedSetting(_id, updatedProps, removedKeys);

if (!settingForCache) {
// unreachable
throw new Error('No document returned after setting was updated due to metadata change, something is wrong with code');
debdutdeb marked this conversation as resolved.
Show resolved Hide resolved
}

this.store.set(settingForCache);

return;
}

const { _id: _, ...settingProps } = settingFromCodeOverwritten;

if (settingStored && isOverwritten) {
if (settingStored.value !== settingFromCodeOverwritten.value) {
const overwrittenKeys = Object.keys(settingFromCodeOverwritten);
const removedKeys = Object.keys(settingStored).filter((key) => !['_updatedAt'].includes(key) && !overwrittenKeys.includes(key));

await this.saveUpdatedSetting(_id, settingProps, removedKeys);
const { value: settingForCache } = await this.saveUpdatedSetting(_id, settingProps, removedKeys);

this.store.set(updatedSettingAfterApplyingOverwrite);
if (!settingForCache) {
// unreachable
throw new Error('No document returned due to an OVERWRITE, something is wrong with the code');
debdutdeb marked this conversation as resolved.
Show resolved Hide resolved
}

this.store.set(settingForCache);
}
return;
}
Expand All @@ -198,9 +210,13 @@ export class SettingsRegistry {
return;
}

await this.model.insertOne(updatedSettingAfterApplyingOverwrite); // no need to emit unless we remove the oplog
const settingOverwrittenDefault = overrideSetting(settingFromCode);

const setting = isOverwritten ? settingFromCodeOverwritten : settingOverwrittenDefault;

await this.model.insertOne(setting); // no need to emit unless we remove the oplog
debdutdeb marked this conversation as resolved.
Show resolved Hide resolved

this.store.set(updatedSettingAfterApplyingOverwrite);
this.store.set(setting);
}

/*
Expand Down Expand Up @@ -275,16 +291,16 @@ export class SettingsRegistry {
_id: string,
settingProps: Omit<Optional<ISetting, 'value'>, '_id'>,
removedKeys?: string[],
): Promise<void> {
await this.model.updateOne(
): Promise<ModifyResult<ISetting>> {
return this.model.findOneAndUpdate(
{ _id },
{
$set: settingProps,
...(removedKeys?.length && {
$unset: removedKeys.reduce((unset, key) => ({ ...unset, [key]: 1 }), {}),
}),
},
{ upsert: true },
{ upsert: true, returnDocument: 'after' },
);
}
}
5 changes: 5 additions & 0 deletions apps/meteor/app/settings/server/functions/settings.mocks.ts
Original file line number Diff line number Diff line change
Expand Up @@ -75,6 +75,11 @@ class SettingsClass {
this.upsertCalls++;
}

findOneAndUpdate({ _id }: { _id: string }, value: any, options?: any) {
this.updateOne({ _id }, value, options);
return { value: this.findOne({ _id }) };
}

updateValueById(id: string, value: any): void {
this.data.set(id, { ...this.data.get(id), value });

Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
import { expect } from 'chai';

import { compareSettingsMetadata } from '../../../../../../app/settings/server/SettingsRegistry';
import { getSettingDefaults } from '../../../../../../app/settings/server/functions/getSettingDefaults';

const testSetting = getSettingDefaults({
_id: 'my_dummy_setting',
type: 'string',
value: 'dummy',
});

describe('#compareSettingsMetadata', () => {
const ignoredKeys = ['value', 'ts', 'createdAt', 'valueSource', 'packageValue', 'processEnvValue', '_updatedAt'];

ignoredKeys.forEach((key) =>
it(`should return true if ${key} changes`, () => {
const copiedSetting: any = { ...testSetting };

if (['ts', 'createdAt', '_updatedAt'].includes(key)) {
copiedSetting[key] = new Date();
} else {
copiedSetting[key] = 'random';
}

expect(compareSettingsMetadata(testSetting, copiedSetting)).to.be.true;
}),
debdutdeb marked this conversation as resolved.
Show resolved Hide resolved
);
});
Original file line number Diff line number Diff line change
Expand Up @@ -3,20 +3,32 @@ import { beforeEach, describe, it } from 'mocha';

import { CachedSettings } from '../../../../../../app/settings/server/CachedSettings';
import { SettingsRegistry } from '../../../../../../app/settings/server/SettingsRegistry';
import { getSettingDefaults } from '../../../../../../app/settings/server/functions/getSettingDefaults';
import { Settings } from '../../../../../../app/settings/server/functions/settings.mocks';

const testSetting = getSettingDefaults({
_id: 'my_dummy_setting',
type: 'string',
value: 'dummy',
});

const settings = new CachedSettings();
settings.initialized();

Settings.settings = settings;

const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

describe('Settings', () => {
beforeEach(() => {
settings.store.clear();
Settings.insertCalls = 0;
Settings.upsertCalls = 0;
Settings.data.clear();
process.env = {};
});

it('should not insert the same setting twice', async () => {
const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });
debdutdeb marked this conversation as resolved.
Show resolved Hide resolved
await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting', true, {
Expand Down Expand Up @@ -76,11 +88,6 @@ describe('Settings', () => {
});

it('should respect override via environment as int', async () => {
const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

process.env.OVERWRITE_SETTING_my_setting = '1';

await settingsRegistry.addGroup('group', async function () {
Expand Down Expand Up @@ -136,10 +143,6 @@ describe('Settings', () => {
it('should respect override via environment as boolean', async () => {
process.env.OVERWRITE_SETTING_my_setting_bool = 'true';

const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });
await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting_bool', false, {
Expand Down Expand Up @@ -193,10 +196,6 @@ describe('Settings', () => {
it('should respect override via environment as string', async () => {
process.env.OVERWRITE_SETTING_my_setting_str = 'hey';

const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });
await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting_str', '', {
Expand Down Expand Up @@ -249,10 +248,6 @@ describe('Settings', () => {
});

it('should work with a setting type multiSelect with a default value', async () => {
const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });
await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting_multiselect', ['a'], {
Expand All @@ -274,11 +269,6 @@ describe('Settings', () => {
it('should respect override via environment as multiSelect', async () => {
process.env.OVERWRITE_SETTING_my_setting_multiselect = '["a","b"]';

const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting_multiselect', ['a'], {
Expand All @@ -303,11 +293,6 @@ describe('Settings', () => {
it('should ignore override via environment as multiSelect if value is invalid', async () => {
process.env.OVERWRITE_SETTING_my_setting_multiselect = '[INVALID_ARRAY]';

const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting_multiselect', ['a'], {
Expand All @@ -331,10 +316,6 @@ describe('Settings', () => {

it('should respect initial value via environment', async () => {
process.env.my_setting = '1';
const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
Expand Down Expand Up @@ -381,11 +362,6 @@ describe('Settings', () => {
});

it('should respect override via environment when changing settings props', async () => {
const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

await settingsRegistry.addGroup('group', async function () {
await this.section('section', async function () {
await this.add('my_setting', 0, {
Expand Down Expand Up @@ -440,13 +416,49 @@ describe('Settings', () => {
.to.not.have.any.keys('section');
});

it('should ignore setting object from code if only value changes and setting already stored', async () => {
await settingsRegistry.add(testSetting._id, testSetting.value, testSetting);

expect(Settings.insertCalls).to.be.equal(1);
Settings.insertCalls = 0;

const settingFromCodeFaked = { ...testSetting, value: Date.now().toString() };

await settingsRegistry.add(settingFromCodeFaked._id, settingFromCodeFaked.value, settingFromCodeFaked);

expect(Settings.insertCalls).to.be.equal(0);
expect(Settings.upsertCalls).to.be.equal(0);
});

it('should ignore value from environment if setting is already stored', async () => {
await settingsRegistry.add(testSetting._id, testSetting.value, testSetting);

process.env[testSetting._id] = Date.now().toString();

await settingsRegistry.add(testSetting._id, testSetting.value, testSetting);

expect(Settings.findOne({ _id: testSetting._id }).value).to.be.equal(testSetting.value);
});

it('should update setting cache synchronously if overwrite is available in enviornment', async () => {
process.env[`OVERWRITE_SETTING_${testSetting._id}`] = Date.now().toString();

await settingsRegistry.add(testSetting._id, testSetting.value, testSetting);

expect(settings.get(testSetting._id)).to.be.equal(process.env[`OVERWRITE_SETTING_${testSetting._id}`]);
});

it('should update cached value with OVERWRITE_SETTING value even if both with-prefixed and without-prefixed variables exist', async () => {
process.env[`OVERWRITE_SETTING_${testSetting._id}`] = Date.now().toString();
process.env[testSetting._id] = Date.now().toString();

await settingsRegistry.add(testSetting._id, testSetting.value, testSetting);

expect(Settings.findOne({ _id: testSetting._id }).value).to.be.equal(process.env[`OVERWRITE_SETTING_${testSetting._id}`]);
});

it('should call `settings.get` callback on setting added', async () => {
return new Promise(async (resolve) => {
const settings = new CachedSettings();
Settings.settings = settings;
settings.initialized();
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

const spiedCallback1 = spy();
const spiedCallback2 = spy();

Expand Down Expand Up @@ -475,9 +487,6 @@ describe('Settings', () => {
return new Promise(async (resolve) => {
const spiedCallback1 = spy();
const spiedCallback2 = spy();
const settings = new CachedSettings();
Settings.settings = settings;
const settingsRegistry = new SettingsRegistry({ store: settings, model: Settings as any });

settings.watch('setting_callback', spiedCallback1, { debounce: 1 });
settings.watchByRegex(/setting_callback/gi, spiedCallback2, { debounce: 1 });
Expand Down
Loading