Skip to content

Commit 30c495e

Browse files
kinyoklionkinyoklion
committed
feat: Add bootstrap support.
1 parent 1fe9665 commit 30c495e

File tree

11 files changed

+395
-10
lines changed

11 files changed

+395
-10
lines changed

‎packages/sdk/browser/__tests__/BrowserClient.test.ts‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ import {
1212

1313
import { BrowserClient } from '../src/BrowserClient';
1414
import { MockHasher } from './MockHasher';
15+
import { goodBootstrapDataWithReasons } from './testBootstrapData';
1516

1617
function mockResponse(value: string, statusCode: number) {
1718
const response: Response = {
@@ -257,4 +258,31 @@ describe('given a mock platform for a BrowserClient', () => {
257258
url: 'http://filtered.com',
258259
});
259260
});
261+
262+
it('can use bootstrap data', async () => {
263+
const client = new BrowserClient(
264+
'client-side-id',
265+
AutoEnvAttributes.Disabled,
266+
{
267+
streaming: false,
268+
logger,
269+
diagnosticOptOut: true,
270+
},
271+
platform,
272+
);
273+
await client.identify(
274+
{ kind: 'user', key: 'bob' },
275+
{
276+
bootstrap: goodBootstrapDataWithReasons,
277+
},
278+
);
279+
280+
expect(client.jsonVariationDetail('json', undefined)).toEqual({
281+
reason: {
282+
kind: 'OFF',
283+
},
284+
value: ['a', 'b', 'c', 'd'],
285+
variationIndex: 1,
286+
});
287+
});
260288
});

‎packages/sdk/browser/__tests__/BrowserDataManager.test.ts‎

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ import BrowserEncoding from '../src/platform/BrowserEncoding';
2424
import BrowserInfo from '../src/platform/BrowserInfo';
2525
import LocalStorage from '../src/platform/LocalStorage';
2626
import { MockHasher } from './MockHasher';
27+
import { goodBootstrapData } from './testBootstrapData';
2728

2829
global.TextEncoder = TextEncoder;
2930

