grpc-host-builder/src/implementationsWrappers/rxToServerWritableStream.js

Lightweight configurator for gRPC host

const { fromEvent } = require("rxjs");
const { takeUntil, catchError } = require("rxjs/operators");

/**
 * @param {import("../index").serviceServerStreamingMethodImplementation<any, any> | import("../index").serviceBidiStreamingMethodImplementation<any, any>} handler
 * @returns {import("../index").handleServerStreamingCall<any, any> | import("../index").handleBidiStreamingCall<any, any>}
 */
module.exports = function(handler) {
  return async call => {
    const result = await handler(call);
    await result
      .pipe(
        takeUntil(fromEvent(call, "cancelled")),
        catchError(err => {
          throw err;
        })
      )
      .forEach(message => call.write(message));
  };
};