OpenHPS/openhps-rest

import { DataFrame, SinkNode, ModelBuilder, Edge } from '@openhps/core';
import { RESTServerNode } from '../RESTServerNode';
import { ServerOptions } from '../ServerOptions';

/**
 * @category Server
 */
export class RESTServerSink<In extends DataFrame> extends SinkNode<In> {
    private _remoteNode: RESTServerNode<In, In>;

    constructor(options: ServerOptions, inputType?: new () => In) {
        super(options);
        this._remoteNode = new RESTServerNode<In, In>(options, inputType);
        this.once('build', this._onRemoteBuild.bind(this));
    }

    private _onRemoteBuild(graphBuilder: ModelBuilder<any, any>): Promise<boolean> {
        // Add a remote node before this node
        this._remoteNode.graph = this.graph;
        this._remoteNode.logger = this.logger;
        graphBuilder.addNode(this._remoteNode);
        graphBuilder.addEdge(new Edge(this, this._remoteNode));
        return this._remoteNode.emitAsync('build', graphBuilder);
    }

    public onPush(data: In): Promise<void> {
        return this._remoteNode.push(data);
    }
}