grpc-promise/lib/request-types/server-stream-request.js

GRPC promisify module for all Request/Response types: standard and stream

class ServerStreamRequest {

  constructor (client, original_function, options = {}) {
    if (options == null) options = {};
    this.queue = [];
    this.client = client;
    this.metadata = options.metadata;
    this.timeout = options.timeout || undefined;
    this.original_function = original_function;
  }

  sendMessage (content = {}) {
    return new Promise((resolve, reject) => {
      // Deadline is advisable to be set
      // It should be a timestamp value in milliseconds
      let deadline = undefined;
      if (this.timeout !== undefined) {
        deadline = Date.now() + this.timeout;
      }
      this.stream = this.original_function.call(this.client, content, this.metadata, { deadline: deadline });
      this.stream.on('error', error => {
        reject(error);
      });
      this.stream.on('data', data => {
        this.queue.push(data);
      });
      this.stream.on('end', () => {
        resolve(this.queue);
      });
    });
  }

}

const makeServerStreamRequest = function (client, originalFunction, options) {
  return function () {
    return new ServerStreamRequest(client, originalFunction, options);
  };
};

module.exports = makeServerStreamRequest;