server-ts: Fix some issues from self-review

This commit is contained in:
Elian Doran 2024-04-02 23:00:04 +03:00
parent 08f0c01eef
commit b84b27692c
No known key found for this signature in database
9 changed files with 40 additions and 168 deletions

View File

@ -21,7 +21,7 @@ interface AttachmentOpts {
* Becca is a backend cache of all notes, branches, and attributes. * Becca is a backend cache of all notes, branches, and attributes.
* There's a similar frontend cache Froca, and share cache Shaca. * There's a similar frontend cache Froca, and share cache Shaca.
*/ */
class Becca { export default class Becca {
loaded!: boolean; loaded!: boolean;
notes!: Record<string, BNote>; notes!: Record<string, BNote>;
@ -280,4 +280,12 @@ class Becca {
} }
} }
export = Becca; /**
* This interface contains the data that is shared across all the objects of a given derived class of {@link AbstractBeccaEntity}.
* For example, all BAttributes will share their content, but all BBranches will have another set of this data.
*/
export interface ConstructorData<T extends AbstractBeccaEntity<T>> {
primaryKeyName: string;
entityName: string;
hashedProperties: (keyof T)[];
}

View File

@ -1,6 +1,6 @@
"use strict"; "use strict";
import Becca = require("./becca-interface"); import Becca from "./becca-interface";
const becca = new Becca(); const becca = new Becca();

View File

