Optimize a few queries

This commit is contained in:
Scott Nonnenberg 2021-12-20 13:04:02 -08:00 committed by GitHub
parent b08691b35b
commit 60a53656af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
27 changed files with 1288 additions and 186 deletions

View File

@ -60,7 +60,9 @@ exports.createConversation = async ({
await sleep(index * 100); await sleep(index * 100);
window.SignalContext.log.info(`Create message ${index + 1}`); window.SignalContext.log.info(`Create message ${index + 1}`);
const message = await createRandomMessage({ conversationId }); const message = await createRandomMessage({ conversationId });
return Signal.Data.saveMessage(message); return Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
}) })
); );
}; };

View File

@ -65,7 +65,13 @@ exports.processNext = async ({
const upgradeDuration = Date.now() - upgradeStartTime; const upgradeDuration = Date.now() - upgradeStartTime;
const saveStartTime = Date.now(); const saveStartTime = Date.now();
await Promise.all(upgradedMessages.map(message => saveMessage(message))); await Promise.all(
upgradedMessages.map(message =>
saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
})
)
);
const saveDuration = Date.now() - saveStartTime; const saveDuration = Date.now() - saveStartTime;
const totalDuration = Date.now() - startTime; const totalDuration = Date.now() - startTime;

View File

@ -1701,7 +1701,9 @@ export async function startApp(): Promise<void> {
}; };
}); });
await window.Signal.Data.saveMessages(newMessageAttributes); await window.Signal.Data.saveMessages(newMessageAttributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
log.info('Expiration start timestamp cleanup: complete'); log.info('Expiration start timestamp cleanup: complete');
@ -2369,7 +2371,9 @@ export async function startApp(): Promise<void> {
messagesToSave.push(message.attributes); messagesToSave.push(message.attributes);
} }
}); });
await window.Signal.Data.saveMessages(messagesToSave); await window.Signal.Data.saveMessages(messagesToSave, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
function onReconnect() { function onReconnect() {
// We disable notifications on first connect, but the same applies to reconnect. In // We disable notifications on first connect, but the same applies to reconnect. In

View File

@ -1716,6 +1716,7 @@ export async function createGroupV2({
}; };
await window.Signal.Data.saveMessages([createdTheGroupMessage], { await window.Signal.Data.saveMessages([createdTheGroupMessage], {
forceSave: true, forceSave: true,
ourUuid,
}); });
const model = new window.Whisper.Message(createdTheGroupMessage); const model = new window.Whisper.Message(createdTheGroupMessage);
window.MessageController.register(model.id, model); window.MessageController.register(model.id, model);
@ -2861,6 +2862,7 @@ async function updateGroup(
if (changeMessagesToSave.length > 0) { if (changeMessagesToSave.length > 0) {
await window.Signal.Data.saveMessages(changeMessagesToSave, { await window.Signal.Data.saveMessages(changeMessagesToSave, {
forceSave: true, forceSave: true,
ourUuid: ourUuid.toString(),
}); });
changeMessagesToSave.forEach(changeMessage => { changeMessagesToSave.forEach(changeMessage => {
const model = new window.Whisper.Message(changeMessage); const model = new window.Whisper.Message(changeMessage);

View File

@ -481,7 +481,9 @@ async function markMessageFailed(
): Promise<void> { ): Promise<void> {
message.markFailed(); message.markFailed();
message.saveErrors(errors, { skipSave: true }); message.saveErrors(errors, { skipSave: true });
await window.Signal.Data.saveMessage(message.attributes); await window.Signal.Data.saveMessage(message.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
function didSendToEveryone(message: Readonly<MessageModel>): boolean { function didSendToEveryone(message: Readonly<MessageModel>): boolean {

View File

@ -58,6 +58,7 @@ export class ReactionJobQueue extends JobQueue<ReactionJobData> {
{ attempt, log }: Readonly<{ attempt: number; log: LoggerType }> { attempt, log }: Readonly<{ attempt: number; log: LoggerType }>
): Promise<void> { ): Promise<void> {
const { messageId } = data; const { messageId } = data;
const ourUuid = window.textsecure.storage.user.getCheckedUuid().toString();
const timeRemaining = timestamp + MAX_RETRY_TIME - Date.now(); const timeRemaining = timestamp + MAX_RETRY_TIME - Date.now();
const isFinalAttempt = attempt >= MAX_ATTEMPTS; const isFinalAttempt = attempt >= MAX_ATTEMPTS;
@ -100,7 +101,7 @@ export class ReactionJobQueue extends JobQueue<ReactionJobData> {
`could not react to ${messageId}. Removing this pending reaction` `could not react to ${messageId}. Removing this pending reaction`
); );
markReactionFailed(message, pendingReaction); markReactionFailed(message, pendingReaction);
await window.Signal.Data.saveMessage(message.attributes); await window.Signal.Data.saveMessage(message.attributes, { ourUuid });
return; return;
} }
@ -109,7 +110,7 @@ export class ReactionJobQueue extends JobQueue<ReactionJobData> {
`reacting to message ${messageId} ran out of time. Giving up on sending it` `reacting to message ${messageId} ran out of time. Giving up on sending it`
); );
markReactionFailed(message, pendingReaction); markReactionFailed(message, pendingReaction);
await window.Signal.Data.saveMessage(message.attributes); await window.Signal.Data.saveMessage(message.attributes, { ourUuid });
return; return;
} }
@ -257,7 +258,7 @@ export class ReactionJobQueue extends JobQueue<ReactionJobData> {
} }
await handleCommonJobRequestError({ err, log, timeRemaining }); await handleCommonJobRequestError({ err, log, timeRemaining });
} finally { } finally {
await window.Signal.Data.saveMessage(message.attributes); await window.Signal.Data.saveMessage(message.attributes, { ourUuid });
} }
} }
} }

View File

@ -278,7 +278,9 @@ async function _finishJob(
): Promise<void> { ): Promise<void> {
if (message) { if (message) {
logger.info(`attachment_downloads/_finishJob for job id: ${id}`); logger.info(`attachment_downloads/_finishJob for job id: ${id}`);
await saveMessage(message.attributes); await saveMessage(message.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
await removeAttachmentDownloadJob(id); await removeAttachmentDownloadJob(id);

View File

@ -1622,6 +1622,7 @@ export class ConversationModel extends window.Backbone
if (eliminated > 0) { if (eliminated > 0) {
log.warn(`cleanModels: Eliminated ${eliminated} messages without an id`); log.warn(`cleanModels: Eliminated ${eliminated} messages without an id`);
} }
const ourUuid = window.textsecure.storage.user.getCheckedUuid().toString();
let upgraded = 0; let upgraded = 0;
for (let max = result.length, i = 0; i < max; i += 1) { for (let max = result.length, i = 0; i < max; i += 1) {
@ -1635,7 +1636,7 @@ export class ConversationModel extends window.Backbone
const upgradedMessage = await upgradeMessageSchema(attributes); const upgradedMessage = await upgradeMessageSchema(attributes);
message.set(upgradedMessage); message.set(upgradedMessage);
// eslint-disable-next-line no-await-in-loop // eslint-disable-next-line no-await-in-loop
await window.Signal.Data.saveMessage(upgradedMessage); await window.Signal.Data.saveMessage(upgradedMessage, { ourUuid });
upgraded += 1; upgraded += 1;
} }
} }
@ -1950,6 +1951,7 @@ export class ConversationModel extends window.Backbone
options: { isLocalAction?: boolean } = {} options: { isLocalAction?: boolean } = {}
): Promise<void> { ): Promise<void> {
const { isLocalAction } = options; const { isLocalAction } = options;
const ourUuid = window.textsecure.storage.user.getCheckedUuid().toString();
let messages: Array<MessageAttributesType> | undefined; let messages: Array<MessageAttributesType> | undefined;
do { do {
@ -1991,7 +1993,9 @@ export class ConversationModel extends window.Backbone
const registered = window.MessageController.register(m.id, m); const registered = window.MessageController.register(m.id, m);
const shouldSave = await registered.queueAttachmentDownloads(); const shouldSave = await registered.queueAttachmentDownloads();
if (shouldSave) { if (shouldSave) {
await window.Signal.Data.saveMessage(registered.attributes); await window.Signal.Data.saveMessage(registered.attributes, {
ourUuid,
});
} }
}) })
); );
@ -2838,7 +2842,9 @@ export class ConversationModel extends window.Backbone
// this type does not fully implement the interface it is expected to // this type does not fully implement the interface it is expected to
} as unknown as MessageAttributesType; } as unknown as MessageAttributesType;
const id = await window.Signal.Data.saveMessage(message); const id = await window.Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
new window.Whisper.Message({ new window.Whisper.Message({
@ -2878,7 +2884,9 @@ export class ConversationModel extends window.Backbone
// this type does not fully implement the interface it is expected to // this type does not fully implement the interface it is expected to
} as unknown as MessageAttributesType; } as unknown as MessageAttributesType;
const id = await window.Signal.Data.saveMessage(message); const id = await window.Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
new window.Whisper.Message({ new window.Whisper.Message({
@ -2914,7 +2922,9 @@ export class ConversationModel extends window.Backbone
// this type does not fully implement the interface it is expected to // this type does not fully implement the interface it is expected to
} as unknown as MessageAttributesType; } as unknown as MessageAttributesType;
const id = await window.Signal.Data.saveMessage(message); const id = await window.Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
new window.Whisper.Message({ new window.Whisper.Message({
@ -2970,7 +2980,9 @@ export class ConversationModel extends window.Backbone
// TODO: DESKTOP-722 // TODO: DESKTOP-722
} as unknown as MessageAttributesType; } as unknown as MessageAttributesType;
const id = await window.Signal.Data.saveMessage(message); const id = await window.Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
new window.Whisper.Message({ new window.Whisper.Message({
@ -3030,7 +3042,9 @@ export class ConversationModel extends window.Backbone
// TODO: DESKTOP-722 // TODO: DESKTOP-722
} as unknown as MessageAttributesType; } as unknown as MessageAttributesType;
const id = await window.Signal.Data.saveMessage(message); const id = await window.Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
new window.Whisper.Message({ new window.Whisper.Message({
@ -3090,7 +3104,9 @@ export class ConversationModel extends window.Backbone
// TODO: DESKTOP-722 // TODO: DESKTOP-722
} as unknown as MessageAttributesType; } as unknown as MessageAttributesType;
const id = await window.Signal.Data.saveMessage(message); const id = await window.Signal.Data.saveMessage(message, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
new window.Whisper.Message({ new window.Whisper.Message({
@ -3131,7 +3147,10 @@ export class ConversationModel extends window.Backbone
const id = await window.Signal.Data.saveMessage( const id = await window.Signal.Data.saveMessage(
// TODO: DESKTOP-722 // TODO: DESKTOP-722
message as MessageAttributesType message as MessageAttributesType,
{
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
}
); );
const model = window.MessageController.register( const model = window.MessageController.register(
id, id,
@ -3948,6 +3967,7 @@ export class ConversationModel extends window.Backbone
await window.Signal.Data.saveMessage(message.attributes, { await window.Signal.Data.saveMessage(message.attributes, {
jobToInsert, jobToInsert,
forceSave: true, forceSave: true,
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
}); });
} }
); );
@ -4388,7 +4408,9 @@ export class ConversationModel extends window.Backbone
// TODO: DESKTOP-722 // TODO: DESKTOP-722
} as unknown as MessageAttributesType); } as unknown as MessageAttributesType);
const id = await window.Signal.Data.saveMessage(model.attributes); const id = await window.Signal.Data.saveMessage(model.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
model.set({ id }); model.set({ id });
@ -4484,7 +4506,9 @@ export class ConversationModel extends window.Backbone
// TODO: DESKTOP-722 // TODO: DESKTOP-722
} as unknown as MessageAttributesType); } as unknown as MessageAttributesType);
const id = await window.Signal.Data.saveMessage(model.attributes); const id = await window.Signal.Data.saveMessage(model.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
model.set({ id }); model.set({ id });
@ -4519,7 +4543,9 @@ export class ConversationModel extends window.Backbone
// TODO: DESKTOP-722 // TODO: DESKTOP-722
} as unknown as MessageAttributesType); } as unknown as MessageAttributesType);
const id = await window.Signal.Data.saveMessage(model.attributes); const id = await window.Signal.Data.saveMessage(model.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
model.set({ id }); model.set({ id });
const message = window.MessageController.register(model.id, model); const message = window.MessageController.register(model.id, model);

View File

@ -965,7 +965,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
this.getConversation()?.debouncedUpdateLastMessage?.(); this.getConversation()?.debouncedUpdateLastMessage?.();
if (shouldPersist) { if (shouldPersist) {
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
await window.Signal.Data.deleteSentProtoByMessageId(this.id); await window.Signal.Data.deleteSentProtoByMessageId(this.id);
@ -1099,7 +1101,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
} }
if (!skipSave && !this.doNotSave) { if (!skipSave && !this.doNotSave) {
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
} }
@ -1161,7 +1165,10 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
await normalMessageSendJobQueue.add( await normalMessageSendJobQueue.add(
{ messageId: this.id, conversationId: conversation.id }, { messageId: this.id, conversationId: conversation.id },
async jobToInsert => { async jobToInsert => {
await window.Signal.Data.saveMessage(this.attributes, { jobToInsert }); await window.Signal.Data.saveMessage(this.attributes, {
jobToInsert,
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
); );
} }
@ -1256,7 +1263,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
} }
if (!this.doNotSave) { if (!this.doNotSave) {
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
const sendStateByConversationId = { const sendStateByConversationId = {
@ -1403,7 +1412,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
} }
if (!this.doNotSave) { if (!this.doNotSave) {
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
updateLeftPane(); updateLeftPane();
@ -1460,7 +1471,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
this.saveErrors(errors, { skipSave: true }); this.saveErrors(errors, { skipSave: true });
} }
} finally { } finally {
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
if (updateLeftPane) { if (updateLeftPane) {
updateLeftPane(); updateLeftPane();
@ -1571,7 +1584,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
return result; return result;
} }
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
return result; return result;
}); });
}; };
@ -2097,7 +2112,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
originalMessage.attributes originalMessage.attributes
); );
originalMessage.set(upgradedMessage); originalMessage.set(upgradedMessage);
await window.Signal.Data.saveMessage(upgradedMessage); await window.Signal.Data.saveMessage(upgradedMessage, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
} catch (error) { } catch (error) {
log.error( log.error(
@ -2264,7 +2281,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
sendStateByConversationId, sendStateByConversationId,
unidentifiedDeliveries: [...unidentifiedDeliveriesSet], unidentifiedDeliveries: [...unidentifiedDeliveriesSet],
}); });
await window.Signal.Data.saveMessage(toUpdate.attributes); await window.Signal.Data.saveMessage(toUpdate.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
confirm(); confirm();
return; return;
@ -3016,7 +3035,9 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
log.info( log.info(
`modifyTargetMessage/${this.idForLogging()}: Changes in second run; saving.` `modifyTargetMessage/${this.idForLogging()}: Changes in second run; saving.`
); );
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
} }
@ -3166,13 +3187,16 @@ export class MessageModel extends window.Backbone.Model<MessageAttributesType> {
); );
await window.Signal.Data.saveMessage(this.attributes, { await window.Signal.Data.saveMessage(this.attributes, {
jobToInsert, jobToInsert,
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
}); });
}); });
} else { } else {
await reactionJobQueue.add(jobData); await reactionJobQueue.add(jobData);
} }
} else if (shouldPersist) { } else if (shouldPersist) {
await window.Signal.Data.saveMessage(this.attributes); await window.Signal.Data.saveMessage(this.attributes, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
} }

