mirror of
https://github.com/zadam/trilium.git
synced 2025-01-17 04:27:56 +08:00
fix sending sync rows via WebSocket after transaction is committed
This commit is contained in:
parent
c34a9b96e4
commit
f117ae4915
3 changed files with 16 additions and 31 deletions
|
@ -33,8 +33,12 @@ function isEntityEventsDisabled() {
|
|||
return !!namespace.get('disableEntityEvents');
|
||||
}
|
||||
|
||||
function getSyncRows() {
|
||||
return namespace.get('syncRows') || [];
|
||||
function getAndClearSyncRows() {
|
||||
const syncRows = namespace.get('syncRows') || [];
|
||||
|
||||
namespace.set('syncRows', []);
|
||||
|
||||
return syncRows;
|
||||
}
|
||||
|
||||
function addSyncRow(syncRow) {
|
||||
|
@ -68,7 +72,7 @@ module.exports = {
|
|||
disableEntityEvents,
|
||||
isEntityEventsDisabled,
|
||||
reset,
|
||||
getSyncRows,
|
||||
getAndClearSyncRows,
|
||||
addSyncRow,
|
||||
getEntityFromCache,
|
||||
setEntityToCache
|
||||
|
|
|
@ -1,7 +1,6 @@
|
|||
"use strict";
|
||||
|
||||
const log = require('./log');
|
||||
const cls = require('./cls');
|
||||
const Database = require('better-sqlite3');
|
||||
const dataDir = require('./data_dir');
|
||||
|
||||
|
@ -74,21 +73,6 @@ function stmt(sql) {
|
|||
return statementCache[sql];
|
||||
}
|
||||
|
||||
function beginTransaction() {
|
||||
// DEFERRED means that the transaction does not actually start until the database is first accessed.
|
||||
// Internally, the BEGIN DEFERRED statement merely sets a flag on the database connection that turns off
|
||||
// the automatic commit that would normally occur when the last statement finishes.
|
||||
return stmt("BEGIN DEFERRED").run();
|
||||
}
|
||||
|
||||
function commit() {
|
||||
return stmt("COMMIT").run();
|
||||
}
|
||||
|
||||
function rollback() {
|
||||
return stmt("ROLLBACK").run();
|
||||
}
|
||||
|
||||
function getRow(query, params = []) {
|
||||
return wrap(query, s => s.get(params));
|
||||
}
|
||||
|
@ -213,7 +197,9 @@ function wrap(query, func) {
|
|||
function transactional(func) {
|
||||
const ret = dbConnection.transaction(func).deferred();
|
||||
|
||||
require('./ws.js').sendPingToAllClients();
|
||||
if (!dbConnection.inTransaction) { // i.e. transaction was really committed (and not just savepoint released)
|
||||
require('./ws.js').sendTransactionSyncsToAllClients();
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
|
|
@ -7,7 +7,6 @@ const syncMutexService = require('./sync_mutex');
|
|||
const protectedSessionService = require('./protected_session');
|
||||
|
||||
let webSocketServer;
|
||||
let lastAcceptedSyncIds = {};
|
||||
|
||||
function init(httpServer, sessionParser) {
|
||||
webSocketServer = new WebSocket.Server({
|
||||
|
@ -28,8 +27,6 @@ function init(httpServer, sessionParser) {
|
|||
webSocketServer.on('connection', (ws, req) => {
|
||||
ws.id = utils.randomString(10);
|
||||
|
||||
lastAcceptedSyncIds[ws.id] = 0;
|
||||
|
||||
console.log(`websocket client connected`);
|
||||
|
||||
ws.on('message', async messageJson => {
|
||||
|
@ -39,8 +36,6 @@ function init(httpServer, sessionParser) {
|
|||
log.info('JS Error: ' + message.error + '\r\nStack: ' + message.stack);
|
||||
}
|
||||
else if (message.type === 'ping') {
|
||||
lastAcceptedSyncIds[ws.id] = message.lastSyncId;
|
||||
|
||||
await syncMutexService.doExclusively(() => sendPing(ws));
|
||||
}
|
||||
else {
|
||||
|
@ -97,9 +92,7 @@ function fillInAdditionalProperties(sync) {
|
|||
}
|
||||
}
|
||||
|
||||
function sendPing(client) {
|
||||
const syncRows = cls.getSyncRows();
|
||||
|
||||
function sendPing(client, syncRows = []) {
|
||||
for (const sync of syncRows) {
|
||||
try {
|
||||
fillInAdditionalProperties(sync);
|
||||
|
@ -119,10 +112,12 @@ function sendPing(client) {
|
|||
});
|
||||
}
|
||||
|
||||
function sendPingToAllClients() {
|
||||
function sendTransactionSyncsToAllClients() {
|
||||
if (webSocketServer) {
|
||||
const syncRows = cls.getAndClearSyncRows();
|
||||
|
||||
webSocketServer.clients.forEach(function each(client) {
|
||||
sendPing(client);
|
||||
sendPing(client, syncRows);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
@ -140,5 +135,5 @@ module.exports = {
|
|||
sendMessageToAllClients,
|
||||
syncPullInProgress,
|
||||
syncPullFinished,
|
||||
sendPingToAllClients
|
||||
sendTransactionSyncsToAllClients
|
||||
};
|
||||
|
|
Loading…
Reference in a new issue