98 lines
4.1 KiB
JavaScript
98 lines
4.1 KiB
JavaScript
|
/* sheetjs (C) 2013-present SheetJS -- https://sheetjs.com */
|
||
|
import { utils } from "xlsx";
|
||
|
|
||
|
/* Generate worksheet from database */
|
||
|
async function redis_to_ws(client) {
|
||
|
/* Get a list of every item in the database */
|
||
|
const keys = await client.KEYS("*");
|
||
|
|
||
|
/* Collect the full contents */
|
||
|
const data = [];
|
||
|
for(let key of keys) {
|
||
|
const type = await client.TYPE(key);
|
||
|
let value;
|
||
|
switch(type) {
|
||
|
case "string": value = await client.GET(key); break;
|
||
|
case "list": value = await client.LRANGE(key, 0, -1); break;
|
||
|
case "set": value = await client.SMEMBERS(key); break;
|
||
|
case "zset": value = await client.ZRANGE_WITHSCORES(key, 0, -1); break;
|
||
|
case "hash": value = await client.HGETALL(key); break;
|
||
|
default: console.warn(`unsupported type ${type}`); break
|
||
|
}
|
||
|
data.push({key, type, value});
|
||
|
}
|
||
|
|
||
|
/* Create a new worksheet and add the string table */
|
||
|
const ws = utils.aoa_to_sheet([["Strings"]]);
|
||
|
utils.sheet_add_aoa(ws, data.filter(r => r.type == "string").map(r => [r.key, r.value]), {origin: "A3"});
|
||
|
|
||
|
/* Add the other types */
|
||
|
let C = 2;
|
||
|
data.forEach(row => {
|
||
|
switch(row.type) {
|
||
|
case "set":
|
||
|
case "list": {
|
||
|
/* `value` is an array. aoa prepends type and key, then transposes to make a column */
|
||
|
var aoa = [row.type == "set" ? "Set" : "List", row.key].concat(row.value).map(x => ([x]));
|
||
|
utils.sheet_add_aoa(ws, aoa, {origin: utils.encode_col(C) + "1"});
|
||
|
++C;
|
||
|
} break;
|
||
|
case "zset": {
|
||
|
/* `value` is an object with value/score keys. generate array with map and prepend metadata */
|
||
|
var aoa = [["Sorted"], [row.key]].concat(row.value.map(r => ([r.value, r.score])));
|
||
|
utils.sheet_add_aoa(ws, aoa, {origin: utils.encode_col(C) + "1"});
|
||
|
C += 2;
|
||
|
} break;
|
||
|
case "hash": {
|
||
|
/* `value` is an object. Object.entries returns an array of arrays */
|
||
|
var aoa = [["Hash"], [row.key]].concat(Object.entries(row.value));
|
||
|
utils.sheet_add_aoa(ws, aoa, {origin: utils.encode_col(C) + "1"});
|
||
|
C += 2;
|
||
|
} break;
|
||
|
case "string": break;
|
||
|
}
|
||
|
});
|
||
|
|
||
|
return ws;
|
||
|
}
|
||
|
|
||
|
/* Generate array of Redis commands */
|
||
|
async function ws_to_redis(ws) {
|
||
|
const cmds = [];
|
||
|
/* Extract data from the worksheet */
|
||
|
const aoa = utils.sheet_to_json(ws, { header: 1 });
|
||
|
/* Iterate over the values in the first row */
|
||
|
aoa[0].forEach((type, C) => {
|
||
|
/* The name is exepcted in the second row, same column, unless type is Strings */
|
||
|
if(type != "Strings" && !aoa[1][C]) throw new Error(`Column ${utils.encode_col(C)} missing name!`)
|
||
|
switch(type) {
|
||
|
case "Strings": {
|
||
|
if(aoa[0][C+1]) throw new Error(`${type} requires 2 columns!`);
|
||
|
/* For each row starting with SheetJS 2, check if key and value are present */
|
||
|
for(let R = 2; R < aoa.length; ++R)
|
||
|
if(aoa[R]?.[C] != null && aoa[R]?.[C+1] != null)
|
||
|
/* When key and value are present, emit a SET command */
|
||
|
cmds.push(["SET", [aoa[R][C], aoa[R][C+1]]]);
|
||
|
} break;
|
||
|
case "Set":
|
||
|
case "List": {
|
||
|
/* SADD (Set) / RPUSH (List) second argument is an array of values to add to the set */
|
||
|
cmds.push([ type == "Set" ? "SADD" : "RPUSH", [ aoa[1][C], aoa.slice(2).map(r => r[C]).filter(x => x != null) ]]);
|
||
|
} break;
|
||
|
case "Sorted": {
|
||
|
/* ZADD second argument is an array of objects with `value` and `score` */
|
||
|
if(aoa[0][C+1]) throw new Error(`${type} requires 2 columns!`);
|
||
|
cmds.push([ "ZADD", [ aoa[1][C], aoa.slice(2).map(r => ({value: r[C], score:r[C+1]})).filter(x => x.value != null && x.score != null) ]]);
|
||
|
} break;
|
||
|
case "Hash": {
|
||
|
/* HSET second argument is an object. `Object.fromEntries` generates object from an array of K/V pairs */
|
||
|
if(aoa[0][C+1]) throw new Error(`${type} requires 2 columns!`);
|
||
|
cmds.push([ "HSET", [ aoa[1][C], Object.fromEntries(aoa.slice(2).map(r => [r[C], r[C+1]]).filter(x => x[0] != null && x[1] != null)) ]]);
|
||
|
} break;
|
||
|
default: console.error(`Unrecognized column type ${type}`); break;
|
||
|
}
|
||
|
})
|
||
|
return cmds;
|
||
|
}
|
||
|
|
||
|
export { ws_to_redis, redis_to_ws };
|