StreamToBuffer.js 814 B

12345678910111213141516171819202122232425262728
  1. const { LoggerStream, WritableBuffer } = require('@overleaf/stream-utils')
  2. const Settings = require('@overleaf/settings')
  3. const logger = require('@overleaf/logger/logging-manager')
  4. const { pipeline } = require('node:stream/promises')
  5. const { callbackify } = require('node:util')
  6. module.exports = {
  7. streamToBuffer: callbackify(streamToBuffer),
  8. promises: {
  9. streamToBuffer,
  10. },
  11. }
  12. async function streamToBuffer(projectId, docId, stream) {
  13. const loggerTransform = new LoggerStream(
  14. Settings.max_doc_length,
  15. (size, isFlush) => {
  16. logger.warn(
  17. { projectId, docId, size, finishedReading: isFlush },
  18. 'potentially large doc pulled down from gcs'
  19. )
  20. }
  21. )
  22. const buffer = new WritableBuffer()
  23. await pipeline(stream, loggerTransform, buffer)
  24. return buffer.contents()
  25. }