ChatGPT/ChatGPT-Next-Web-LangChain-Gpt-4-All
1
0
Fork 0
mirror of https://github.com/Yanyutin753/ChatGPT-Next-Web-LangChain-Gpt-4-All.git synced 2025-10-17 16:44:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: support custom s3 service

Browse Source
This commit is contained in:
Hk-Gosuto
2023-12-10 21:02:49 +08:00
parent eb7711f832
commit b84da5e120
6 changed files with 88 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/utils/r2_file_storage.ts → app/utils/s3_file_storage.ts
View File

@@ -12,13 +12,19 @@ const R2_ACCESS_KEY_ID = process.env.R2_ACCESS_KEY_ID;
const R2_SECRET_ACCESS_KEY = process.env.R2_SECRET_ACCESS_KEY;
const R2_BUCKET = process.env.R2_BUCKET;
const S3_ENDPOINT = process.env.S3_ENDPOINT;
const S3_ACCESS_KEY_ID = process.env.S3_ACCESS_KEY_ID;
const S3_SECRET_ACCESS_KEY = process.env.S3_SECRET_ACCESS_KEY;
const S3_BUCKET = process.env.S3_BUCKET;
const getR2Client = () => {
return new S3Client({
region: "auto",
endpoint: `https://${R2_ACCOUNT_ID}.r2.cloudflarestorage.com`,
endpoint:
S3_ENDPOINT ?? `https://${R2_ACCOUNT_ID}.r2.cloudflarestorage.com`,
credentials: {
accessKeyId: R2_ACCESS_KEY_ID!,
secretAccessKey: R2_SECRET_ACCESS_KEY!,
accessKeyId: S3_ACCESS_KEY_ID ?? R2_ACCESS_KEY_ID!,
secretAccessKey: S3_SECRET_ACCESS_KEY ?? R2_SECRET_ACCESS_KEY!,
},
});
};
@@ -27,7 +33,7 @@ export default class S3FileStorage {
static async get(fileName: string) {
const file = await getR2Client().send(
new GetObjectCommand({
Bucket: R2_BUCKET,
Bucket: S3_BUCKET ?? R2_BUCKET,
Key: fileName,
}),
);
@@ -43,7 +49,7 @@ export default class S3FileStorage {
const signedUrl = await getSignedUrl(
getR2Client(),
new PutObjectCommand({
Bucket: R2_BUCKET,
Bucket: S3_BUCKET ?? R2_BUCKET,
Key: fileName,
}),
{ expiresIn: 60 },
@@ -59,7 +65,7 @@ export default class S3FileStorage {
return `/api/file/${fileName}`;
} catch (e) {
console.error("[R2]", e);
console.error("[S3]", e);
throw e;
}
}