90 lines
2.7 KiB
JavaScript
90 lines
2.7 KiB
JavaScript
import Promise from 'bluebird';
|
|
import { ChannelStateSizeError } from '../errors';
|
|
import db from '../database';
|
|
import { LoggerFactory } from '@calzoneman/jsli';
|
|
|
|
const LOGGER = LoggerFactory.getLogger('dbstore');
|
|
|
|
const SIZE_LIMIT = 1048576;
|
|
const QUERY_CHANNEL_DATA = 'SELECT `key`, `value` FROM channel_data WHERE channel_id = ?';
|
|
|
|
function queryAsync(query, substitutions) {
|
|
return new Promise((resolve, reject) => {
|
|
db.query(query, substitutions, (err, res) => {
|
|
if (err) {
|
|
if (!(err instanceof Error)) {
|
|
err = new Error(err);
|
|
}
|
|
reject(err);
|
|
} else {
|
|
resolve(res);
|
|
}
|
|
});
|
|
});
|
|
}
|
|
|
|
function buildUpdateQuery(numEntries) {
|
|
const values = [];
|
|
for (let i = 0; i < numEntries; i++) {
|
|
values.push('(?, ?, ?)');
|
|
}
|
|
|
|
return `INSERT INTO channel_data VALUES ${values.join(', ')} ` +
|
|
'ON DUPLICATE KEY UPDATE `value` = VALUES(`value`)';
|
|
}
|
|
|
|
export class DatabaseStore {
|
|
load(id, channelName) {
|
|
if (!id || id === 0) {
|
|
return Promise.reject(new Error(`Cannot load state for [${channelName}]: ` +
|
|
`id was passed as [${id}]`));
|
|
}
|
|
|
|
return queryAsync(QUERY_CHANNEL_DATA, [id]).then(rows => {
|
|
const data = {};
|
|
rows.forEach(row => {
|
|
try {
|
|
data[row.key] = JSON.parse(row.value);
|
|
} catch (e) {
|
|
LOGGER.error(`Channel data for channel "${channelName}", ` +
|
|
`key "${row.key}" is invalid: ${e}`);
|
|
}
|
|
});
|
|
|
|
return data;
|
|
});
|
|
}
|
|
|
|
save(id, channelName, data) {
|
|
if (!id || id === 0) {
|
|
return Promise.reject(new Error(`Cannot save state for [${channelName}]: ` +
|
|
`id was passed as [${id}]`));
|
|
}
|
|
|
|
let totalSize = 0;
|
|
let rowCount = 0;
|
|
const substitutions = [];
|
|
for (const key in data) {
|
|
if (typeof data[key] === 'undefined') {
|
|
continue;
|
|
}
|
|
rowCount++;
|
|
const value = JSON.stringify(data[key]);
|
|
totalSize += value.length;
|
|
substitutions.push(id);
|
|
substitutions.push(key);
|
|
substitutions.push(value);
|
|
}
|
|
|
|
if (totalSize > SIZE_LIMIT) {
|
|
return Promise.reject(new ChannelStateSizeError(
|
|
'Channel state size is too large', {
|
|
limit: SIZE_LIMIT,
|
|
actual: totalSize
|
|
}));
|
|
}
|
|
|
|
return queryAsync(buildUpdateQuery(rowCount), substitutions);
|
|
}
|
|
}
|