@@ -126,6 +127,7 @@ describe('given a BrowserDataManager with mocked dependencies', () => {
126127
upsert: jest.fn(),
127128
on: jest.fn(),
128129
off: jest.fn(),
130+
setBootstrap: jest.fn(),
129131
} as unknown as jest.Mocked<FlagManager>;
130132

131133
browserConfig = validateOptions({}, logger);
@@ -310,6 +312,36 @@ describe('given a BrowserDataManager with mocked dependencies', () => {
310312
expect(platform.requests.createEventSource).not.toHaveBeenCalled();
311313
});
312314

315+
it('uses data from bootstrap and does not make an initial poll', async () => {
316+
const context = Context.fromLDContext({ kind: 'user', key: 'test-user' });
317+
const identifyOptions: BrowserIdentifyOptions = {
318+
bootstrap: goodBootstrapData,
319+
};
320+
const identifyResolve = jest.fn();
321+
const identifyReject = jest.fn();
322+
323+
flagManager.loadCached.mockResolvedValue(true);
324+
325+
await dataManager.identify(identifyResolve, identifyReject, context, identifyOptions);
326+
327+
expect(logger.debug).toHaveBeenCalledWith(
328+
'[BrowserDataManager] Identify - Initialization completed from bootstrap',
329+
);
330+
331+
expect(flagManager.loadCached).not.toHaveBeenCalledWith(context);
332+
expect(identifyResolve).toHaveBeenCalled();
333+
expect(flagManager.init).not.toHaveBeenCalled();
334+
expect(flagManager.setBootstrap).toHaveBeenCalledWith(expect.anything(), {
335+
cat: { version: 2, flag: { version: 2, variation: 1, value: false } },
336+
json: { version: 3, flag: { version: 3, variation: 1, value: ['a', 'b', 'c', 'd'] } },
337+
killswitch: { version: 5, flag: { version: 5, variation: 0, value: true } },
338+
'my-boolean-flag': { version: 11, flag: { version: 11, variation: 1, value: false } },
339+
'string-flag': { version: 3, flag: { version: 3, variation: 1, value: 'is bob' } },
340+
});
341+
expect(platform.requests.createEventSource).not.toHaveBeenCalled();
342+
expect(platform.requests.fetch).not.toHaveBeenCalled();
343+
});
344+
313345
it('should identify from polling when there are no cached flags', async () => {
314346
const context = Context.fromLDContext({ kind: 'user', key: 'test-user' });
315347
const identifyOptions: BrowserIdentifyOptions = {};

‎packages/sdk/browser/__tests__/bootstrap.test.ts‎

Lines changed: 149 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,149 @@
1+
import { jest } from '@jest/globals';
2+
3+
import { readFlagsFromBootstrap } from '../src/bootstrap';
4+
import { goodBootstrapData, goodBootstrapDataWithReasons } from './testBootstrapData';
5+
6+
it('can read valid bootstrap data', () => {
7+
const logger = {
8+
debug: jest.fn(),
9+
info: jest.fn(),
10+
warn: jest.fn(),
11+
error: jest.fn(),
12+
};
13+
14+
const readData = readFlagsFromBootstrap(logger, goodBootstrapData);
15+
expect(readData).toEqual({
16+
cat: { version: 2, flag: { version: 2, variation: 1, value: false } },
17+
json: { version: 3, flag: { version: 3, variation: 1, value: ['a', 'b', 'c', 'd'] } },
18+
killswitch: { version: 5, flag: { version: 5, variation: 0, value: true } },
19+
'my-boolean-flag': { version: 11, flag: { version: 11, variation: 1, value: false } },
20+
'string-flag': { version: 3, flag: { version: 3, variation: 1, value: 'is bob' } },
21+
});
22+
expect(logger.debug).not.toHaveBeenCalled();
23+
expect(logger.info).not.toHaveBeenCalled();
24+
expect(logger.warn).not.toHaveBeenCalled();
25+
expect(logger.error).not.toHaveBeenCalled();
26+
});
27+
28+
it('can read valid bootstrap data with reasons', () => {
29+
const logger = {
30+
debug: jest.fn(),
31+
info: jest.fn(),
32+
warn: jest.fn(),
33+
error: jest.fn(),
34+
};
35+
36+
const readData = readFlagsFromBootstrap(logger, goodBootstrapDataWithReasons);
37+
expect(readData).toEqual({
38+
cat: {
39+
version: 2,
40+
flag: {
41+
version: 2,
42+
variation: 1,
43+
value: false,
44+
reason: {
45+
kind: 'OFF',
46+
},
47+
},
48+
},
49+
json: {
50+
version: 3,
51+
flag: {
52+
version: 3,
53+
variation: 1,
54+
value: ['a', 'b', 'c', 'd'],
55+
reason: {
56+
kind: 'OFF',
57+
},
58+
},
59+
},
60+
killswitch: {
61+
version: 5,
62+
flag: {
63+
version: 5,
64+
variation: 0,
65+
value: true,
66+
reason: {
67+
kind: 'FALLTHROUGH',
68+
},
69+
},
70+
},
71+
'my-boolean-flag': {
72+
version: 11,
73+
flag: {
74+
version: 11,
75+
variation: 1,
76+
value: false,
77+
reason: {
78+
kind: 'OFF',
79+
},
80+
},
81+
},
82+
'string-flag': {
83+
version: 3,
84+
flag: {
85+
version: 3,
86+
variation: 1,
87+
value: 'is bob',
88+
reason: {
89+
kind: 'OFF',
90+
},
91+
},
92+
},
93+
});
94+
expect(logger.debug).not.toHaveBeenCalled();
95+
expect(logger.info).not.toHaveBeenCalled();
96+
expect(logger.warn).not.toHaveBeenCalled();
97+
expect(logger.error).not.toHaveBeenCalled();
98+
});
99+
100+
it('can read old bootstrap data', () => {
101+
const logger = {
102+
debug: jest.fn(),
103+
info: jest.fn(),
104+
warn: jest.fn(),
105+
error: jest.fn(),
106+
};
107+
108+
const oldData: any = { ...goodBootstrapData };
109+
delete oldData.$flagsState;
110+
111+
const readData = readFlagsFromBootstrap(logger, oldData);
112+
expect(readData).toEqual({
113+
cat: { version: 0, flag: { version: 0, value: false } },
114+
json: { version: 0, flag: { version: 0, value: ['a', 'b', 'c', 'd'] } },
115+
killswitch: { version: 0, flag: { version: 0, value: true } },
116+
'my-boolean-flag': { version: 0, flag: { version: 0, value: false } },
117+
'string-flag': { version: 0, flag: { version: 0, value: 'is bob' } },
118+
});
119+
expect(logger.debug).not.toHaveBeenCalled();
120+
expect(logger.info).not.toHaveBeenCalled();
121+
expect(logger.warn).toHaveBeenCalledWith(
122+
'LaunchDarkly client was initialized with bootstrap data that did not' +
123+
' include flag metadata. Events may not be sent correctly.',
124+
);
125+
expect(logger.warn).toHaveBeenCalledTimes(1);
126+
expect(logger.error).not.toHaveBeenCalled();
127+
});
128+
129+
it('can handle invalid bootstrap data', () => {
130+
const logger = {
131+
debug: jest.fn(),
132+
info: jest.fn(),
133+
warn: jest.fn(),
134+
error: jest.fn(),
135+
};
136+
137+
const invalid: any = { $valid: false, $flagsState: {} };
138+
139+
const readData = readFlagsFromBootstrap(logger, invalid);
140+
expect(readData).toEqual({});
141+
expect(logger.debug).not.toHaveBeenCalled();
142+
expect(logger.info).not.toHaveBeenCalled();
143+
expect(logger.warn).toHaveBeenCalledWith(
144+
'LaunchDarkly bootstrap data is not available because the back end' +
145+
' could not read the flags.',
146+
);
147+
expect(logger.warn).toHaveBeenCalledTimes(1);
148+
expect(logger.error).not.toHaveBeenCalled();
149+
});

‎packages/sdk/browser/__tests__/testBootstrapData.ts‎

Lines changed: 76 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,76 @@
1+
export const goodBootstrapData = {
2+
cat: false,
3+
json: ['a', 'b', 'c', 'd'],
4+
killswitch: true,
5+
'my-boolean-flag': false,
6+
'string-flag': 'is bob',
7+
$flagsState: {
8+
cat: {
9+
variation: 1,
10+
version: 2,
11+
},
12+
json: {
13+
variation: 1,
14+
version: 3,
15+
},
16+
killswitch: {
17+
variation: 0,
18+
version: 5,
19+
},
20+
'my-boolean-flag': {
21+
variation: 1,
22+
version: 11,
23+
},
24+
'string-flag': {
25+
variation: 1,
26+
version: 3,
27+
},
28+
},
29+
$valid: true,
30+
};
31+
32+
export const goodBootstrapDataWithReasons = {
33+
cat: false,
34+
json: ['a', 'b', 'c', 'd'],
35+
killswitch: true,
36+
'my-boolean-flag': false,
37+
'string-flag': 'is bob',
38+
$flagsState: {
39+
cat: {
40+
variation: 1,
41+
version: 2,
42+
reason: {
43+
kind: 'OFF',
44+
},
45+
},
46+
json: {
47+
variation: 1,
48+
version: 3,
49+
reason: {
50+
kind: 'OFF',
51+
},
52+
},
53+
killswitch: {
54+
variation: 0,
55+
version: 5,
56+
reason: {
57+
kind: 'FALLTHROUGH',
58+
},
59+
},
60+
'my-boolean-flag': {
61+
variation: 1,
62+
version: 11,
63+
reason: {
64+
kind: 'OFF',
65+
},
66+
},
67+
'string-flag': {
68+
variation: 1,
69+
version: 3,
70+
reason: {
71+
kind: 'OFF',
72+
},
73+
},
74+
},
75+
$valid: true,
76+
};

‎packages/sdk/browser/src/BrowserDataManager.ts‎

Lines changed: 30 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ import {
1313
Requestor,
1414
} from '@launchdarkly/js-client-sdk-common';
1515

16+
import { readFlagsFromBootstrap } from './bootstrap';
1617
import { BrowserIdentifyOptions } from './BrowserIdentifyOptions';
1718
import { ValidatedOptions } from './options';
1819

@@ -82,14 +83,29 @@ export default class BrowserDataManager extends BaseDataManager {
8283
this.setConnectionParams();
8384
}
8485
this.secureModeHash = browserIdentifyOptions?.hash;
85-
if (await this.flagManager.loadCached(context)) {
86-
this.debugLog('Identify - Flags loaded from cache. Continuing to initialize via a poll.');
87-
}
88-
const plainContextString = JSON.stringify(Context.toLDContext(context));
89-
const requestor = this.getRequestor(plainContextString);
9086

9187
// TODO: Handle wait for network results in a meaningful way. SDK-707
9288

89+
if (browserIdentifyOptions?.bootstrap) {
90+
this.finishIdentifyFromBootstrap(context, browserIdentifyOptions.bootstrap, identifyResolve);
91+
} else {
92+
if (await this.flagManager.loadCached(context)) {
93+
this.debugLog('Identify - Flags loaded from cache. Continuing to initialize via a poll.');
94+
}
95+
const plainContextString = JSON.stringify(Context.toLDContext(context));
96+
const requestor = this.getRequestor(plainContextString);
97+
await this.finishIdentifyFromPoll(requestor, context, identifyResolve, identifyReject);
98+
}
99+
100+
this.updateStreamingState();
101+
}
102+
103+
private async finishIdentifyFromPoll(
104+
requestor: Requestor,
105+
context: Context,
106+
identifyResolve: () => void,
107+
identifyReject: (err: Error) => void,
108+
) {
93109
try {
94110
const payload = await requestor.requestPayload();
95111
const listeners = this.createStreamListeners(context, identifyResolve);
@@ -98,8 +114,16 @@ export default class BrowserDataManager extends BaseDataManager {
98114
} catch (e: any) {
99115
identifyReject(e);
100116
}
117+
}
101118

102-
this.updateStreamingState();
119+
private finishIdentifyFromBootstrap(
120+
context: Context,
121+
bootstrap: unknown,
122+
identifyResolve: () => void,
123+
) {
124+
this.flagManager.setBootstrap(context, readFlagsFromBootstrap(this.logger, bootstrap));
125+
this.debugLog('Identify - Initialization completed from bootstrap');
126+
identifyResolve();
103127
}
104128

105129
setForcedStreaming(streaming?: boolean) {

0 commit comments

Comments
 (0)