80 lines
2.4 KiB
JavaScript
80 lines
2.4 KiB
JavaScript
"use strict";
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
|
|
var _utils = require("./utils/index.cjs");
|
|
var _database = require("@planetscale/database");
|
|
var _nodeFetchNative = require("node-fetch-native");
|
|
const DRIVER_NAME = "planetscale";
|
|
module.exports = (0, _utils.defineDriver)((opts = {}) => {
|
|
opts.table = opts.table || "storage";
|
|
let _connection;
|
|
const getConnection = () => {
|
|
if (!_connection) {
|
|
if (!opts.url) {
|
|
throw (0, _utils.createRequiredError)(DRIVER_NAME, "url");
|
|
}
|
|
_connection = (0, _database.connect)({
|
|
url: opts.url,
|
|
fetch: _nodeFetchNative.fetch
|
|
});
|
|
if (opts.boostCache) {
|
|
_connection.execute("SET @@boost_cached_queries = true;").catch(error => {
|
|
console.error("[unstorage] [planetscale] Failed to enable cached queries:", error);
|
|
});
|
|
}
|
|
}
|
|
return _connection;
|
|
};
|
|
return {
|
|
name: DRIVER_NAME,
|
|
options: opts,
|
|
getInstance: getConnection,
|
|
hasItem: async key => {
|
|
const res = await getConnection().execute(`SELECT EXISTS (SELECT 1 FROM ${opts.table} WHERE id = :key) as value;`, {
|
|
key
|
|
});
|
|
return rows(res)[0]?.value == "1";
|
|
},
|
|
getItem: async key => {
|
|
const res = await getConnection().execute(`SELECT value from ${opts.table} WHERE id=:key;`, {
|
|
key
|
|
});
|
|
return rows(res)[0]?.value ?? null;
|
|
},
|
|
setItem: async (key, value) => {
|
|
await getConnection().execute(`INSERT INTO ${opts.table} (id, value) VALUES (:key, :value) ON DUPLICATE KEY UPDATE value = :value;`, {
|
|
key,
|
|
value
|
|
});
|
|
},
|
|
removeItem: async key => {
|
|
await getConnection().execute(`DELETE FROM ${opts.table} WHERE id=:key;`, {
|
|
key
|
|
});
|
|
},
|
|
getMeta: async key => {
|
|
const res = await getConnection().execute(`SELECT created_at, updated_at from ${opts.table} WHERE id=:key;`, {
|
|
key
|
|
});
|
|
return {
|
|
birthtime: rows(res)[0]?.created_at,
|
|
mtime: rows(res)[0]?.updated_at
|
|
};
|
|
},
|
|
getKeys: async (base = "") => {
|
|
const res = await getConnection().execute(`SELECT id from ${opts.table} WHERE id LIKE :base;`, {
|
|
base: `${base}%`
|
|
});
|
|
return rows(res).map(r => r.id);
|
|
},
|
|
clear: async () => {
|
|
await getConnection().execute(`DELETE FROM ${opts.table};`);
|
|
}
|
|
};
|
|
});
|
|
function rows(res) {
|
|
return res.rows || [];
|
|
} |