Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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
33 changes: 26 additions & 7 deletions src/cache.ts
Original file line number Diff line number Diff line change
Expand Up @@ -34,18 +34,21 @@ export class ListWatch<T extends KubernetesObject> implements ObjectCache<T>, In
private readonly watch: Watch;
private readonly listFn: ListPromise<T>;
private readonly labelSelector?: string;
private readonly fieldSelector?: string;

public constructor(
path: string,
watch: Watch,
listFn: ListPromise<T>,
autoStart: boolean = true,
labelSelector?: string,
fieldSelector?: string,
) {
this.path = path;
this.watch = watch;
this.listFn = listFn;
this.labelSelector = labelSelector;
this.fieldSelector = fieldSelector;

this.callbackCache[ADD] = [];
this.callbackCache[UPDATE] = [];
Expand Down Expand Up @@ -140,7 +143,10 @@ export class ListWatch<T extends KubernetesObject> implements ObjectCache<T>, In

private async doneHandler(err: any): Promise<void> {
this._stop();
if (err && err.statusCode === 410) {
if (
err &&
((err as { statusCode?: number }).statusCode === 410 || (err as { code?: number }).code === 410)
) {
this.resourceVersion = '';
} else if (err) {
this.callbackCache[ERROR].forEach((elt: ErrorCallback) => elt(err));
Expand All @@ -162,17 +168,21 @@ export class ListWatch<T extends KubernetesObject> implements ObjectCache<T>, In
}
this.objects = deleteItems(this.objects, list.items, this.callbackCache[DELETE].slice());
this.addOrUpdateItems(list.items);
this.resourceVersion = list.metadata!.resourceVersion || '';
this.resourceVersion = list.metadata ? list.metadata!.resourceVersion || '' : '';
}
const queryParams = {
resourceVersion: this.resourceVersion,
} as {
resourceVersion: string | undefined;
labelSelector: string | undefined;
fieldSelector: string | undefined;
};
if (this.labelSelector !== undefined) {
queryParams.labelSelector = ObjectSerializer.serialize(this.labelSelector, 'string');
}
if (this.fieldSelector !== undefined) {
queryParams.fieldSelector = ObjectSerializer.serialize(this.fieldSelector, 'string');
}
this.request = await this.watch.watch(
this.path,
queryParams,
Expand All @@ -182,6 +192,9 @@ export class ListWatch<T extends KubernetesObject> implements ObjectCache<T>, In
}

private addOrUpdateItems(items: T[]): void {
if (items === undefined || items === null) {
return;
}
items.forEach((obj: T) => {
addOrUpdateObject(
this.objects,
Expand All @@ -192,13 +205,18 @@ export class ListWatch<T extends KubernetesObject> implements ObjectCache<T>, In
});
}

private watchHandler(phase: string, obj: T, watchObj?: any): void {
private async watchHandler(
phase: string,
obj: T,
watchObj?: { type: string; object: KubernetesObject },
): Promise<void> {
switch (phase) {
case 'ERROR':
if ((obj as { code?: number }).code === 410) {
this.resourceVersion = '';
}
break;
// We don't restart here, because it should be handled by the watch exiting if necessary
return;
case 'ADDED':
case 'MODIFIED':
addOrUpdateObject(
Expand All @@ -215,15 +233,16 @@ export class ListWatch<T extends KubernetesObject> implements ObjectCache<T>, In
// nothing to do, here for documentation, mostly.
break;
}
if (watchObj && watchObj.metadata) {
this.resourceVersion = watchObj.metadata.resourceVersion;
}
this.resourceVersion = obj.metadata ? obj.metadata!.resourceVersion || '' : '';
}
}

// exported for testing
export function cacheMapFromList<T extends KubernetesObject>(newObjects: T[]): CacheMap<T> {
const objects: CacheMap<T> = new Map();
if (newObjects === undefined || newObjects === null) {
return objects;
}
// build up the new list
for (const obj of newObjects) {
let namespaceObjects = objects.get(obj.metadata!.namespace || '');
Expand Down
122 changes: 96 additions & 26 deletions src/cache_test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -235,6 +235,7 @@ describe('ListWatchCache', () => {
watchHandler('ADDED', {
metadata: {
name: 'name3',
resourceVersion: 'blah',
} as V1ObjectMeta,
} as V1Namespace);

Expand All @@ -245,40 +246,28 @@ describe('ListWatchCache', () => {
} as V1ObjectMeta,
} as V1Namespace);

watchHandler(
'DELETED',
{
metadata: {
name: 'name2',
resourceVersion: 'blah',
} as V1ObjectMeta,
} as V1Namespace,
{
metadata: {
resourceVersion: '54321',
},
},
);
watchHandler('DELETED', {
metadata: {
name: 'name2',
resourceVersion: '54321',
} as V1ObjectMeta,
} as V1Namespace);

const [addResult, updateResult, deleteResult] = await Promise.all([
addPromise,
updatePromise,
deletePromise,
]);
deepStrictEqual(addResult.metadata, { name: 'name3' });
deepStrictEqual(addResult.metadata, { name: 'name3', resourceVersion: 'blah' });
deepStrictEqual(updateResult.metadata, { name: 'name3', resourceVersion: 'baz' });
deepStrictEqual(deleteResult.metadata, { name: 'name2', resourceVersion: 'blah' });
deepStrictEqual(deleteResult.metadata, { name: 'name2', resourceVersion: '54321' });
strictEqual(informer.latestResourceVersion(), '54321');

watchHandler(
'BOOKMARK',
{},
{
metadata: {
resourceVersion: '5454',
},
watchHandler('BOOKMARK', {
metadata: {
resourceVersion: '5454',
},
);
});
strictEqual(informer.latestResourceVersion(), '5454');
});

Expand Down Expand Up @@ -1205,9 +1194,10 @@ describe('ListWatchCache', () => {
{
metadata: {
name: 'name3',
resourceVersion: '23456',
} as V1ObjectMeta,
} as V1Namespace,
{ metadata: { resourceVersion: '23456' } },
{ type: 'ADDED', metadata: { resourceVersion: '23456' } },
);

await informer.stop();
Expand Down Expand Up @@ -1259,9 +1249,87 @@ describe('ListWatchCache', () => {
{
metadata: {
name: 'name3',
resourceVersion: '23456',
} as V1ObjectMeta,
} as V1Namespace,
{ metadata: { resourceVersion: '23456' } },
{ type: 'ADDED', metadata: { resourceVersion: '23456' } },
);

await informer.stop();

let errorEmitted = false;
informer.on('error', () => (errorEmitted = true));

promise = new Promise((resolve) => {
mock.when(
fakeWatch.watch(mock.anything(), mock.anything(), mock.anything(), mock.anything()),
).thenCall(() => {
resolve({});
});
});

informer.start();
await promise;

const [, , , doneHandler] = mock.capture(fakeWatch.watch).last();

const object = {
kind: 'Status',
apiVersion: 'v1',
metadata: {},
status: 'Failure',
message: 'too old resource version: 12345 (1234)',
reason: 'Expired',
code: 410,
};
await watchHandler('ERROR', object, { type: 'ERROR', object });
await doneHandler(null);

mock.verify(
fakeWatch.watch(mock.anything(), mock.anything(), mock.anything(), mock.anything()),
).thrice();
strictEqual(errorEmitted, false);
strictEqual(listCalls, 2);
});

it('should list if the watch errors from the last version', async () => {
const fakeWatch = mock.mock(Watch);

let listCalls = 0;
const listFn: ListPromise<V1Namespace> = function (): Promise<V1NamespaceList> {
return new Promise<V1NamespaceList>((resolve, reject) => {
listCalls++;
resolve({
metadata: {
resourceVersion: '12345',
} as V1ListMeta,
items: [],
} as V1NamespaceList);
});
};
let promise = new Promise((resolve) => {
mock.when(
fakeWatch.watch(mock.anything(), mock.anything(), mock.anything(), mock.anything()),
).thenCall(() => {
resolve({});
});
});

const informer = new ListWatch('/some/path', mock.instance(fakeWatch), listFn, false);

informer.start();
await promise;

const [, , watchHandler] = mock.capture(fakeWatch.watch).last();
watchHandler(
'ADDED',
{
metadata: {
name: 'name3',
resourceVersion: '23456',
} as V1ObjectMeta,
} as V1Namespace,
{ type: 'ADDED', metadata: { resourceVersion: '23456' } },
);

await informer.stop();
Expand Down Expand Up @@ -1335,6 +1403,8 @@ describe('ListWatchCache', () => {
);

await informer.stop();
strictEqual(listCalls, 1);
listCalls = 0;

let errorEmitted = false;
informer.on('error', () => (errorEmitted = true));
Expand Down