View File

@ -1076,7 +1076,11 @@ async function getMessageCount(conversationId?: string) {
async function saveMessage( async function saveMessage(
data: MessageType, data: MessageType,
options: { jobToInsert?: Readonly<StoredJob>; forceSave?: boolean } = {} options: {
jobToInsert?: Readonly<StoredJob>;
forceSave?: boolean;
ourUuid: UUIDStringType;
}
) { ) {
const id = await channels.saveMessage(_cleanMessageData(data), { const id = await channels.saveMessage(_cleanMessageData(data), {
...options, ...options,
@ -1091,7 +1095,7 @@ async function saveMessage(
async function saveMessages( async function saveMessages(
arrayOfMessages: Array<MessageType>, arrayOfMessages: Array<MessageType>,
options?: { forceSave?: boolean } options: { forceSave?: boolean; ourUuid: UUIDStringType }
) { ) {
await channels.saveMessages( await channels.saveMessages(
arrayOfMessages.map(message => _cleanMessageData(message)), arrayOfMessages.map(message => _cleanMessageData(message)),

View File

@ -371,14 +371,15 @@ export type DataInterface = {
getMessageCount: (conversationId?: string) => Promise<number>; getMessageCount: (conversationId?: string) => Promise<number>;
saveMessage: ( saveMessage: (
data: MessageType, data: MessageType,
options?: { options: {
jobToInsert?: StoredJob; jobToInsert?: StoredJob;
forceSave?: boolean; forceSave?: boolean;
ourUuid: UUIDStringType;
} }
) => Promise<string>; ) => Promise<string>;
saveMessages: ( saveMessages: (
arrayOfMessages: Array<MessageType>, arrayOfMessages: Array<MessageType>,
options?: { forceSave?: boolean } options: { forceSave?: boolean; ourUuid: UUIDStringType }
) => Promise<void>; ) => Promise<void>;
removeMessage: (id: string) => Promise<void>; removeMessage: (id: string) => Promise<void>;
removeMessages: (ids: Array<string>) => Promise<void>; removeMessages: (ids: Array<string>) => Promise<void>;

View File

@ -1688,20 +1688,7 @@ function hasUserInitiatedMessages(conversationId: string): boolean {
SELECT 1 FROM messages SELECT 1 FROM messages
WHERE WHERE
conversationId = $conversationId AND conversationId = $conversationId AND
(type IS NULL isUserInitiatedMessage = 1
OR
type NOT IN (
'change-number-notification',
'group-v1-migration',
'group-v2-change',
'keychange',
'message-history-unsynced',
'profile-change',
'story',
'universal-timer-notification',
'verified-change'
)
)
LIMIT 1 LIMIT 1
); );
` `
@ -1714,17 +1701,19 @@ function hasUserInitiatedMessages(conversationId: string): boolean {
function saveMessageSync( function saveMessageSync(
data: MessageType, data: MessageType,
options: { options: {
jobToInsert?: StoredJob;
forceSave?: boolean;
alreadyInTransaction?: boolean; alreadyInTransaction?: boolean;
db?: Database; db?: Database;
} = {} forceSave?: boolean;
jobToInsert?: StoredJob;
ourUuid: UUIDStringType;
}
): string { ): string {
const { const {
jobToInsert,
forceSave,
alreadyInTransaction, alreadyInTransaction,
db = getInstance(), db = getInstance(),
forceSave,
jobToInsert,
ourUuid,
} = options; } = options;
if (!alreadyInTransaction) { if (!alreadyInTransaction) {
@ -1741,6 +1730,7 @@ function saveMessageSync(
const { const {
body, body,
conversationId, conversationId,
groupV2Change,
hasAttachments, hasAttachments,
hasFileAttachments, hasFileAttachments,
hasVisualMediaAttachments, hasVisualMediaAttachments,
@ -1772,6 +1762,7 @@ function saveMessageSync(
hasAttachments: hasAttachments ? 1 : 0, hasAttachments: hasAttachments ? 1 : 0,
hasFileAttachments: hasFileAttachments ? 1 : 0, hasFileAttachments: hasFileAttachments ? 1 : 0,
hasVisualMediaAttachments: hasVisualMediaAttachments ? 1 : 0, hasVisualMediaAttachments: hasVisualMediaAttachments ? 1 : 0,
isChangeCreatedByUs: groupV2Change?.from === ourUuid ? 1 : 0,
isErased: isErased ? 1 : 0, isErased: isErased ? 1 : 0,
isViewOnce: isViewOnce ? 1 : 0, isViewOnce: isViewOnce ? 1 : 0,
received_at: received_at || null, received_at: received_at || null,
@ -1801,6 +1792,7 @@ function saveMessageSync(
hasAttachments = $hasAttachments, hasAttachments = $hasAttachments,
hasFileAttachments = $hasFileAttachments, hasFileAttachments = $hasFileAttachments,
hasVisualMediaAttachments = $hasVisualMediaAttachments, hasVisualMediaAttachments = $hasVisualMediaAttachments,
isChangeCreatedByUs = $isChangeCreatedByUs,
isErased = $isErased, isErased = $isErased,
isViewOnce = $isViewOnce, isViewOnce = $isViewOnce,
received_at = $received_at, received_at = $received_at,
@ -1843,6 +1835,7 @@ function saveMessageSync(
hasAttachments, hasAttachments,
hasFileAttachments, hasFileAttachments,
hasVisualMediaAttachments, hasVisualMediaAttachments,
isChangeCreatedByUs,
isErased, isErased,
isViewOnce, isViewOnce,
received_at, received_at,
@ -1866,6 +1859,7 @@ function saveMessageSync(
$hasAttachments, $hasAttachments,
$hasFileAttachments, $hasFileAttachments,
$hasVisualMediaAttachments, $hasVisualMediaAttachments,
$isChangeCreatedByUs,
$isErased, $isErased,
$isViewOnce, $isViewOnce,
$received_at, $received_at,
@ -1895,10 +1889,11 @@ function saveMessageSync(
async function saveMessage( async function saveMessage(
data: MessageType, data: MessageType,
options?: { options: {
jobToInsert?: StoredJob; jobToInsert?: StoredJob;
forceSave?: boolean; forceSave?: boolean;
alreadyInTransaction?: boolean; alreadyInTransaction?: boolean;
ourUuid: UUIDStringType;
} }
): Promise<string> { ): Promise<string> {
return saveMessageSync(data, options); return saveMessageSync(data, options);
@ -1906,15 +1901,14 @@ async function saveMessage(
async function saveMessages( async function saveMessages(
arrayOfMessages: Array<MessageType>, arrayOfMessages: Array<MessageType>,
options?: { forceSave?: boolean } options: { forceSave?: boolean; ourUuid: UUIDStringType }
): Promise<void> { ): Promise<void> {
const db = getInstance(); const db = getInstance();
const { forceSave } = options || {};
db.transaction(() => { db.transaction(() => {
for (const message of arrayOfMessages) { for (const message of arrayOfMessages) {
assertSync( assertSync(
saveMessageSync(message, { forceSave, alreadyInTransaction: true }) saveMessageSync(message, { ...options, alreadyInTransaction: true })
); );
} }
})(); })();
@ -2070,7 +2064,7 @@ async function getUnreadByConversationAndMarkRead({
expirationStartTimestamp IS NULL OR expirationStartTimestamp IS NULL OR
expirationStartTimestamp > $expirationStartTimestamp expirationStartTimestamp > $expirationStartTimestamp
) AND ) AND
expireTimer IS NOT NULL AND expireTimer > 0 AND
conversationId = $conversationId AND conversationId = $conversationId AND
storyId IS $storyId AND storyId IS $storyId AND
received_at <= $newestUnreadAt; received_at <= $newestUnreadAt;
@ -2161,7 +2155,7 @@ async function getUnreadReactionsAndMarkRead({
FROM reactions FROM reactions
JOIN messages on messages.id IS reactions.messageId JOIN messages on messages.id IS reactions.messageId
WHERE WHERE
unread IS NOT 0 AND unread > 0 AND
messages.conversationId IS $conversationId AND messages.conversationId IS $conversationId AND
messages.received_at <= $newestUnreadAt AND messages.received_at <= $newestUnreadAt AND
messages.storyId IS $storyId messages.storyId IS $storyId
@ -2504,31 +2498,9 @@ function getLastConversationActivity({
SELECT json FROM messages SELECT json FROM messages
WHERE WHERE
conversationId = $conversationId AND conversationId = $conversationId AND
(type IS NULL shouldAffectActivity IS 1 AND
OR isTimerChangeFromSync IS 0 AND
type NOT IN ( isGroupLeaveEventFromOther IS 0
'change-number-notification',
'group-v1-migration',
'keychange',
'message-history-unsynced',
'profile-change',
'story',
'universal-timer-notification',
'verified-change'
)
) AND
(
json_extract(json, '$.expirationTimerUpdate.fromSync') IS NULL
OR
json_extract(json, '$.expirationTimerUpdate.fromSync') != 1
) AND NOT
(
type IS 'group-v2-change' AND
json_extract(json, '$.groupV2Change.from') IS NOT $ourUuid AND
json_extract(json, '$.groupV2Change.details.length') IS 1 AND
json_extract(json, '$.groupV2Change.details[0].type') IS 'member-remove' AND
json_extract(json, '$.groupV2Change.details[0].uuid') IS NOT $ourUuid
)
ORDER BY received_at DESC, sent_at DESC ORDER BY received_at DESC, sent_at DESC
LIMIT 1; LIMIT 1;
` `
@ -2557,29 +2529,12 @@ function getLastConversationPreview({
SELECT json FROM messages SELECT json FROM messages
WHERE WHERE
conversationId = $conversationId AND conversationId = $conversationId AND
shouldAffectPreview IS 1 AND
isGroupLeaveEventFromOther IS 0 AND
( (
expiresAt IS NULL OR expiresAt IS NULL
(expiresAt > $now)
) AND
(
type IS NULL
OR OR
type NOT IN ( expiresAt > $now
'change-number-notification',
'group-v1-migration',
'message-history-unsynced',
'profile-change',
'story',
'universal-timer-notification',
'verified-change'
)
) AND NOT
(
type IS 'group-v2-change' AND
json_extract(json, '$.groupV2Change.from') IS NOT $ourUuid AND
json_extract(json, '$.groupV2Change.details.length') IS 1 AND
json_extract(json, '$.groupV2Change.details[0].type') IS 'member-remove' AND
json_extract(json, '$.groupV2Change.details[0].uuid') IS NOT $ourUuid
) )
ORDER BY received_at DESC, sent_at DESC ORDER BY received_at DESC, sent_at DESC
LIMIT 1; LIMIT 1;

View File

@ -11,7 +11,7 @@ import { createOrUpdate, getById, removeById } from '../util';
import type { EmptyQuery, Query } from '../util'; import type { EmptyQuery, Query } from '../util';
import type { ItemKeyType } from '../Interface'; import type { ItemKeyType } from '../Interface';
function getOurUuid(db: Database): string | undefined { export function getOurUuid(db: Database): string | undefined {
const UUID_ID: ItemKeyType = 'uuid_id'; const UUID_ID: ItemKeyType = 'uuid_id';
const row: { json: string } | undefined = db const row: { json: string } | undefined = db

View File

@ -0,0 +1,141 @@
// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import type { Database } from 'better-sqlite3';
import type { LoggerType } from '../../types/Logging';
import { getOurUuid } from './41-uuid-keys';
import type { Query } from '../util';
export default function updateToSchemaVersion47(
currentVersion: number,
db: Database,
logger: LoggerType
): void {
if (currentVersion >= 47) {
return;
}
db.transaction(() => {
db.exec(
`
DROP INDEX messages_conversation;
ALTER TABLE messages
DROP COLUMN isStory;
ALTER TABLE messages
ADD COLUMN isStory INTEGER
GENERATED ALWAYS AS (type IS 'story');
ALTER TABLE messages
ADD COLUMN isChangeCreatedByUs INTEGER NOT NULL DEFAULT 0;
ALTER TABLE messages
ADD COLUMN shouldAffectActivity INTEGER
GENERATED ALWAYS AS (
type IS NULL
OR
type NOT IN (
'change-number-notification',
'group-v1-migration',
'message-history-unsynced',
'profile-change',
'story',
'universal-timer-notification',
'verified-change',
'keychange'
)
);
ALTER TABLE messages
ADD COLUMN shouldAffectPreview INTEGER
GENERATED ALWAYS AS (
type IS NULL
OR
type NOT IN (
'change-number-notification',
'group-v1-migration',
'message-history-unsynced',
'profile-change',
'story',
'universal-timer-notification',
'verified-change'
)
);
ALTER TABLE messages
ADD COLUMN isUserInitiatedMessage INTEGER
GENERATED ALWAYS AS (
type IS NULL
OR
type NOT IN (
'change-number-notification',
'group-v1-migration',
'message-history-unsynced',
'profile-change',
'story',
'universal-timer-notification',
'verified-change',
'group-v2-change',
'keychange'
)
);
ALTER TABLE messages
ADD COLUMN isTimerChangeFromSync INTEGER
GENERATED ALWAYS AS (
json_extract(json, '$.expirationTimerUpdate.fromSync') IS 1
);
ALTER TABLE messages
ADD COLUMN isGroupLeaveEvent INTEGER
GENERATED ALWAYS AS (
type IS 'group-v2-change' AND
json_array_length(json_extract(json, '$.groupV2Change.details')) IS 1 AND
json_extract(json, '$.groupV2Change.details[0].type') IS 'member-remove' AND
json_extract(json, '$.groupV2Change.from') IS NOT NULL AND
json_extract(json, '$.groupV2Change.from') IS json_extract(json, '$.groupV2Change.details[0].uuid')
);
ALTER TABLE messages
ADD COLUMN isGroupLeaveEventFromOther INTEGER
GENERATED ALWAYS AS (
isGroupLeaveEvent IS 1
AND
isChangeCreatedByUs IS 0
);
CREATE INDEX messages_conversation ON messages
(conversationId, isStory, storyId, received_at, sent_at);
CREATE INDEX messages_preview ON messages
(conversationId, shouldAffectPreview, isGroupLeaveEventFromOther, expiresAt, received_at, sent_at);
CREATE INDEX messages_activity ON messages
(conversationId, shouldAffectActivity, isTimerChangeFromSync, isGroupLeaveEventFromOther, received_at, sent_at);
CREATE INDEX message_user_initiated ON messages (isUserInitiatedMessage);
`
);
const ourUuid = getOurUuid(db);
if (!ourUuid) {
logger.warn('updateToSchemaVersion47: our UUID not found');
} else {
db.prepare<Query>(
`
UPDATE messages SET
isChangeCreatedByUs = json_extract(json, '$.groupV2Change.from') IS $ourUuid;
`
).run({
ourUuid,
});
}
db.pragma('user_version = 47');
})();
logger.info('updateToSchemaVersion47: success!');
}

View File

@ -22,6 +22,7 @@ import updateToSchemaVersion43 from './43-gv2-uuid';
import updateToSchemaVersion44 from './44-badges'; import updateToSchemaVersion44 from './44-badges';
import updateToSchemaVersion45 from './45-stories'; import updateToSchemaVersion45 from './45-stories';
import updateToSchemaVersion46 from './46-optimize-stories'; import updateToSchemaVersion46 from './46-optimize-stories';
import updateToSchemaVersion47 from './47-further-optimize';
function updateToSchemaVersion1( function updateToSchemaVersion1(
currentVersion: number, currentVersion: number,
@ -1907,6 +1908,7 @@ export const SCHEMA_VERSIONS = [
updateToSchemaVersion44, updateToSchemaVersion44,
updateToSchemaVersion45, updateToSchemaVersion45,
updateToSchemaVersion46, updateToSchemaVersion46,
updateToSchemaVersion47,
]; ];
export function updateSchema(db: Database, logger: LoggerType): void { export function updateSchema(db: Database, logger: LoggerType): void {

View File

@ -1333,7 +1333,8 @@ function cancelMessagesPendingConversationVerification(): ThunkAction<
}); });
await window.Signal.Data.saveMessages( await window.Signal.Data.saveMessages(
messagesStopped.map(message => message.attributes) messagesStopped.map(message => message.attributes),
{ ourUuid: window.textsecure.storage.user.getCheckedUuid().toString() }
); );
}; };
} }

View File

@ -75,6 +75,7 @@ describe('Conversations', () => {
// Saving to db and updating the convo's last message // Saving to db and updating the convo's last message
await window.Signal.Data.saveMessage(message.attributes, { await window.Signal.Data.saveMessage(message.attributes, {
forceSave: true, forceSave: true,
ourUuid,
}); });
message = window.MessageController.register(message.id, message); message = window.MessageController.register(message.id, message);
await window.Signal.Data.updateConversation(conversation.attributes); await window.Signal.Data.updateConversation(conversation.attributes);

View File

@ -32,6 +32,7 @@ describe('sql/allMedia', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -62,7 +63,10 @@ describe('sql/allMedia', () => {
hasVisualMediaAttachments: true, hasVisualMediaAttachments: true,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -79,6 +83,7 @@ describe('sql/allMedia', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -112,7 +117,10 @@ describe('sql/allMedia', () => {
hasVisualMediaAttachments: true, hasVisualMediaAttachments: true,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -131,6 +139,7 @@ describe('sql/allMedia', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -161,7 +170,10 @@ describe('sql/allMedia', () => {
hasFileAttachments: true, hasFileAttachments: true,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -178,6 +190,7 @@ describe('sql/allMedia', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -211,7 +224,10 @@ describe('sql/allMedia', () => {
hasFileAttachments: true, hasFileAttachments: true,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);

View File

@ -0,0 +1,548 @@
// Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
import { assert } from 'chai';
import dataInterface from '../../sql/Client';
import { UUID } from '../../types/UUID';
import type { UUIDStringType } from '../../types/UUID';
import type { MessageAttributesType } from '../../model-types.d';
const {
removeAll,
_getAllMessages,
saveMessages,
getLastConversationMessages,
} = dataInterface;
function getUuid(): UUIDStringType {
return UUID.generate().toString();
}
describe('sql/conversationSummary', () => {
beforeEach(async () => {
await removeAll();
});
describe('getLastConversationMessages', () => {
it('returns the latest message in current conversation', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
type: 'outgoing',
conversationId,
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'outgoing',
conversationId,
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
const message3: MessageAttributesType = {
id: getUuid(),
body: 'message 3',
type: 'outgoing',
conversationId: getUuid(),
sent_at: now + 3,
received_at: now + 3,
timestamp: now + 3,
};
await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.activity?.body, message2.body, 'activity');
assert.strictEqual(messages.preview?.body, message2.body, 'preview');
assert.isTrue(messages.hasUserInitiatedMessages);
});
it('preview excludes several message types, allows type = NULL', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
// @ts-expect-error We're forcing a null type here for testing
type: null,
conversationId,
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'change-number-notification',
conversationId,
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
const message3: MessageAttributesType = {
id: getUuid(),
body: 'message 3',
type: 'group-v1-migration',
conversationId,
sent_at: now + 3,
received_at: now + 3,
timestamp: now + 3,
};
const message4: MessageAttributesType = {
id: getUuid(),
body: 'message 4',
type: 'message-history-unsynced',
conversationId,
sent_at: now + 4,
received_at: now + 4,
timestamp: now + 4,
};
const message5: MessageAttributesType = {
id: getUuid(),
body: 'message 5',
type: 'profile-change',
conversationId,
sent_at: now + 5,
received_at: now + 5,
timestamp: now + 5,
};
const message6: MessageAttributesType = {
id: getUuid(),
body: 'message 6',
type: 'story',
conversationId,
sent_at: now + 6,
received_at: now + 6,
timestamp: now + 6,
};
const message7: MessageAttributesType = {
id: getUuid(),
body: 'message 7',
type: 'universal-timer-notification',
conversationId,
sent_at: now + 7,
received_at: now + 7,
timestamp: now + 7,
};
const message8: MessageAttributesType = {
id: getUuid(),
body: 'message 8',
type: 'verified-change',
conversationId,
sent_at: now + 8,
received_at: now + 8,
timestamp: now + 8,
};
await saveMessages(
[
message1,
message2,
message3,
message4,
message5,
message6,
message7,
message8,
],
{
forceSave: true,
ourUuid,
}
);
assert.lengthOf(await _getAllMessages(), 8);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.preview?.body, message1.body);
});
it('activity excludes several message types, allows type = NULL', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
// @ts-expect-error We're forcing a null type here for testing
type: null,
conversationId,
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'change-number-notification',
conversationId,
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
const message3: MessageAttributesType = {
id: getUuid(),
body: 'message 3',
type: 'group-v1-migration',
conversationId,
sent_at: now + 3,
received_at: now + 3,
timestamp: now + 3,
};
const message4: MessageAttributesType = {
id: getUuid(),
body: 'message 4',
type: 'keychange',
conversationId,
sent_at: now + 4,
received_at: now + 4,
timestamp: now + 4,
};
const message5: MessageAttributesType = {
id: getUuid(),
body: 'message 5',
type: 'message-history-unsynced',
conversationId,
sent_at: now + 5,
received_at: now + 5,
timestamp: now + 5,
};
const message6: MessageAttributesType = {
id: getUuid(),
body: 'message 6',
type: 'profile-change',
conversationId,
sent_at: now + 6,
received_at: now + 6,
timestamp: now + 6,
};
const message7: MessageAttributesType = {
id: getUuid(),
body: 'message 7',
type: 'story',
conversationId,
sent_at: now + 7,
received_at: now + 7,
timestamp: now + 7,
};
const message8: MessageAttributesType = {
id: getUuid(),
body: 'message 8',
type: 'universal-timer-notification',
conversationId,
sent_at: now + 8,
received_at: now + 8,
timestamp: now + 8,
};
const message9: MessageAttributesType = {
id: getUuid(),
body: 'message 9',
type: 'verified-change',
conversationId,
sent_at: now + 9,
received_at: now + 9,
timestamp: now + 9,
};
await saveMessages(
[
message1,
message2,
message3,
message4,
message5,
message6,
message7,
message8,
message9,
],
{
forceSave: true,
ourUuid,
}
);
assert.lengthOf(await _getAllMessages(), 9);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.activity?.body, message1.body);
});
it('activity excludes expirationTimerUpdates with fromSync = true, includes fromSync = undefined', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
type: 'outgoing',
conversationId,
expirationTimerUpdate: {
expireTimer: 10,
source: 'you',
},
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'outgoing',
conversationId,
expirationTimerUpdate: {
expireTimer: 10,
fromSync: true,
},
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
await saveMessages([message1, message2], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 2);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.activity?.body, message1.body);
});
it('activity excludes expirationTimerUpdates with fromSync = true, includes fromSync = false', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
type: 'outgoing',
conversationId,
expirationTimerUpdate: {
expireTimer: 10,
source: 'you',
fromSync: false,
},
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'outgoing',
conversationId,
expirationTimerUpdate: {
expireTimer: 10,
fromSync: true,
},
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
await saveMessages([message1, message2], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 2);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.activity?.body, message1.body);
});
it('preview excludes expired message, includes non-disappearing message', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
type: 'outgoing',
conversationId,
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'outgoing',
conversationId,
expirationStartTimestamp: now - 2 * 1000,
expireTimer: 1,
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
await saveMessages([message1, message2], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 2);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.preview?.body, message1.body);
});
it('preview excludes expired message, includes non-disappearing message', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1',
type: 'outgoing',
conversationId,
expirationStartTimestamp: now,
expireTimer: 30,
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2',
type: 'outgoing',
conversationId,
expirationStartTimestamp: now - 2 * 1000,
expireTimer: 1,
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
await saveMessages([message1, message2], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 2);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.preview?.body, message1.body);
});
it('excludes group v2 change events where someone else leaves a group', async () => {
assert.lengthOf(await _getAllMessages(), 0);
const now = Date.now();
const conversationId = getUuid();
const otherUuid = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = {
id: getUuid(),
body: 'message 1 - removing ourselves',
type: 'group-v2-change',
conversationId,
groupV2Change: {
from: ourUuid,
details: [
{
type: 'member-remove',
uuid: ourUuid,
},
],
},
sent_at: now + 1,
received_at: now + 1,
timestamp: now + 1,
};
const message2: MessageAttributesType = {
id: getUuid(),
body: 'message 2 - someone else leaving',
type: 'group-v2-change',
conversationId,
groupV2Change: {
from: otherUuid,
details: [
{
type: 'member-remove',
uuid: otherUuid,
},
],
},
sent_at: now + 2,
received_at: now + 2,
timestamp: now + 2,
};
await saveMessages([message1, message2], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 2);
const messages = await getLastConversationMessages({
conversationId,
ourUuid,
});
assert.strictEqual(messages.activity?.body, message1.body, 'activity');
assert.strictEqual(messages.preview?.body, message1.body, 'preview');
assert.isFalse(messages.hasUserInitiatedMessages);
});
});
});

View File

@ -31,6 +31,7 @@ describe('sql/fullTextSearch', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1 - generic string', body: 'message 1 - generic string',
@ -59,7 +60,10 @@ describe('sql/fullTextSearch', () => {
timestamp: now, timestamp: now,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -68,7 +72,7 @@ describe('sql/fullTextSearch', () => {
assert.strictEqual(searchResults[0].id, message2.id); assert.strictEqual(searchResults[0].id, message2.id);
message3.body = 'message 3 - unique string'; message3.body = 'message 3 - unique string';
await saveMessage(message3); await saveMessage(message3, { ourUuid });
const searchResults2 = await searchMessages('unique'); const searchResults2 = await searchMessages('unique');
assert.lengthOf(searchResults2, 2); assert.lengthOf(searchResults2, 2);
@ -81,6 +85,7 @@ describe('sql/fullTextSearch', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1 - unique string', body: 'message 1 - unique string',
@ -111,7 +116,10 @@ describe('sql/fullTextSearch', () => {
isViewOnce: true, isViewOnce: true,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -120,7 +128,7 @@ describe('sql/fullTextSearch', () => {
assert.strictEqual(searchResults[0].id, message1.id); assert.strictEqual(searchResults[0].id, message1.id);
message1.body = 'message 3 - unique string'; message1.body = 'message 3 - unique string';
await saveMessage(message3); await saveMessage(message3, { ourUuid });
const searchResults2 = await searchMessages('unique'); const searchResults2 = await searchMessages('unique');
assert.lengthOf(searchResults2, 1); assert.lengthOf(searchResults2, 1);
@ -132,6 +140,7 @@ describe('sql/fullTextSearch', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1 - unique string', body: 'message 1 - unique string',
@ -162,7 +171,10 @@ describe('sql/fullTextSearch', () => {
storyId: getUuid(), storyId: getUuid(),
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -171,7 +183,7 @@ describe('sql/fullTextSearch', () => {
assert.strictEqual(searchResults[0].id, message1.id); assert.strictEqual(searchResults[0].id, message1.id);
message1.body = 'message 3 - unique string'; message1.body = 'message 3 - unique string';
await saveMessage(message3); await saveMessage(message3, { ourUuid });
const searchResults2 = await searchMessages('unique'); const searchResults2 = await searchMessages('unique');
assert.lengthOf(searchResults2, 1); assert.lengthOf(searchResults2, 1);

View File

@ -39,6 +39,7 @@ describe('sql/markRead', () => {
const start = Date.now(); const start = Date.now();
const readAt = start + 20; const readAt = start + 20;
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -117,6 +118,7 @@ describe('sql/markRead', () => {
[message1, message2, message3, message4, message5, message6, message7], [message1, message2, message3, message4, message5, message6, message7],
{ {
forceSave: true, forceSave: true,
ourUuid,
} }
); );
@ -175,6 +177,7 @@ describe('sql/markRead', () => {
const readAt = start + 20; const readAt = start + 20;
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -258,6 +261,7 @@ describe('sql/markRead', () => {
[message1, message2, message3, message4, message5, message6, message7], [message1, message2, message3, message4, message5, message6, message7],
{ {
forceSave: true, forceSave: true,
ourUuid,
} }
); );
@ -297,6 +301,7 @@ describe('sql/markRead', () => {
const readAt = start + 20; const readAt = start + 20;
const conversationId = getUuid(); const conversationId = getUuid();
const expireTimer = 15; const expireTimer = 15;
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -356,6 +361,7 @@ describe('sql/markRead', () => {
await saveMessages([message1, message2, message3, message4, message5], { await saveMessages([message1, message2, message3, message4, message5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.strictEqual( assert.strictEqual(
@ -409,6 +415,7 @@ describe('sql/markRead', () => {
const start = Date.now(); const start = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -459,6 +466,7 @@ describe('sql/markRead', () => {
await saveMessages([message1, message2, message3, message4, message5], { await saveMessages([message1, message2, message3, message4, message5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.lengthOf(await _getAllMessages(), 5); assert.lengthOf(await _getAllMessages(), 5);
@ -553,6 +561,7 @@ describe('sql/markRead', () => {
const start = Date.now(); const start = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -606,6 +615,7 @@ describe('sql/markRead', () => {
await saveMessages([message1, message2, message3, message4, message5], { await saveMessages([message1, message2, message3, message4, message5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.lengthOf(await _getAllMessages(), 5); assert.lengthOf(await _getAllMessages(), 5);

View File

@ -1,13 +1,17 @@
// Copyright 2021 Signal Messenger, LLC // Copyright 2021 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only // SPDX-License-Identifier: AGPL-3.0-only
import { v4 as getGuid } from 'uuid';
import { assert } from 'chai'; import { assert } from 'chai';
import dataInterface from '../../sql/Client'; import dataInterface from '../../sql/Client';
import { UUID } from '../../types/UUID';
import type { UUIDStringType } from '../../types/UUID';
import { constantTimeEqual, getRandomBytes } from '../../Crypto'; import { constantTimeEqual, getRandomBytes } from '../../Crypto';
function getUuid(): UUIDStringType {
return UUID.generate().toString();
}
const { const {
_getAllSentProtoMessageIds, _getAllSentProtoMessageIds,
_getAllSentProtoRecipients, _getAllSentProtoRecipients,
@ -37,9 +41,9 @@ describe('sql/sendLog', () => {
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[getGuid()]: [1, 2], [getUuid()]: [1, 2],
}, },
}); });
const allProtos = await getAllSentProtos(); const allProtos = await getAllSentProtos();
@ -69,10 +73,10 @@ describe('sql/sendLog', () => {
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid(), getGuid()], messageIds: [getUuid(), getUuid()],
recipients: { recipients: {
[getGuid()]: [1, 2], [getUuid()]: [1, 2],
[getGuid()]: [1], [getUuid()]: [1],
}, },
}); });
@ -88,21 +92,22 @@ describe('sql/sendLog', () => {
}); });
it('trigger deletes payload when referenced message is deleted', async () => { it('trigger deletes payload when referenced message is deleted', async () => {
const id = getGuid(); const id = getUuid();
const timestamp = Date.now(); const timestamp = Date.now();
const ourUuid = getUuid();
await saveMessage( await saveMessage(
{ {
id, id,
body: 'some text', body: 'some text',
conversationId: getGuid(), conversationId: getUuid(),
received_at: timestamp, received_at: timestamp,
sent_at: timestamp, sent_at: timestamp,
timestamp, timestamp,
type: 'outgoing', type: 'outgoing',
}, },
{ forceSave: true } { forceSave: true, ourUuid }
); );
const bytes = getRandomBytes(128); const bytes = getRandomBytes(128);
@ -114,7 +119,7 @@ describe('sql/sendLog', () => {
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [id], messageIds: [id],
recipients: { recipients: {
[getGuid()]: [1, 2], [getUuid()]: [1, 2],
}, },
}); });
const allProtos = await getAllSentProtos(); const allProtos = await getAllSentProtos();
@ -133,9 +138,9 @@ describe('sql/sendLog', () => {
it('supports adding duplicates', async () => { it('supports adding duplicates', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const messageIds = [getGuid()]; const messageIds = [getUuid()];
const recipients = { const recipients = {
[getGuid()]: [1], [getUuid()]: [1],
}; };
const proto1 = { const proto1 = {
contentHint: 7, contentHint: 7,
@ -170,7 +175,7 @@ describe('sql/sendLog', () => {
it('handles duplicates, adding new recipients if needed', async () => { it('handles duplicates, adding new recipients if needed', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const messageIds = [getGuid()]; const messageIds = [getUuid()];
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
@ -184,7 +189,7 @@ describe('sql/sendLog', () => {
const id = await insertSentProto(proto, { const id = await insertSentProto(proto, {
messageIds, messageIds,
recipients: { recipients: {
[getGuid()]: [1], [getUuid()]: [1],
}, },
}); });
@ -192,7 +197,7 @@ describe('sql/sendLog', () => {
assert.lengthOf(await _getAllSentProtoMessageIds(), 1); assert.lengthOf(await _getAllSentProtoMessageIds(), 1);
assert.lengthOf(await _getAllSentProtoRecipients(), 1); assert.lengthOf(await _getAllSentProtoRecipients(), 1);
const recipientUuid = getGuid(); const recipientUuid = getUuid();
await insertProtoRecipients({ await insertProtoRecipients({
id, id,
recipientUuid, recipientUuid,
@ -225,21 +230,21 @@ describe('sql/sendLog', () => {
timestamp: timestamp - 15, timestamp: timestamp - 15,
}; };
await insertSentProto(proto1, { await insertSentProto(proto1, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[getGuid()]: [1], [getUuid()]: [1],
}, },
}); });
await insertSentProto(proto2, { await insertSentProto(proto2, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[getGuid()]: [1, 2], [getUuid()]: [1, 2],
}, },
}); });
await insertSentProto(proto3, { await insertSentProto(proto3, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[getGuid()]: [1, 2, 3], [getUuid()]: [1, 2, 3],
}, },
}); });
@ -268,7 +273,7 @@ describe('sql/sendLog', () => {
assert.lengthOf(await _getAllSentProtoMessageIds(), 0); assert.lengthOf(await _getAllSentProtoMessageIds(), 0);
assert.lengthOf(await _getAllSentProtoRecipients(), 0); assert.lengthOf(await _getAllSentProtoRecipients(), 0);
const messageId = getGuid(); const messageId = getUuid();
const timestamp = Date.now(); const timestamp = Date.now();
const proto1 = { const proto1 = {
contentHint: 1, contentHint: 1,
@ -286,22 +291,22 @@ describe('sql/sendLog', () => {
timestamp: timestamp - 20, timestamp: timestamp - 20,
}; };
await insertSentProto(proto1, { await insertSentProto(proto1, {
messageIds: [messageId, getGuid()], messageIds: [messageId, getUuid()],
recipients: { recipients: {
[getGuid()]: [1, 2], [getUuid()]: [1, 2],
[getGuid()]: [1], [getUuid()]: [1],
}, },
}); });
await insertSentProto(proto2, { await insertSentProto(proto2, {
messageIds: [messageId], messageIds: [messageId],
recipients: { recipients: {
[getGuid()]: [1], [getUuid()]: [1],
}, },
}); });
await insertSentProto(proto3, { await insertSentProto(proto3, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[getGuid()]: [1], [getUuid()]: [1],
}, },
}); });
@ -321,15 +326,15 @@ describe('sql/sendLog', () => {
it('does not delete payload if recipient remains', async () => { it('does not delete payload if recipient remains', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid1 = getGuid(); const recipientUuid1 = getUuid();
const recipientUuid2 = getGuid(); const recipientUuid2 = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[recipientUuid1]: [1, 2], [recipientUuid1]: [1, 2],
[recipientUuid2]: [1], [recipientUuid2]: [1],
@ -352,15 +357,15 @@ describe('sql/sendLog', () => {
it('deletes payload if no recipients remain', async () => { it('deletes payload if no recipients remain', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid1 = getGuid(); const recipientUuid1 = getUuid();
const recipientUuid2 = getGuid(); const recipientUuid2 = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[recipientUuid1]: [1, 2], [recipientUuid1]: [1, 2],
[recipientUuid2]: [1], [recipientUuid2]: [1],
@ -401,15 +406,15 @@ describe('sql/sendLog', () => {
it('deletes multiple recipients in a single transaction', async () => { it('deletes multiple recipients in a single transaction', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid1 = getGuid(); const recipientUuid1 = getUuid();
const recipientUuid2 = getGuid(); const recipientUuid2 = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[recipientUuid1]: [1, 2], [recipientUuid1]: [1, 2],
[recipientUuid2]: [1], [recipientUuid2]: [1],
@ -446,8 +451,8 @@ describe('sql/sendLog', () => {
it('returns matching payload', async () => { it('returns matching payload', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid = getGuid(); const recipientUuid = getUuid();
const messageIds = [getGuid(), getGuid()]; const messageIds = [getUuid(), getUuid()];
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
@ -482,7 +487,7 @@ describe('sql/sendLog', () => {
it('returns matching payload with no messageIds', async () => { it('returns matching payload with no messageIds', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid = getGuid(); const recipientUuid = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
@ -517,14 +522,14 @@ describe('sql/sendLog', () => {
it('returns nothing if payload does not have recipient', async () => { it('returns nothing if payload does not have recipient', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid = getGuid(); const recipientUuid = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[recipientUuid]: [1, 2], [recipientUuid]: [1, 2],
}, },
@ -536,7 +541,7 @@ describe('sql/sendLog', () => {
const actual = await getSentProtoByRecipient({ const actual = await getSentProtoByRecipient({
now: timestamp, now: timestamp,
timestamp, timestamp,
recipientUuid: getGuid(), recipientUuid: getUuid(),
}); });
assert.isUndefined(actual); assert.isUndefined(actual);
@ -545,14 +550,14 @@ describe('sql/sendLog', () => {
it('returns nothing if timestamp does not match', async () => { it('returns nothing if timestamp does not match', async () => {
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid = getGuid(); const recipientUuid = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[recipientUuid]: [1, 2], [recipientUuid]: [1, 2],
}, },
@ -574,14 +579,14 @@ describe('sql/sendLog', () => {
const TWO_DAYS = 2 * 24 * 60 * 60 * 1000; const TWO_DAYS = 2 * 24 * 60 * 60 * 1000;
const timestamp = Date.now(); const timestamp = Date.now();
const recipientUuid = getGuid(); const recipientUuid = getUuid();
const proto = { const proto = {
contentHint: 1, contentHint: 1,
proto: getRandomBytes(128), proto: getRandomBytes(128),
timestamp, timestamp,
}; };
await insertSentProto(proto, { await insertSentProto(proto, {
messageIds: [getGuid()], messageIds: [getUuid()],
recipients: { recipients: {
[recipientUuid]: [1, 2], [recipientUuid]: [1, 2],
}, },

View File

@ -28,6 +28,7 @@ describe('sql/stories', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const sourceUuid = getUuid(); const sourceUuid = getUuid();
const ourUuid = getUuid();
const story1: MessageAttributesType = { const story1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -82,6 +83,7 @@ describe('sql/stories', () => {
await saveMessages([story1, story2, story3, story4, story5], { await saveMessages([story1, story2, story3, story4, story5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.lengthOf(await _getAllMessages(), 5); assert.lengthOf(await _getAllMessages(), 5);
@ -149,6 +151,8 @@ describe('sql/stories', () => {
const start = Date.now(); const start = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const story1: MessageAttributesType = { const story1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -197,6 +201,7 @@ describe('sql/stories', () => {
await saveMessages([story1, story2, story3, story4, story5], { await saveMessages([story1, story2, story3, story4, story5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.lengthOf(await _getAllMessages(), 5); assert.lengthOf(await _getAllMessages(), 5);

View File

@ -35,6 +35,8 @@ describe('sql/timelineFetches', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -85,6 +87,7 @@ describe('sql/timelineFetches', () => {
await saveMessages([message1, message2, message3, message4, message5], { await saveMessages([message1, message2, message3, message4, message5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.lengthOf(await _getAllMessages(), 5); assert.lengthOf(await _getAllMessages(), 5);
@ -105,6 +108,8 @@ describe('sql/timelineFetches', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'story', body: 'story',
@ -135,7 +140,10 @@ describe('sql/timelineFetches', () => {
timestamp: now, timestamp: now,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -152,6 +160,8 @@ describe('sql/timelineFetches', () => {
const target = Date.now(); const target = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -180,7 +190,10 @@ describe('sql/timelineFetches', () => {
timestamp: target + 10, timestamp: target + 10,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -198,6 +211,8 @@ describe('sql/timelineFetches', () => {
const target = Date.now(); const target = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -226,7 +241,10 @@ describe('sql/timelineFetches', () => {
timestamp: target, timestamp: target,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -248,6 +266,8 @@ describe('sql/timelineFetches', () => {
const target = Date.now(); const target = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -276,7 +296,10 @@ describe('sql/timelineFetches', () => {
timestamp: target, timestamp: target,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -299,6 +322,8 @@ describe('sql/timelineFetches', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -349,6 +374,7 @@ describe('sql/timelineFetches', () => {
await saveMessages([message1, message2, message3, message4, message5], { await saveMessages([message1, message2, message3, message4, message5], {
forceSave: true, forceSave: true,
ourUuid,
}); });
assert.lengthOf(await _getAllMessages(), 5); assert.lengthOf(await _getAllMessages(), 5);
@ -368,6 +394,8 @@ describe('sql/timelineFetches', () => {
const now = Date.now(); const now = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -398,7 +426,10 @@ describe('sql/timelineFetches', () => {
timestamp: now + 20, timestamp: now + 20,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -416,6 +447,8 @@ describe('sql/timelineFetches', () => {
const target = Date.now(); const target = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -444,7 +477,10 @@ describe('sql/timelineFetches', () => {
timestamp: target + 10, timestamp: target + 10,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -462,6 +498,8 @@ describe('sql/timelineFetches', () => {
const target = Date.now(); const target = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const ourUuid = getUuid();
const message1: MessageAttributesType = { const message1: MessageAttributesType = {
id: getUuid(), id: getUuid(),
body: 'message 1', body: 'message 1',
@ -490,7 +528,10 @@ describe('sql/timelineFetches', () => {
timestamp: target, timestamp: target,
}; };
await saveMessages([message1, message2, message3], { forceSave: true }); await saveMessages([message1, message2, message3], {
forceSave: true,
ourUuid,
});
assert.lengthOf(await _getAllMessages(), 3); assert.lengthOf(await _getAllMessages(), 3);
@ -514,6 +555,7 @@ describe('sql/timelineFetches', () => {
const target = Date.now(); const target = Date.now();
const conversationId = getUuid(); const conversationId = getUuid();
const storyId = getUuid(); const storyId = getUuid();
const ourUuid = getUuid();
const story: MessageAttributesType = { const story: MessageAttributesType = {
id: getUuid(), id: getUuid(),
@ -605,7 +647,7 @@ describe('sql/timelineFetches', () => {
newestInStory, newestInStory,
newest, newest,
], ],
{ forceSave: true } { forceSave: true, ourUuid }
); );
assert.lengthOf(await _getAllMessages(), 8); assert.lengthOf(await _getAllMessages(), 8);

View File

@ -910,35 +910,110 @@ describe('SQL migrations test', () => {
}); });
}); });
describe('updateToSchemaVersion46', () => { describe('updateToSchemaVersion47', () => {
it('creates new auto-generated isStory field', () => { it('creates and pre-populates new isChangeCreatedByUs field', () => {
const STORY_ID_1 = generateGuid(); const OTHER_UUID = generateGuid();
const MESSAGE_ID_1 = generateGuid(); const MESSAGE_ID_1 = generateGuid();
const MESSAGE_ID_2 = generateGuid(); const MESSAGE_ID_2 = generateGuid();
const CONVERSATION_ID = generateGuid(); const CONVERSATION_ID = generateGuid();
updateToVersion(46); updateToVersion(46);
const uuidItem = JSON.stringify({
value: `${OUR_UUID}.4`,
});
const changeFromUs = JSON.stringify({
groupV2Change: {
from: OUR_UUID,
details: [
{
type: 'member-remove',
uuid: OTHER_UUID,
},
],
},
});
const changeFromOther = JSON.stringify({
groupV2Change: {
from: OTHER_UUID,
details: [
{
type: 'member-remove',
uuid: OUR_UUID,
},
],
},
});
db.exec( db.exec(
` `
INSERT INTO items (id, json) VALUES ('uuid_id', '${uuidItem}');
INSERT INTO messages INSERT INTO messages
(id, storyId, conversationId, type, body) (id, conversationId, type, json)
VALUES VALUES
('${MESSAGE_ID_1}', '${STORY_ID_1}', '${CONVERSATION_ID}', 'story', 'story 1'), ('${MESSAGE_ID_1}', '${CONVERSATION_ID}', 'outgoing', '${changeFromUs}'),
('${MESSAGE_ID_2}', null, '${CONVERSATION_ID}', 'outgoing', 'reply to story 1'); ('${MESSAGE_ID_2}', '${CONVERSATION_ID}', 'outgoing', '${changeFromOther}');
` `
); );
updateToVersion(47);
assert.strictEqual( assert.strictEqual(
db.prepare('SELECT COUNT(*) FROM messages;').pluck().get(), db.prepare('SELECT COUNT(*) FROM messages;').pluck().get(),
2 2
); );
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isChangeCreatedByUs IS 0;'
)
.pluck()
.get(),
1,
'zero'
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isChangeCreatedByUs IS 1;'
)
.pluck()
.get(),
1,
'one'
);
});
it('creates new auto-generated isStory field', () => {
const STORY_ID_1 = generateGuid();
const MESSAGE_ID_1 = generateGuid();
const MESSAGE_ID_2 = generateGuid();
const MESSAGE_ID_3 = generateGuid();
const CONVERSATION_ID = generateGuid();
updateToVersion(47);
db.exec(
`
INSERT INTO messages
(id, storyId, conversationId, type, body)
VALUES
('${MESSAGE_ID_1}', '${STORY_ID_1}', '${CONVERSATION_ID}', 'story', 'story 1'),
('${MESSAGE_ID_2}', null, '${CONVERSATION_ID}', 'outgoing', 'reply to story 1'),
('${MESSAGE_ID_3}', null, '${CONVERSATION_ID}', null, 'null type!');
`
);
assert.strictEqual(
db.prepare('SELECT COUNT(*) FROM messages;').pluck().get(),
3
);
assert.strictEqual( assert.strictEqual(
db db
.prepare('SELECT COUNT(*) FROM messages WHERE isStory IS 0;') .prepare('SELECT COUNT(*) FROM messages WHERE isStory IS 0;')
.pluck() .pluck()
.get(), .get(),
1 2
); );
assert.strictEqual( assert.strictEqual(
db db
@ -949,8 +1024,217 @@ describe('SQL migrations test', () => {
); );
}); });
it('creates new auto-generated shouldAffectActivity/shouldAffectPreview/isUserInitiatedMessage fields', () => {
const MESSAGE_ID_1 = generateGuid();
const MESSAGE_ID_2 = generateGuid();
const MESSAGE_ID_3 = generateGuid();
const MESSAGE_ID_4 = generateGuid();
const CONVERSATION_ID = generateGuid();
updateToVersion(47);
db.exec(
`
INSERT INTO messages
(id, conversationId, type)
VALUES
('${MESSAGE_ID_1}', '${CONVERSATION_ID}', 'story'),
('${MESSAGE_ID_2}', '${CONVERSATION_ID}', 'keychange'),
('${MESSAGE_ID_3}', '${CONVERSATION_ID}', 'outgoing'),
('${MESSAGE_ID_4}', '${CONVERSATION_ID}', 'group-v2-change');
`
);
assert.strictEqual(
db.prepare('SELECT COUNT(*) FROM messages;').pluck().get(),
4
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE shouldAffectPreview IS 1;'
)
.pluck()
.get(),
3
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE shouldAffectActivity IS 1;'
)
.pluck()
.get(),
2
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isUserInitiatedMessage IS 1;'
)
.pluck()
.get(),
1
);
});
it('creates new auto-generated isTimerChangeFromSync fields', () => {
const MESSAGE_ID_1 = generateGuid();
const MESSAGE_ID_2 = generateGuid();
const MESSAGE_ID_3 = generateGuid();
const CONVERSATION_ID = generateGuid();
updateToVersion(47);
const timerUpdate = JSON.stringify({
expirationTimerUpdate: {
expireTimer: 30,
fromSync: false,
},
});
const timerUpdateFromSync = JSON.stringify({
expirationTimerUpdate: {
expireTimer: 30,
fromSync: true,
},
});
db.exec(
`
INSERT INTO messages
(id, conversationId, type, json)
VALUES
('${MESSAGE_ID_1}', '${CONVERSATION_ID}', 'outgoing', '${timerUpdate}'),
('${MESSAGE_ID_2}', '${CONVERSATION_ID}', 'outgoing', '${timerUpdateFromSync}'),
('${MESSAGE_ID_3}', '${CONVERSATION_ID}', 'outgoing', '{}');
`
);
assert.strictEqual(
db.prepare('SELECT COUNT(*) FROM messages;').pluck().get(),
3
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isTimerChangeFromSync IS 1;'
)
.pluck()
.get(),
1
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isTimerChangeFromSync IS 0;'
)
.pluck()
.get(),
2
);
});
it('creates new auto-generated isGroupLeaveEvent fields', () => {
const MESSAGE_ID_1 = generateGuid();
const MESSAGE_ID_2 = generateGuid();
const MESSAGE_ID_3 = generateGuid();
const MESSAGE_ID_4 = generateGuid();
const MESSAGE_ID_5 = generateGuid();
const CONVERSATION_ID = generateGuid();
const FIRST_UUID = generateGuid();
const SECOND_UUID = generateGuid();
const THIRD_UUID = generateGuid();
updateToVersion(47);
const memberRemoveByOther = JSON.stringify({
groupV2Change: {
from: FIRST_UUID,
details: [
{
type: 'member-remove',
uuid: SECOND_UUID,
},
],
},
});
const memberLeave = JSON.stringify({
groupV2Change: {
from: FIRST_UUID,
details: [
{
type: 'member-remove',
uuid: FIRST_UUID,
},
],
},
});
const multipleRemoves = JSON.stringify({
groupV2Change: {
from: FIRST_UUID,
details: [
{
type: 'member-remove',
uuid: SECOND_UUID,
},
{
type: 'member-remove',
uuid: THIRD_UUID,
},
],
},
});
const memberAdd = JSON.stringify({
groupV2Change: {
from: FIRST_UUID,
details: [
{
type: 'member-add',
uuid: FIRST_UUID,
},
],
},
});
db.exec(
`
INSERT INTO messages
(id, conversationId, type, json)
VALUES
('${MESSAGE_ID_1}', '${CONVERSATION_ID}', 'outgoing', '${memberLeave}'),
('${MESSAGE_ID_2}', '${CONVERSATION_ID}', 'group-v2-change', '${memberRemoveByOther}'),
('${MESSAGE_ID_3}', '${CONVERSATION_ID}', 'group-v2-change', '${memberLeave}'),
('${MESSAGE_ID_4}', '${CONVERSATION_ID}', 'group-v2-change', '${multipleRemoves}'),
('${MESSAGE_ID_5}', '${CONVERSATION_ID}', 'group-v2-change', '${memberAdd}');
`
);
assert.strictEqual(
db.prepare('SELECT COUNT(*) FROM messages;').pluck().get(),
5
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isGroupLeaveEvent IS 1;'
)
.pluck()
.get(),
1
);
assert.strictEqual(
db
.prepare(
'SELECT COUNT(*) FROM messages WHERE isGroupLeaveEvent IS 0;'
)
.pluck()
.get(),
4
);
});
it('ensures that index is used for getOlderMessagesByConversation', () => { it('ensures that index is used for getOlderMessagesByConversation', () => {
updateToVersion(46); updateToVersion(47);
const { detail } = db const { detail } = db
.prepare( .prepare(

View File

@ -12,7 +12,9 @@ const updateMessageBatcher = createBatcher<MessageAttributesType>({
maxSize: 50, maxSize: 50,
processBatch: async (messageAttrs: Array<MessageAttributesType>) => { processBatch: async (messageAttrs: Array<MessageAttributesType>) => {
log.info('updateMessageBatcher', messageAttrs.length); log.info('updateMessageBatcher', messageAttrs.length);
await window.Signal.Data.saveMessages(messageAttrs); await window.Signal.Data.saveMessages(messageAttrs, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
}, },
}); });
@ -22,7 +24,9 @@ export function queueUpdateMessage(messageAttr: MessageAttributesType): void {
if (shouldBatch) { if (shouldBatch) {
updateMessageBatcher.add(messageAttr); updateMessageBatcher.add(messageAttr);
} else { } else {
window.Signal.Data.saveMessage(messageAttr); window.Signal.Data.saveMessage(messageAttr, {
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
});
} }
} }
@ -38,6 +42,7 @@ export const saveNewMessageBatcher = createWaitBatcher<MessageAttributesType>({
log.info('saveNewMessageBatcher', messageAttrs.length); log.info('saveNewMessageBatcher', messageAttrs.length);
await window.Signal.Data.saveMessages(messageAttrs, { await window.Signal.Data.saveMessages(messageAttrs, {
forceSave: true, forceSave: true,
ourUuid: window.textsecure.storage.user.getCheckedUuid().toString(),
}); });
}, },
}); });

View File

@ -1475,6 +1475,7 @@ export class ConversationView extends window.Backbone.View<ConversationModel> {
const DEFAULT_DOCUMENTS_FETCH_COUNT = 150; const DEFAULT_DOCUMENTS_FETCH_COUNT = 150;
const conversationId = this.model.get('id'); const conversationId = this.model.get('id');
const ourUuid = window.textsecure.storage.user.getCheckedUuid().toString();
const getProps = async () => { const getProps = async () => {
const rawMedia = const rawMedia =
@ -1505,7 +1506,7 @@ export class ConversationView extends window.Backbone.View<ConversationModel> {
// eslint-disable-next-line no-await-in-loop // eslint-disable-next-line no-await-in-loop
rawMedia[i] = await upgradeMessageSchema(message); rawMedia[i] = await upgradeMessageSchema(message);
// eslint-disable-next-line no-await-in-loop // eslint-disable-next-line no-await-in-loop
await window.Signal.Data.saveMessage(rawMedia[i]); await window.Signal.Data.saveMessage(rawMedia[i], { ourUuid });
} }
} }