@ -9,7 +9,7 @@ import cls = require('../../services/cls');
import log = require('../../services/log'); import log = require('../../services/log');
import protectedSessionService = require('../../services/protected_session'); import protectedSessionService = require('../../services/protected_session');
import blobService = require('../../services/blob'); import blobService = require('../../services/blob');
import Becca = require('../becca-interface'); import Becca, { ConstructorData } from '../becca-interface';
let becca: Becca | null = null; let becca: Becca | null = null;
@ -18,16 +18,6 @@ interface ContentOpts {
forceFrontendReload?: boolean; forceFrontendReload?: boolean;
} }
/**
* This interface contains the data that is shared across all the objects of a given derived class of {@link AbstractBeccaEntity}.
* For example, all BAttributes will share their content, but all BBranches will have another set of this data.
*/
interface ConstructorData<T extends AbstractBeccaEntity<T>> {
primaryKeyName: string;
entityName: string;
hashedProperties: (keyof T)[];
}
/** /**
* Base class for all backend entities. * Base class for all backend entities.
* *

View File

@ -1,18 +1,24 @@
import AbstractBeccaEntity = require("./abstract_becca_entity");
import { BlobRow } from "./rows"; import { BlobRow } from "./rows";
// TODO: Why this does not extend the abstract becca? // TODO: Why this does not extend the abstract becca?
class BBlob { class BBlob extends AbstractBeccaEntity<BBlob> {
static get entityName() { return "blobs"; } static get entityName() { return "blobs"; }
static get primaryKeyName() { return "blobId"; } static get primaryKeyName() { return "blobId"; }
static get hashedProperties() { return ["blobId", "content"]; } static get hashedProperties() { return ["blobId", "content"]; }
blobId: string; blobId!: string;
content: string | Buffer; content!: string | Buffer;
contentLength: number; contentLength!: number;
dateModified: string; dateModified!: string;
utcDateModified: string; utcDateModified!: string;
constructor(row: BlobRow) { constructor(row: BlobRow) {
super();
this.updateFromRow(row);
}
updateFromRow(row: BlobRow): void {
this.blobId = row.blobId; this.blobId = row.blobId;
this.content = row.content; this.content = row.content;
this.contentLength = row.contentLength; this.contentLength = row.contentLength;
@ -20,6 +26,10 @@ class BBlob {
this.utcDateModified = row.utcDateModified; this.utcDateModified = row.utcDateModified;
} }
init() {
// Nothing to do.
}
getPojo() { getPojo() {
return { return {
blobId: this.blobId, blobId: this.blobId,

View File

@ -11,6 +11,7 @@ import AbstractBeccaEntity = require('./abstract_becca_entity');
class BRecentNote extends AbstractBeccaEntity<BRecentNote> { class BRecentNote extends AbstractBeccaEntity<BRecentNote> {
static get entityName() { return "recent_notes"; } static get entityName() { return "recent_notes"; }
static get primaryKeyName() { return "noteId"; } static get primaryKeyName() { return "noteId"; }
static get hashedProperties() { return ["noteId", "notePath"]; }
noteId!: string; noteId!: string;
notePath!: string; notePath!: string;

View File

@ -107,7 +107,3 @@ export interface NoteRow {
utcDateCreated: string; utcDateCreated: string;
utcDateModified: string; utcDateModified: string;
} }
export interface AttributeRow {
}

View File

@ -1,3 +1,5 @@
import { ConstructorData } from './becca-interface';
import AbstractBeccaEntity = require('./entities/abstract_becca_entity');
import BAttachment = require('./entities/battachment'); import BAttachment = require('./entities/battachment');
import BAttribute = require('./entities/battribute'); import BAttribute = require('./entities/battribute');
import BBlob = require('./entities/bblob'); import BBlob = require('./entities/bblob');
@ -8,7 +10,9 @@ import BOption = require('./entities/boption');
import BRecentNote = require('./entities/brecent_note'); import BRecentNote = require('./entities/brecent_note');
import BRevision = require('./entities/brevision'); import BRevision = require('./entities/brevision');
const ENTITY_NAME_TO_ENTITY: Record<string, any> = { type EntityClass = new (row?: any) => AbstractBeccaEntity<any>;
const ENTITY_NAME_TO_ENTITY: Record<string, ConstructorData<any> & EntityClass> = {
"attachments": BAttachment, "attachments": BAttachment,
"attributes": BAttribute, "attributes": BAttribute,
"blobs": BBlob, "blobs": BBlob,

View File

@ -1,137 +0,0 @@
const backupService = require('./backup');
const sql = require('./sql');
const fs = require('fs-extra');
const log = require('./log');
const utils = require('./utils');
const resourceDir = require('./resource_dir');
const appInfo = require('./app_info');
const cls = require('./cls');
async function migrate() {
const currentDbVersion = getDbVersion();
if (currentDbVersion < 214) {
log.error("Direct migration from your current version is not supported. Please upgrade to the latest v0.60.4 first and only then to this version.");
utils.crash();
return;
}
// backup before attempting migration
await backupService.backupNow(
// creating a special backup for version 0.60.4, the changes in 0.61 are major.
currentDbVersion === 214
? `before-migration-v060`
: 'before-migration'
);
const migrations = fs.readdirSync(resourceDir.MIGRATIONS_DIR).map(file => {
const match = file.match(/^([0-9]{4})__([a-zA-Z0-9_ ]+)\.(sql|js)$/);
if (!match) {
return null;
}
const dbVersion = parseInt(match[1]);
if (dbVersion > currentDbVersion) {
const name = match[2];
const type = match[3];
return {
dbVersion: dbVersion,
name: name,
file: file,
type: type
};
} else {
return null;
}
}).filter(el => !!el);
migrations.sort((a, b) => a.dbVersion - b.dbVersion);
// all migrations are executed in one transaction - upgrade either succeeds, or the user can stay at the old version
// otherwise if half of the migrations succeed, user can't use any version - DB is too "new" for the old app,
// and too old for the new app version.
cls.setMigrationRunning(true);
sql.transactional(() => {
for (const mig of migrations) {
try {
log.info(`Attempting migration to version ${mig.dbVersion}`);
executeMigration(mig);
sql.execute(`UPDATE options
SET value = ?
WHERE name = ?`, [mig.dbVersion.toString(), "dbVersion"]);
log.info(`Migration to version ${mig.dbVersion} has been successful.`);
} catch (e) {
log.error(`error during migration to version ${mig.dbVersion}: ${e.stack}`);
log.error("migration failed, crashing hard"); // this is not very user-friendly :-/
utils.crash();
break; // crash() above does not seem to work right away
}
}
});
if (currentDbVersion === 214) {
// special VACUUM after the big migration
log.info("VACUUMing database, this might take a while ...");
sql.execute("VACUUM");
}
}
function executeMigration(mig) {
if (mig.type === 'sql') {
const migrationSql = fs.readFileSync(`${resourceDir.MIGRATIONS_DIR}/${mig.file}`).toString('utf8');
console.log(`Migration with SQL script: ${migrationSql}`);
sql.executeScript(migrationSql);
} else if (mig.type === 'js') {
console.log("Migration with JS module");
const migrationModule = require(`${resourceDir.MIGRATIONS_DIR}/${mig.file}`);
migrationModule();
} else {
throw new Error(`Unknown migration type '${mig.type}'`);
}
}
function getDbVersion() {
return parseInt(sql.getValue("SELECT value FROM options WHERE name = 'dbVersion'"));
}
function isDbUpToDate() {
const dbVersion = getDbVersion();
const upToDate = dbVersion >= appInfo.dbVersion;
if (!upToDate) {
log.info(`App db version is ${appInfo.dbVersion}, while db version is ${dbVersion}. Migration needed.`);
}
return upToDate;
}
async function migrateIfNecessary() {
const currentDbVersion = getDbVersion();
if (currentDbVersion > appInfo.dbVersion && process.env.TRILIUM_IGNORE_DB_VERSION !== 'true') {
log.error(`Current DB version ${currentDbVersion} is newer than the current DB version ${appInfo.dbVersion}, which means that it was created by a newer and incompatible version of Trilium. Upgrade to the latest version of Trilium to resolve this issue.`);
utils.crash();
}
if (!isDbUpToDate()) {
await migrate();
}
}
module.exports = {
migrateIfNecessary,
isDbUpToDate
};

View File

@ -1,13 +1,13 @@
const path = require('path'); const path = require('path');
const url = require("url"); const url = require("url");
const port = require('./port.ts'); const port = require('./port');
const optionService = require('./options'); const optionService = require('./options');
const env = require('./env'); const env = require('./env');
const log = require('./log'); const log = require('./log');
const sqlInit = require('./sql_init'); const sqlInit = require('./sql_init');
const cls = require('./cls'); const cls = require('./cls');
const keyboardActionsService = require('./keyboard_actions'); const keyboardActionsService = require('./keyboard_actions');
const {ipcMain} = require('electron'); const { ipcMain } = require('electron');
// Prevent the window being garbage collected // Prevent the window being garbage collected
/** @type {Electron.BrowserWindow} */ /** @type {Electron.BrowserWindow} */
@ -18,7 +18,7 @@ let setupWindow;
async function createExtraWindow(extraWindowHash) { async function createExtraWindow(extraWindowHash) {
const spellcheckEnabled = optionService.getOptionBool('spellCheckEnabled'); const spellcheckEnabled = optionService.getOptionBool('spellCheckEnabled');
const {BrowserWindow} = require('electron'); const { BrowserWindow } = require('electron');
const win = new BrowserWindow({ const win = new BrowserWindow({
width: 1000, width: 1000,
@ -55,7 +55,7 @@ async function createMainWindow(app) {
const spellcheckEnabled = optionService.getOptionBool('spellCheckEnabled'); const spellcheckEnabled = optionService.getOptionBool('spellCheckEnabled');
const {BrowserWindow} = require('electron'); // should not be statically imported const { BrowserWindow } = require('electron'); // should not be statically imported
mainWindow = new BrowserWindow({ mainWindow = new BrowserWindow({
x: mainWindowState.x, x: mainWindowState.x,
@ -128,7 +128,7 @@ function getIcon() {
} }
async function createSetupWindow() { async function createSetupWindow() {
const {BrowserWindow} = require('electron'); // should not be statically imported const { BrowserWindow } = require('electron'); // should not be statically imported
setupWindow = new BrowserWindow({ setupWindow = new BrowserWindow({
width: 800, width: 800,
height: 800, height: 800,
@ -152,7 +152,7 @@ function closeSetupWindow() {
} }
async function registerGlobalShortcuts() { async function registerGlobalShortcuts() {
const {globalShortcut} = require('electron'); const { globalShortcut } = require('electron');
await sqlInit.dbReady; await sqlInit.dbReady;