feat: use luxon for TTL

This commit is contained in:
2025-12-18 08:51:53 -05:00
parent 45f6f678b6
commit 7cf0c4c1c7
2 changed files with 29 additions and 25 deletions

View File

@@ -1,5 +1,4 @@
import { OpenAPIHono, createRoute, z } from "@hono/zod-openapi"; import { OpenAPIHono, createRoute, z } from "@hono/zod-openapi";
import type { HonoRequest } from "hono";
import { AnilistUpdateType } from "~/libs/anilist/updateType.ts"; import { AnilistUpdateType } from "~/libs/anilist/updateType.ts";
import { maybeScheduleNextAiringEpisode } from "~/libs/maybeScheduleNextAiringEpisode"; import { maybeScheduleNextAiringEpisode } from "~/libs/maybeScheduleNextAiringEpisode";

View File

@@ -1,6 +1,7 @@
import type { TypedDocumentNode } from "@graphql-typed-document-node/core"; import type { TypedDocumentNode } from "@graphql-typed-document-node/core";
import { DurableObject } from "cloudflare:workers"; import { DurableObject } from "cloudflare:workers";
import { print } from "graphql"; import { print } from "graphql";
import { DateTime } from "luxon";
import { z } from "zod"; import { z } from "zod";
import { import {
@@ -99,7 +100,7 @@ export class AnilistDurableObject extends DurableObject {
}); });
return data?.Media; return data?.Media;
}, },
60 * 60 * 1000, DateTime.now().plus({ hours: 1 }),
); );
} }
@@ -114,7 +115,7 @@ export class AnilistDurableObject extends DurableObject {
}); });
return data?.Page; return data?.Page;
}, },
60 * 60 * 1000, DateTime.now().plus({ hours: 1 }),
); );
} }
@@ -156,7 +157,7 @@ export class AnilistDurableObject extends DurableObject {
limit, limit,
}); });
}, },
24 * 60 * 60 * 1000, DateTime.now().plus({ days: 1 }),
); );
} }
@@ -169,15 +170,14 @@ export class AnilistDurableObject extends DurableObject {
return this.handleCachedRequest( return this.handleCachedRequest(
`popular:${JSON.stringify({ page, limit, season, seasonYear })}`, `popular:${JSON.stringify({ page, limit, season, seasonYear })}`,
async () => { async () => {
const data = await this.fetchFromAnilist(GetPopularTitlesQuery, { return this.fetchFromAnilist(GetPopularTitlesQuery, {
page, page,
limit, limit,
season, season,
seasonYear, seasonYear,
}); }).then((data) => data?.Page);
return data?.Page;
}, },
24 * 60 * 60 * 1000, DateTime.now().plus({ days: 1 }),
); );
} }
@@ -191,7 +191,7 @@ export class AnilistDurableObject extends DurableObject {
}); });
return data?.Page; return data?.Page;
}, },
24 * 60 * 60 * 1000, DateTime.now().plus({ days: 1 }),
); );
} }
@@ -210,7 +210,7 @@ export class AnilistDurableObject extends DurableObject {
}); });
return data?.Page; return data?.Page;
}, },
24 * 60 * 60 * 1000, DateTime.now().plus({ days: 1 }),
); );
} }
@@ -218,10 +218,10 @@ export class AnilistDurableObject extends DurableObject {
return this.handleCachedRequest( return this.handleCachedRequest(
`user:${token}`, `user:${token}`,
async () => { async () => {
const data = await this.fetchFromAnilist(GetUserQuery, {}, token); const data = await this.fetchFromAnilist(GetUserQuery, {}, { token });
return data?.Viewer; return data?.Viewer;
}, },
60 * 60 * 24 * 30 * 1000, DateTime.now().plus({ days: 30 }),
); );
} }
@@ -232,11 +232,11 @@ export class AnilistDurableObject extends DurableObject {
const data = await this.fetchFromAnilist( const data = await this.fetchFromAnilist(
GetUserProfileQuery, GetUserProfileQuery,
{ token }, { token },
token, { token },
); );
return data?.Viewer; return data?.Viewer;
}, },
60 * 60 * 24 * 30 * 1000, DateTime.now().plus({ days: 30 }),
); );
} }
@@ -248,7 +248,7 @@ export class AnilistDurableObject extends DurableObject {
const data = await this.fetchFromAnilist( const data = await this.fetchFromAnilist(
MarkEpisodeAsWatchedMutation, MarkEpisodeAsWatchedMutation,
{ titleId, episodeNumber }, { titleId, episodeNumber },
token, { token },
); );
return data?.SaveMediaListEntry; return data?.SaveMediaListEntry;
} }
@@ -257,7 +257,7 @@ export class AnilistDurableObject extends DurableObject {
const data = await this.fetchFromAnilist( const data = await this.fetchFromAnilist(
MarkTitleAsWatchedMutation, MarkTitleAsWatchedMutation,
{ titleId }, { titleId },
token, { token },
); );
return data?.SaveMediaListEntry; return data?.SaveMediaListEntry;
} }
@@ -266,7 +266,7 @@ export class AnilistDurableObject extends DurableObject {
async handleCachedRequest<T>( async handleCachedRequest<T>(
key: string, key: string,
fetcher: () => Promise<T>, fetcher: () => Promise<T>,
ttl?: number | ((data: T) => number | undefined), ttl?: DateTime | ((data: T) => DateTime | undefined),
) { ) {
const cache = await this.state.storage.get(key); const cache = await this.state.storage.get(key);
console.debug(`Retrieving request ${key} from cache:`, cache != null); console.debug(`Retrieving request ${key} from cache:`, cache != null);
@@ -278,9 +278,8 @@ export class AnilistDurableObject extends DurableObject {
await this.state.storage.put(key, result); await this.state.storage.put(key, result);
const calculatedTtl = typeof ttl === "function" ? ttl(result) : ttl; const calculatedTtl = typeof ttl === "function" ? ttl(result) : ttl;
if (calculatedTtl) {
if (calculatedTtl && calculatedTtl > 0) { const alarmTime = calculatedTtl.toMillis();
const alarmTime = Date.now() + calculatedTtl;
await this.state.storage.setAlarm(alarmTime); await this.state.storage.setAlarm(alarmTime);
await this.state.storage.put(`alarm:${key}`, alarmTime); await this.state.storage.put(`alarm:${key}`, alarmTime);
} }
@@ -307,8 +306,11 @@ export class AnilistDurableObject extends DurableObject {
async fetchFromAnilist<Result = any, Variables = any>( async fetchFromAnilist<Result = any, Variables = any>(
query: TypedDocumentNode<Result, Variables>, query: TypedDocumentNode<Result, Variables>,
variables: Variables, variables: Variables,
token?: string | undefined, {
): Promise<Result> { token,
shouldRetryOnRateLimit = true,
}: { token?: string | undefined; shouldRetryOnRateLimit?: boolean } = {},
): Promise<Result | undefined> {
const headers: any = { const headers: any = {
"Content-Type": "application/json", "Content-Type": "application/json",
}; };
@@ -338,14 +340,17 @@ export class AnilistDurableObject extends DurableObject {
}); });
// 1. Handle Rate Limiting (429) // 1. Handle Rate Limiting (429)
if (response.status === 429) { if (shouldRetryOnRateLimit && response.status === 429) {
const retryAfter = await response const retryAfter = await response
.json() .json<{ headers: Record<string, string> }>()
.then(({ headers }) => new Headers(headers).get("Retry-After")); .then(({ headers }) => new Headers(headers).get("Retry-After"));
console.log("429, retrying in", retryAfter); console.log("429, retrying in", retryAfter);
await sleep(Number(retryAfter || 1) * 1000); // specific fallback or ensure logic await sleep(Number(retryAfter || 1) * 1000); // specific fallback or ensure logic
return this.fetchFromAnilist(query, variables, token); return this.fetchFromAnilist(query, variables, {
token,
shouldRetryOnRateLimit: false,
});
} }
// 2. Handle HTTP Errors (like 404 or 500) // 2. Handle HTTP Errors (like 404 or 500)