mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-07 20:31:06 -05:00
asyncify MessageManager
GitOrigin-RevId: 6b89b70f7c19fc9e71d60b634808030f324f40fe
This commit is contained in:
parent
49e2e02105
commit
01cdb4025e
1 changed files with 28 additions and 38 deletions
|
@ -2,8 +2,9 @@ let MessageManager
|
|||
const { db, ObjectId } = require('../../mongodb')
|
||||
const metrics = require('@overleaf/metrics')
|
||||
const logger = require('@overleaf/logger')
|
||||
const { callbackify } = require('util')
|
||||
|
||||
function createMessage(roomId, userId, content, timestamp, callback) {
|
||||
async function createMessage(roomId, userId, content, timestamp) {
|
||||
let newMessageOpts = {
|
||||
content,
|
||||
room_id: roomId,
|
||||
|
@ -11,64 +12,53 @@ function createMessage(roomId, userId, content, timestamp, callback) {
|
|||
timestamp,
|
||||
}
|
||||
newMessageOpts = _ensureIdsAreObjectIds(newMessageOpts)
|
||||
db.messages.insertOne(newMessageOpts, function (error, confirmation) {
|
||||
if (error) {
|
||||
return callback(error)
|
||||
}
|
||||
newMessageOpts._id = confirmation.insertedId
|
||||
callback(null, newMessageOpts)
|
||||
})
|
||||
const confirmation = await db.messages.insertOne(newMessageOpts)
|
||||
newMessageOpts._id = confirmation.insertedId
|
||||
return newMessageOpts
|
||||
}
|
||||
|
||||
function getMessages(roomId, limit, before, callback) {
|
||||
async function getMessages(roomId, limit, before) {
|
||||
let query = { room_id: roomId }
|
||||
if (before) {
|
||||
query.timestamp = { $lt: before }
|
||||
}
|
||||
query = _ensureIdsAreObjectIds(query)
|
||||
db.messages.find(query).sort({ timestamp: -1 }).limit(limit).toArray(callback)
|
||||
return db.messages.find(query).sort({ timestamp: -1 }).limit(limit).toArray()
|
||||
}
|
||||
|
||||
function findAllMessagesInRooms(roomIds, callback) {
|
||||
db.messages
|
||||
async function findAllMessagesInRooms(roomIds) {
|
||||
return db.messages
|
||||
.find({
|
||||
room_id: { $in: roomIds },
|
||||
})
|
||||
.toArray(callback)
|
||||
.toArray()
|
||||
}
|
||||
|
||||
function deleteAllMessagesInRoom(roomId, callback) {
|
||||
db.messages.deleteMany(
|
||||
{
|
||||
room_id: roomId,
|
||||
},
|
||||
callback
|
||||
)
|
||||
async function deleteAllMessagesInRoom(roomId) {
|
||||
await db.messages.deleteMany({
|
||||
room_id: roomId,
|
||||
})
|
||||
}
|
||||
|
||||
function updateMessage(roomId, messageId, content, timestamp, callback) {
|
||||
async function updateMessage(roomId, messageId, content, timestamp) {
|
||||
const query = _ensureIdsAreObjectIds({
|
||||
_id: messageId,
|
||||
room_id: roomId,
|
||||
})
|
||||
db.messages.updateOne(
|
||||
query,
|
||||
{
|
||||
$set: {
|
||||
content,
|
||||
edited_at: timestamp,
|
||||
},
|
||||
await db.messages.updateOne(query, {
|
||||
$set: {
|
||||
content,
|
||||
edited_at: timestamp,
|
||||
},
|
||||
callback
|
||||
)
|
||||
})
|
||||
}
|
||||
|
||||
function deleteMessage(roomId, messageId, callback) {
|
||||
async function deleteMessage(roomId, messageId) {
|
||||
const query = _ensureIdsAreObjectIds({
|
||||
_id: messageId,
|
||||
room_id: roomId,
|
||||
})
|
||||
db.messages.deleteOne(query, callback)
|
||||
await db.messages.deleteOne(query)
|
||||
}
|
||||
|
||||
function _ensureIdsAreObjectIds(query) {
|
||||
|
@ -85,12 +75,12 @@ function _ensureIdsAreObjectIds(query) {
|
|||
}
|
||||
|
||||
module.exports = MessageManager = {
|
||||
createMessage,
|
||||
getMessages,
|
||||
findAllMessagesInRooms,
|
||||
deleteAllMessagesInRoom,
|
||||
updateMessage,
|
||||
deleteMessage,
|
||||
createMessage: callbackify(createMessage),
|
||||
getMessages: callbackify(getMessages),
|
||||
findAllMessagesInRooms: callbackify(findAllMessagesInRooms),
|
||||
deleteAllMessagesInRoom: callbackify(deleteAllMessagesInRoom),
|
||||
updateMessage: callbackify(updateMessage),
|
||||
deleteMessage: callbackify(deleteMessage),
|
||||
}
|
||||
;[
|
||||
'createMessage',
|
||||
|
|
Loading…
Reference in a new issue