kibana/examples/bfetch_explorer/server/plugin.ts
2020-05-22 09:08:58 +02:00

69 lines
2.3 KiB
TypeScript

/*
* Licensed to Elasticsearch B.V. under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch B.V. licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
import { Subject } from 'rxjs';
import { Plugin, CoreSetup, CoreStart } from '../../../src/core/server';
import { BfetchServerSetup, BfetchServerStart } from '../../../src/plugins/bfetch/server';
export interface BfetchExplorerSetupPlugins {
bfetch: BfetchServerSetup;
}
export interface BfetchExplorerStartPlugins {
bfetch: BfetchServerStart;
}
export class BfetchExplorerPlugin implements Plugin {
public setup(core: CoreSetup, plugins: BfetchExplorerSetupPlugins) {
plugins.bfetch.addStreamingResponseRoute<string, string>('/bfetch_explorer/count', () => ({
getResponseStream: ({ data }: any) => {
const subject = new Subject<string>();
const countTo = Number(data);
for (let cnt = 1; cnt <= countTo; cnt++) {
setTimeout(() => {
subject.next(String(cnt));
}, cnt * 1000);
}
setTimeout(() => {
subject.complete();
}, countTo * 1000);
return subject;
},
}));
plugins.bfetch.addBatchProcessingRoute<{ num: number }, { num: number }>(
'/bfetch_explorer/double',
() => ({
onBatchItem: async ({ num }) => {
// Validate inputs.
if (num < 0) throw new Error('Invalid number');
// Wait number of specified milliseconds.
await new Promise((r) => setTimeout(r, num));
// Double the number and send it back.
return { num: 2 * num };
},
})
);
}
public start(core: CoreStart, plugins: BfetchExplorerStartPlugins) {}
public stop() {}
}