From 6bf4b90c90f15f4ab60833bddf5b5756d1a6b1f6 Mon Sep 17 00:00:00 2001 From: Elizabeth Alexander Hunt Date: Thu, 2 Jul 2026 11:55:17 -0700 Subject: Init --- Homework/cs5260/consumer/dao/S3WidgetDao.ts | 93 +++++++++++++++++++++++++++++ 1 file changed, 93 insertions(+) create mode 100644 Homework/cs5260/consumer/dao/S3WidgetDao.ts (limited to 'Homework/cs5260/consumer/dao/S3WidgetDao.ts') diff --git a/Homework/cs5260/consumer/dao/S3WidgetDao.ts b/Homework/cs5260/consumer/dao/S3WidgetDao.ts new file mode 100644 index 0000000..d5aee72 --- /dev/null +++ b/Homework/cs5260/consumer/dao/S3WidgetDao.ts @@ -0,0 +1,93 @@ +import { WidgetT, Widget } from "@/shared/schema"; +import { WidgetDAO } from "."; +import { + DeleteObjectCommand, + GetObjectCommand, + NoSuchKey, + PutObjectCommand, + S3Client, +} from "@aws-sdk/client-s3"; +import { S3WidgetKeyGenerator } from "./keys"; +import { TracingLogger } from "@/consumer/log"; + +export class S3WidgetDAO implements WidgetDAO { + private bucket: string; + private s3Client: S3Client; + private widgetKeyGenerator: S3WidgetKeyGenerator; + private logger: TracingLogger; + + constructor( + bucket: string, + s3Client: S3Client, + widgetKeyGenerator: S3WidgetKeyGenerator, + logger: TracingLogger + ) { + this.bucket = bucket; + this.s3Client = s3Client; + this.widgetKeyGenerator = widgetKeyGenerator; + this.logger = logger; + } + + public async retrieve(widget: WidgetT) { + const key = this.widgetKeyGenerator.fromWidget(widget); + + this.logger.info( + `getting Widget=(${widget.id}) in Bucket=(${this.bucket})` + ); + + const retrievalRequest = new GetObjectCommand({ + Bucket: this.bucket, + Key: key, + }); + + try { + const objectResp = await this.s3Client.send(retrievalRequest); + const textRepr = await objectResp.Body?.transformToString(); + + if (textRepr) { + this.logger.info(`retrieved WidgetObject=(${widget.id}) from S3`); + return Widget.parse(JSON.parse(textRepr)); + } + } catch (e: unknown) { + if (e instanceof NoSuchKey) { + return; + } + throw e; + } + } + + public async save(widget: WidgetT) { + const key = this.widgetKeyGenerator.fromWidget(widget); + + this.logger.info(`saving Widget=(${widget.id}) in Bucket=(${this.bucket})`); + + const putReq = new PutObjectCommand({ + Bucket: this.bucket, + Key: key, + Body: JSON.stringify(widget, null, 2), + }); + + await this.s3Client.send(putReq); + + this.logger.info(`successfully saved Widget=(${widget.id}) in S3`); + return widget; + } + + public async delete(widget: WidgetT) { + const key = this.widgetKeyGenerator.fromWidget(widget); + + this.logger.info( + `deleting Widget=(${widget.id}) in Bucket=(${this.bucket})` + ); + + const deleteReq = new DeleteObjectCommand({ + Key: key, + Bucket: this.bucket, + }); + + await this.s3Client.send(deleteReq); + + this.logger.info(`successfully deleted Widget=(${widget.id}) in S3`); + return widget; + } +} -- cgit v1.3