aboutsummaryrefslogtreecommitdiff
path: root/apps/cf-ai-backend/src/utils/kvBulkInsert.ts
blob: 622364128a7208f7ceabb8b8cfed4b72f7cc596c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import { KVBulkItem } from "../types";

export const bulkInsertKv = async (
	credentials: {
		CF_KV_AUTH_TOKEN: string;
		KV_NAMESPACE_ID: string;
		CF_ACCOUNT_ID: string;
	},
	keyData: {
		chunkIds: Array<string>;
		urlid: string;
	},
) => {
	const data: Array<KVBulkItem> = keyData.chunkIds.map((chunkId) => ({
		key: chunkId,
		value: keyData.urlid,
		base64: false,
	}));

	try {
		const response = await fetch(
			`https://api.cloudflare.com/client/v4/accounts/${credentials.CF_ACCOUNT_ID}/storage/kv/namespaces/${credentials.KV_NAMESPACE_ID}/bulk`,
			{
				method: "PUT",
				headers: {
					Authorization: `Bearer ${credentials.CF_KV_AUTH_TOKEN}`,
					"Content-Type": "application/json",
				},
				body: JSON.stringify(data),
			},
		);

		if (!response.ok) {
			throw new Error(
				`can't insert bulk to kv because ${response.status} ${response.statusText} ${JSON.stringify(response.body)}`,
			);
		}
		return await response.json();
	} catch (e) {
		//dosomething
		throw e;
	}
};