1
0
silverbullet/plugos/syscalls/store.knex_node.ts

83 lines
2.2 KiB
TypeScript
Raw Normal View History

import {Knex} from "knex";
import {SysCallMapping} from "../system";
2022-03-25 11:03:06 +00:00
type Item = {
page: string;
key: string;
value: any;
};
export type KV = {
key: string;
value: any;
};
export async function ensureTable(db: Knex<any, unknown>, tableName: string) {
if (!(await db.schema.hasTable(tableName))) {
await db.schema.createTable(tableName, (table) => {
table.string("key");
table.text("value");
table.primary(["key"]);
});
console.log(`Created table ${tableName}`);
}
}
export function storeSyscalls(
2022-03-25 11:03:06 +00:00
db: Knex<any, unknown>,
tableName: string
): SysCallMapping {
const apiObj: SysCallMapping = {
"store.delete": async (ctx, key: string) => {
2022-03-28 13:25:05 +00:00
await db<Item>(tableName).where({ key }).del();
2022-03-25 11:03:06 +00:00
},
"store.deletePrefix": async (ctx, prefix: string) => {
2022-03-25 11:03:06 +00:00
return db<Item>(tableName).andWhereLike("key", `${prefix}%`).del();
},
"store.deleteAll": async (ctx) => {
2022-03-25 11:03:06 +00:00
await db<Item>(tableName).del();
},
"store.set": async (ctx, key: string, value: any) => {
2022-03-25 11:03:06 +00:00
let changed = await db<Item>(tableName)
.where({ key })
.update("value", JSON.stringify(value));
if (changed === 0) {
await db<Item>(tableName).insert({
key,
value: JSON.stringify(value),
});
}
},
2022-03-28 13:25:05 +00:00
// TODO: Optimize
"store.batchSet": async (ctx, kvs: KV[]) => {
2022-03-25 11:03:06 +00:00
for (let { key, value } of kvs) {
await apiObj["store.set"](ctx, key, value);
2022-03-28 13:25:05 +00:00
}
},
"store.batchDelete": async (ctx, keys: string[]) => {
2022-03-28 13:25:05 +00:00
for (let key of keys) {
await apiObj["store.delete"](ctx, key);
2022-03-25 11:03:06 +00:00
}
},
"store.get": async (ctx, key: string): Promise<any | null> => {
2022-03-25 11:03:06 +00:00
let result = await db<Item>(tableName).where({ key }).select("value");
if (result.length) {
return JSON.parse(result[0].value);
} else {
return null;
}
},
"store.queryPrefix": async (ctx, prefix: string) => {
2022-03-25 11:03:06 +00:00
return (
await db<Item>(tableName)
.andWhereLike("key", `${prefix}%`)
.select("key", "value")
).map(({ key, value }) => ({
key,
value: JSON.parse(value),
}));
},
};
return apiObj;
2022-03-25 11:03:06 +00:00
}