mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-21 20:47:08 -05:00
7f48c67512
* Add `unicorn/prefer-node-protocol` * Fix `unicorn/prefer-node-protocol` ESLint errors * Run `npm run format:fix` * Add sandboxed-module sourceTransformers in mocha setups Fix `no such file or directory, open 'node:fs'` in `sandboxed-module` * Remove `node:` in the SandboxedModule requires * Fix new linting errors with `node:` GitOrigin-RevId: 68f6e31e2191fcff4cb8058dd0a6914c14f59926
82 lines
2.5 KiB
JavaScript
82 lines
2.5 KiB
JavaScript
const crypto = require('node:crypto')
|
|
const benny = require('benny')
|
|
const { Blob } = require('overleaf-editor-core')
|
|
const mongoBackend = require('../storage/lib/blob_store/mongo')
|
|
const postgresBackend = require('../storage/lib/blob_store/postgres')
|
|
const cleanup = require('../test/acceptance/js/storage/support/cleanup')
|
|
|
|
const MONGO_PROJECT_ID = '637386deb4ce3c62acd3848e'
|
|
const POSTGRES_PROJECT_ID = '123'
|
|
|
|
async function run() {
|
|
for (const blobCount of [1, 10, 100, 1000, 10000, 100000, 500000]) {
|
|
await cleanup.everything()
|
|
const blobs = createBlobs(blobCount)
|
|
await insertBlobs(blobs)
|
|
const randomHashes = getRandomHashes(blobs, 100)
|
|
await benny.suite(
|
|
`Read a blob in a project with ${blobCount} blobs`,
|
|
benny.add('Mongo backend', async () => {
|
|
await mongoBackend.findBlob(MONGO_PROJECT_ID, randomHashes[0])
|
|
}),
|
|
benny.add('Postgres backend', async () => {
|
|
await postgresBackend.findBlob(POSTGRES_PROJECT_ID, randomHashes[0])
|
|
}),
|
|
benny.cycle(),
|
|
benny.complete()
|
|
)
|
|
await benny.suite(
|
|
`Read 100 blobs in a project with ${blobCount} blobs`,
|
|
benny.add('Mongo backend', async () => {
|
|
await mongoBackend.findBlobs(MONGO_PROJECT_ID, randomHashes)
|
|
}),
|
|
benny.add('Postgres backend', async () => {
|
|
await postgresBackend.findBlobs(POSTGRES_PROJECT_ID, randomHashes)
|
|
}),
|
|
benny.cycle(),
|
|
benny.complete()
|
|
)
|
|
await benny.suite(
|
|
`Insert a blob in a project with ${blobCount} blobs`,
|
|
benny.add('Mongo backend', async () => {
|
|
const [newBlob] = createBlobs(1)
|
|
await mongoBackend.insertBlob(MONGO_PROJECT_ID, newBlob)
|
|
}),
|
|
benny.add('Postgres backend', async () => {
|
|
const [newBlob] = createBlobs(1)
|
|
await postgresBackend.insertBlob(POSTGRES_PROJECT_ID, newBlob)
|
|
}),
|
|
benny.cycle(),
|
|
benny.complete()
|
|
)
|
|
}
|
|
}
|
|
|
|
function createBlobs(blobCount) {
|
|
const blobs = []
|
|
for (let i = 0; i < blobCount; i++) {
|
|
const hash = crypto.randomBytes(20).toString('hex')
|
|
blobs.push(new Blob(hash, 42, 42))
|
|
}
|
|
return blobs
|
|
}
|
|
|
|
async function insertBlobs(blobs) {
|
|
for (const blob of blobs) {
|
|
await Promise.all([
|
|
mongoBackend.insertBlob(MONGO_PROJECT_ID, blob),
|
|
postgresBackend.insertBlob(POSTGRES_PROJECT_ID, blob),
|
|
])
|
|
}
|
|
}
|
|
|
|
function getRandomHashes(blobs, count) {
|
|
const hashes = []
|
|
for (let i = 0; i < count; i++) {
|
|
const index = Math.floor(Math.random() * blobs.length)
|
|
hashes.push(blobs[index].getHash())
|
|
}
|
|
return hashes
|
|
}
|
|
|
|
module.exports = run
|