mirror of
https://github.com/yu-i-i/overleaf-cep.git
synced 2025-08-01 05:00:09 +02:00

Stream and buffer consolidation GitOrigin-RevId: 284f411e6403e190d2dda3d9ebf806b1935b3949
28 lines
814 B
JavaScript
28 lines
814 B
JavaScript
const { LoggerStream, WritableBuffer } = require('@overleaf/stream-utils')
|
|
const Settings = require('@overleaf/settings')
|
|
const logger = require('@overleaf/logger/logging-manager')
|
|
const { pipeline } = require('node:stream/promises')
|
|
const { callbackify } = require('node:util')
|
|
|
|
module.exports = {
|
|
streamToBuffer: callbackify(streamToBuffer),
|
|
promises: {
|
|
streamToBuffer,
|
|
},
|
|
}
|
|
|
|
async function streamToBuffer(projectId, docId, stream) {
|
|
const loggerTransform = new LoggerStream(
|
|
Settings.max_doc_length,
|
|
(size, isFlush) => {
|
|
logger.warn(
|
|
{ projectId, docId, size, finishedReading: isFlush },
|
|
'potentially large doc pulled down from gcs'
|
|
)
|
|
}
|
|
)
|
|
|
|
const buffer = new WritableBuffer()
|
|
await pipeline(stream, loggerTransform, buffer)
|
|
return buffer.contents()
|
|
}
|