7x31
/
ChatGPT-Next-Web
mirror of https://github.com/Yidadaa/ChatGPT-Next-Web.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add expiration_ttl for kv storage

This commit is contained in:
lloydzhou 2024-07-25 16:56:08 +08:00
parent 7c1bc1f1a1
commit d8afd1af88
3 changed files with 32 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
app/api/artifact/route.ts
View File

@ -4,18 +4,33 @@ import { getServerSideConfig } from "@/app/config/server";
async function handle(req: NextRequest, res: NextResponse) { async function handle(req: NextRequest, res: NextResponse) {
const serverConfig = getServerSideConfig(); const serverConfig = getServerSideConfig();
const storeUrl = (key: string) => const storeUrl = () =>
`https://api.cloudflare.com/client/v4/accounts/${serverConfig.cloudflareAccountId}/storage/kv/namespaces/${serverConfig.cloudflareKVNamespaceId}/values/${key}`; `https://api.cloudflare.com/client/v4/accounts/${serverConfig.cloudflareAccountId}/storage/kv/namespaces/${serverConfig.cloudflareKVNamespaceId}`;
const storeHeaders = () => ({ const storeHeaders = () => ({
Authorization: `Bearer ${serverConfig.cloudflareKVApiKey}`, Authorization: `Bearer ${serverConfig.cloudflareKVApiKey}`,
}); });
if (req.method === "POST") { if (req.method === "POST") {
const clonedBody = await req.text(); const clonedBody = await req.text();
const hashedCode = md5.hash(clonedBody).trim(); const hashedCode = md5.hash(clonedBody).trim();
const res = await fetch(storeUrl(hashedCode), { const body = {
headers: storeHeaders(), key: hashedCode,
value: clonedBody,
};
try {
const ttl = parseInt(serverConfig.cloudflareKVTTL);
if (ttl > 60) {
body["expiration_ttl"] = ttl;
}
} catch (e) {
console.error(e);
}
const res = await fetch(`${storeUrl()}/bulk`, {
headers: {
...storeHeaders(),
"Content-Type": "application/json",
},
method: "PUT", method: "PUT",
body: clonedBody, body: JSON.stringify([body]),
}); });
const result = await res.json(); const result = await res.json();
console.log("save data", result); console.log("save data", result);
@ -32,7 +47,7 @@ async function handle(req: NextRequest, res: NextResponse) {
} }
if (req.method === "GET") { if (req.method === "GET") {
const id = req?.nextUrl?.searchParams?.get("id"); const id = req?.nextUrl?.searchParams?.get("id");
const res = await fetch(storeUrl(id as string), { const res = await fetch(`${storeUrl()}/values/${id}`, {
headers: storeHeaders(), headers: storeHeaders(),
method: "GET", method: "GET",
}); });

11
app/components/artifact.tsx
View File

@ -180,8 +180,17 @@ export function Artifact() {
useEffect(() => { useEffect(() => {
if (id) { if (id) {
fetch(`${ApiPath.Artifact}?id=${id}`) fetch(`${ApiPath.Artifact}?id=${id}`)
.then((res) => {
if (res.status > 300) {
throw Error("can not get content");
}
return res;
})
.then((res) => res.text()) .then((res) => res.text())
.then(setCode); .then(setCode)
.catch((e) => {
showToast(Locale.Export.Artifact.Error);
});
} }
}, [id]); }, [id]);

1
app/config/server.ts
View File

@ -161,6 +161,7 @@ export const getServerSideConfig = () => {
cloudflareAccountId: process.env.CLOUDFLARE_ACCOUNT_ID, cloudflareAccountId: process.env.CLOUDFLARE_ACCOUNT_ID,
cloudflareKVNamespaceId: process.env.CLOUDFLARE_KV_NAMESPACE_ID, cloudflareKVNamespaceId: process.env.CLOUDFLARE_KV_NAMESPACE_ID,
cloudflareKVApiKey: getApiKey(process.env.CLOUDFLARE_KV_API_KEY), cloudflareKVApiKey: getApiKey(process.env.CLOUDFLARE_KV_API_KEY),
cloudflareKVTTL: process.env.CLOUDFLARE_KV_TTL,
gtmId: process.env.GTM_ID, gtmId: process.env.GTM_ID,