1
0
Fork 0

Reworked the systems handling and caching with sqlite

This commit is contained in:
Julien Dessaux 2023-05-31 01:19:31 +02:00
parent afe2db8a35
commit 61b5c8493e
Signed by: adyxax
GPG key ID: F92E51B86E07177E
4 changed files with 129 additions and 46 deletions

View file

@ -1,6 +1,6 @@
CREATE TABLE systems (
id INTEGER PRIMARY KEY,
symbol TEXT NOT NULL UNIQUE,
data TEXT NOT NULL,
updated DATE DEFAULT (datetime('now'))
data JSON NOT NULL,
updated DATE DEFAULT NULL
);
CREATE UNIQUE INDEX systems_data_symbol on systems (json_extract(data, '$.symbol'));

View file

@ -1,7 +1,25 @@
import db from './db.js';
const getSystemStatement = db.prepare(`SELECT data from systems where symbol = ?;`);
const setSystemStatement = db.prepare(`INSERT INTO systems(symbol, data) VALUES (?, ?);`);
const getSystemStatement = db.prepare(`SELECT data from systems where json_extract(data, '$.symbol') = ?;`);
const getSystemUpdatedStatement = db.prepare(`SELECT updated from systems where json_extract(data, '$.symbol') = ?;`);
const setSystemStatement = db.prepare(`INSERT INTO systems(data) VALUES (json(?));`);
const setSystemWaypointsStatement = db.prepare(`UPDATE systems SET data = (SELECT json_set(data, '$.waypoints', json(:waypoints)) FROM systems WHERE json_extract(data, '$.symbol') = :symbol), updated = :date WHERE json_extract(data, '$.symbol') = :symbol;`);
export function init() {
try {
return db.prepare(`INSERT INTO config(key, value) VALUES ('systems_initialized', TRUE);`).run().lastInsertRowid;
} catch (err) {
return null;
}
}
export function isInit() {
try {
return db.prepare(`SELECT value FROM config WHERE key = 'systems_initialized'`).get().value === '1';
} catch (err) {
return false;
}
}
export function getSystem(symbol) {
try {
@ -16,9 +34,34 @@ export function getSystem(symbol) {
}
}
export function setSystem(symbol, data) {
export function getSystemUpdated(symbol) {
try {
return setSystemStatement.run(symbol, JSON.stringify(data)).lastInsertRowid;
const updated = getSystemUpdatedStatement.get(symbol);
if (updated === undefined) {
return null;
}
return updated.updated;
} catch (err) {
console.log(err);
return null;
}
}
export function setSystem(data) {
try {
return setSystemStatement.run(JSON.stringify(data)).lastInsertRowid;
} catch (err) {
return null;
}
}
export function setSystemWaypoints(symbol, waypoints) {
try {
return setSystemWaypointsStatement.run({
date: new Date().toISOString(),
symbol: symbol,
waypoints: JSON.stringify(waypoints),
});
} catch (err) {
console.log(err);
return null;

View file

@ -1,44 +1,23 @@
import * as api from './api.js';
import * as db from '../database/systems.js';
// Retrieves a list of waypoints that have a specific ctx.trait like a SHIPYARD or a MARKETPLACE in the system ctx.symbol
export async function trait(ctx) {
const s = await system(ctx);
return s.filter(s => s.traits.some(t => t.symbol === ctx.trait));
}
// Retrieves a list of waypoints that have a specific ctx.type like ASTEROID_FIELD in the system ctx.symbol
export async function type(ctx, response) {
const s = await system(ctx);
return s.filter(s => s.type === ctx.type);
}
// Retrieves the system's information for ctx.symbol and cache it in the database
export async function system(ctx) {
let s = db.getSystem(ctx.symbol);
if (s === null) {
const response = await api.send({endpoint: `/systems/${ctx.symbol}/waypoints?limit=20&page=1`});
if (response.error !== undefined) {
switch(response.error.code) {
case 404:
throw `Error retrieving waypoints for system ${ctx.symbol}: ${response.error.message}`;
default: // yet unhandled error
throw response;
}
}
if (response.meta !== undefined && response.meta.total > response.meta.limit) {
throw `Error retrieving waypoints for system ${ctx.symbol}: Pagination is not implemented yet!`;
}
s = response.data;
db.setSystem(ctx.symbol, s);
// Retrieves all systems information, should be called only once after registering
export async function init(ctx) {
if (db.isInit()) {
return;
}
return s;
}
export async function systems(ctx) {
const response = await api.send({endpoint: `/systems?limit=20&page=1`});
// TODO pagination
return response;
for (let page=1; true; ++page) {
const response = await api.send({endpoint: `/systems?limit=20&page=${page}`, priority:100});
if (response.error !== undefined) {
throw response;
}
response.data.forEach(system => db.setSystem(system));
if (response.meta.total <= response.meta.limit * page) {
break;
}
}
console.log('Finished retrieving all systems information');
db.init();
}
// Retrieves a shipyard's information for ctx.symbol
@ -47,3 +26,61 @@ export async function shipyard(ctx) {
console.log(systemSymbol);
return await api.send({endpoint: `/systems/${systemSymbol}/waypoints/${ctx.symbol}/shipyard`});
}
// Retrieves the system's information for ctx.symbol and caches it in the database
export async function system(ctx) {
let s = db.getSystem(ctx.symbol);
if (s === null) {
const response = await api.send({endpoint: `/systems/${ctx.symbol}`});
if (response.error !== undefined) {
switch(response.error.code) {
case 404:
throw `Error retrieving info for system ${ctx.symbol}: ${response.error.message}`;
default: // yet unhandled error
throw response;
}
}
s = response.data;
db.setSystem(s);
}
return s;
}
// Retrieves a list of waypoints that have a specific ctx.trait like a SHIPYARD or a MARKETPLACE in the system ctx.symbol
export async function trait(ctx) {
const s = await system(ctx);
return s.waypoints.filter(s => s.traits.some(t => t.symbol === ctx.trait));
}
// Retrieves a list of waypoints that have a specific ctx.type like ASTEROID_FIELD in the system ctx.symbol
export async function type(ctx, response) {
const s = await system(ctx);
return s.waypoints.filter(s => s.type === ctx.type);
}
// Retrieves the system's information for ctx.symbol and caches it in the database
export async function waypoints(ctx) {
await system(ctx);
let updated = db.getSystemUpdated(ctx.symbol);
if (updated === null) {
let waypoints = [];
for (let page=1; true; ++page) {
const response = await api.send({endpoint: `/systems/${ctx.symbol}/waypoints?limit=20&page=${page}`, priority: 98});
if (response.error !== undefined) {
switch(response.error.code) {
case 404:
throw `Error retrieving waypoints for system ${ctx.symbol}: ${response.error.message}`;
default: // yet unhandled error
throw response;
}
}
waypoints = waypoints.concat(response.data);
if (response.meta.total <= response.meta.limit * page) {
break;
}
}
db.setSystemWaypoints(ctx.symbol, waypoints);
return waypoints;
}
return db.getSystem(ctx.symbol).waypoints;
}

View file

@ -109,11 +109,14 @@ default:
case 'systems.shipyards':
api.debugLog(await systems.trait({symbol: process.argv[3], trait: 'SHIPYARD'}));
break;
case 'systems.init':
await systems.init();
break;
case 'systems.system':
api.debugLog(await systems.system({symbol: process.argv[3]}));
break;
case 'systems.systems':
api.debugLog(await systems.systems());
case 'systems.waypoints':
api.debugLog(await systems.waypoints({symbol: process.argv[3]}));
break;
default:
usage();