Review comments, moved swarmNodes database stuff to just be in data.js

pull/157/head
Beaudan 7 years ago
parent 80fa4f866e
commit e735b3ca41

@ -110,6 +110,7 @@ module.exports = {
removeAllSessions, removeAllSessions,
getSwarmNodesByPubkey, getSwarmNodesByPubkey,
saveSwarmNodesForPubKey,
getConversationCount, getConversationCount,
saveConversation, saveConversation,
@ -674,6 +675,16 @@ async function getSwarmNodesByPubkey(pubkey) {
return channels.getSwarmNodesByPubkey(pubkey); return channels.getSwarmNodesByPubkey(pubkey);
} }
async function saveSwarmNodesForPubKey(pubKey, swarmNodes, { Conversation }) {
const conversation = await getConversationById(pubKey, { Conversation });
conversation.set({ swarmNodes });
await updateConversation(
conversation.id,
conversation.attributes,
{ Conversation }
);
}
async function getConversationCount() { async function getConversationCount() {
return channels.getConversationCount(); return channels.getConversationCount();
} }

@ -1,6 +1,6 @@
/* eslint-disable no-await-in-loop */ /* eslint-disable no-await-in-loop */
/* eslint-disable no-loop-func */ /* eslint-disable no-loop-func */
/* global log, dcodeIO, window, callWorker */ /* global log, dcodeIO, window, callWorker, Whisper */
const fetch = require('node-fetch'); const fetch = require('node-fetch');
const _ = require('lodash'); const _ = require('lodash');
@ -105,7 +105,7 @@ class LokiMessageAPI {
throw HTTPError('sendMessage: error response', response.status, result); throw HTTPError('sendMessage: error response', response.status, result);
}; };
let swarmNodes = await window.LokiSnodeAPI.getSwarmNodesByPubkey(pubKey); let swarmNodes = await window.Signal.Data.getSwarmNodesByPubkey(pubKey);
while (successfulRequests < MINIMUM_SUCCESSFUL_REQUESTS) { while (successfulRequests < MINIMUM_SUCCESSFUL_REQUESTS) {
if (!canResolve) { if (!canResolve) {
throw new window.textsecure.DNSResolutionError('Sending messages'); throw new window.textsecure.DNSResolutionError('Sending messages');
@ -123,7 +123,11 @@ class LokiMessageAPI {
new Error('Ran out of swarm nodes to query') new Error('Ran out of swarm nodes to query')
); );
} }
await window.LokiSnodeAPI.saveSwarmNodes(pubKey, swarmNodes); await window.Signal.Data.saveSwarmNodesForPubkey(
pubKey,
swarmNodes,
{ Conversation: Whisper.Conversation }
);
} }
const remainingRequests = const remainingRequests =
MINIMUM_SUCCESSFUL_REQUESTS - completedNodes.length; MINIMUM_SUCCESSFUL_REQUESTS - completedNodes.length;

@ -106,23 +106,6 @@ class LokiSnodeAPI {
return this.ourSwarmNodes; return this.ourSwarmNodes;
} }
async getSwarmNodesByPubkey(pubKey) {
const swarmNodes = await window.Signal.Data.getSwarmNodesByPubkey(pubKey);
return swarmNodes;
}
async saveSwarmNodes(pubKey, swarmNodes) {
const conversation = window.ConversationController.get(pubKey);
conversation.set({ swarmNodes });
await window.Signal.Data.updateConversation(
conversation.id,
conversation.attributes,
{
Conversation: Whisper.Conversation,
}
);
}
async getFreshSwarmNodes(pubKey) { async getFreshSwarmNodes(pubKey) {
if (!(pubKey in this.swarmsPendingReplenish)) { if (!(pubKey in this.swarmsPendingReplenish)) {
this.swarmsPendingReplenish[pubKey] = new Promise(async resolve => { this.swarmsPendingReplenish[pubKey] = new Promise(async resolve => {

@ -12,9 +12,13 @@
const friendKeys = await window.Signal.Data.getPubKeysWithFriendStatus( const friendKeys = await window.Signal.Data.getPubKeysWithFriendStatus(
window.friends.friendRequestStatusEnum.friends window.friends.friendRequestStatusEnum.friends
); );
friendKeys.forEach(pubKey => { await Promise.all(friendKeys.map(async pubKey => {
sendOnlineBroadcastMessage(pubKey); try {
}); await sendOnlineBroadcastMessage(pubKey);
} catch(e) {
log.warn(`Failed to send online broadcast message to ${pubKey}`);
}
}));
} }
async function sendOnlineBroadcastMessage(pubKey) { async function sendOnlineBroadcastMessage(pubKey) {

Loading…
Cancel
Save