Use Postgres as cache/message queue backend

docs/src/content/docs/install/env.mdx

@@ -18,8 +18,7 @@ The URL of the PostgreSQL database, e.g.,
 
 The URL of the Redis server, e.g., `redis://localhost/0`.
 
-If not set, Hollo will use an in-memory cache and in-process job queue, which
-means that the cache and job queue will be lost when the server restarts.
+If not set, Hollo will use PostgreSQL for caching and message queueing.
 
 ###  `HOME_URL` <Badge text="Optional" />
 

package.json

@@ -13,10 +13,11 @@
   "dependencies": {
     "@aws-sdk/client-s3": "^3.577.0",
     "@aws-sdk/credential-providers": "^3.577.0",
-    "@fedify/fedify": "1.0.0-dev.388",
+    "@fedify/fedify": "^1.0.0-dev.410",
     "@fedify/markdown-it-hashtag": "0.2.0",
     "@fedify/markdown-it-mention": "^0.1.1",
-    "@fedify/redis": "^0.1.1",
+    "@fedify/postgres": "^0.1.0-dev.3",
+    "@fedify/redis": "^0.2.0-dev.10",
     "@hono/zod-validator": "^0.2.1",
     "@js-temporal/polyfill": "^0.4.4",
     "@logtape/logtape": "^0.4.2",

src/db.ts

@@ -1,7 +1,7 @@
 import { getLogger } from "@logtape/logtape";
 import type { Logger } from "drizzle-orm/logger";
 import { drizzle } from "drizzle-orm/postgres-js";
-import postgres from "postgres";
+import createPostgres from "postgres";
 import * as schema from "./schema";
 
 // biome-ignore lint/complexity/useLiteralKeys: tsc rants about this (TS4111)
@@ -56,7 +56,7 @@ class LogTapeLogger implements Logger {
   }
 }
 
-const client = postgres(databaseUrl, { connect_timeout: 5 });
-export const db = drizzle(client, { schema, logger: new LogTapeLogger() });
+export const postgres = createPostgres(databaseUrl, { connect_timeout: 5 });
+export const db = drizzle(postgres, { schema, logger: new LogTapeLogger() });
 
 export default db;

src/federation/index.ts

@@ -12,7 +12,6 @@ import {
   Image,
   type KvStore,
   Like,
-  MemoryKvStore,
   type MessageQueue,
   Note,
   PropertyValue,
@@ -26,6 +25,7 @@ import {
   importJwk,
   isActor,
 } from "@fedify/fedify";
+import { PostgresKvStore, PostgresMessageQueue } from "@fedify/postgres";
 import { RedisKvStore, RedisMessageQueue } from "@fedify/redis";
 import { getLogger } from "@logtape/logtape";
 import { parse } from "@std/semver";
@@ -40,7 +40,7 @@ import {
   like,
 } from "drizzle-orm";
 import metadata from "../../package.json" with { type: "json" };
-import db from "../db";
+import { db, postgres } from "../db";
 import { createRedis, getRedisUrl } from "../redis";
 import {
   type NewLike,
@@ -69,13 +69,13 @@ import {
 const logger = getLogger(["hollo", "federation"]);
 
 let kv: KvStore;
-let queue: MessageQueue | undefined;
+let queue: MessageQueue;
 
 if (getRedisUrl() == null) {
-  kv = new MemoryKvStore();
-  queue = undefined;
+  kv = new PostgresKvStore(postgres);
+  queue = new PostgresMessageQueue(postgres);
   logger.warn(
-    "No REDIS_URL is defined, using in-memory store and in-process queue.",
+    "No REDIS_URL is defined, using PostgresKvStore and PostgresMessageQueue.",
   );
 } else {
   kv = new RedisKvStore(createRedis());