unstorage/drivers/planetscale.mjs

Universal Storage Layer

import { defineDriver } from "./utils/index.mjs";
import { connect } from "@planetscale/database";
import { fetch } from "node-fetch-native";
export default defineDriver((opts = {}) => {
  if (!opts.url)
    throw new Error("Database URL is required to use the Planetscale driver.");
  opts.table = opts.table || "storage";
  let _connection;
  const getConnection = () => {
    if (!_connection) {
      _connection = connect({
        url: opts.url,
        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: "planetscale",
    options: opts,
    hasItem: async (key) => {
      const res = await getConnection().execute(
        `SELECT COUNT(id) from ${opts.table} WHERE id=:key;`,
        { key }
      );
      return res.size >= 0;
    },
    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 || [];
}