rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add s3 url cache

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-04-30 13:38:46 -07:00
parent ee952bcc91
commit 29a40f5792
3 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/probo/evidence_service.go
View File

@@ -304,6 +304,7 @@ func (s EvidenceService) GenerateFileURL(
presignedReq, err := presignClient.PresignGetObject(ctx, &s3.GetObjectInput{ presignedReq, err := presignClient.PresignGetObject(ctx, &s3.GetObjectInput{
Bucket: aws.String(s.svc.bucket), Bucket: aws.String(s.svc.bucket),
Key: aws.String(evidence.ObjectKey), Key: aws.String(evidence.ObjectKey),
ResponseCacheControl: aws.String("max-age=3600, public"),
ResponseContentType: aws.String(evidence.MimeType), ResponseContentType: aws.String(evidence.MimeType),
ResponseContentDisposition: aws.String(contentDisposition), ResponseContentDisposition: aws.String(contentDisposition),
}, func(opts *s3.PresignOptions) { }, func(opts *s3.PresignOptions) {

11
pkg/probo/organization_service.go
View File

@@ -18,6 +18,7 @@ import (
"context" "context"
"fmt" "fmt"
"io" "io"
"net/url"
"time" "time"
"github.com/aws/aws-sdk-go-v2/aws" "github.com/aws/aws-sdk-go-v2/aws"
@@ -174,9 +175,15 @@ func (s OrganizationService) GenerateLogoURL(
presignClient := s3.NewPresignClient(s.svc.s3) presignClient := s3.NewPresignClient(s.svc.s3)
encodedFilename := url.QueryEscape(organization.Name)
contentDisposition := fmt.Sprintf("attachment; filename=\"%s\"; filename*=UTF-8''%s",
encodedFilename, encodedFilename)
presignedReq, err := presignClient.PresignGetObject(ctx, &s3.GetObjectInput{ presignedReq, err := presignClient.PresignGetObject(ctx, &s3.GetObjectInput{
Bucket: aws.String(s.svc.bucket), Bucket: aws.String(s.svc.bucket),
Key: aws.String(organization.LogoObjectKey), Key: aws.String(organization.LogoObjectKey),
ResponseCacheControl: aws.String("max-age=3600, public"),
ResponseContentDisposition: aws.String(contentDisposition),
}, func(opts *s3.PresignOptions) { }, func(opts *s3.PresignOptions) {
opts.Expires = expiresIn opts.Expires = expiresIn
}) })

1
pkg/probo/vendor_compliance_report_service.go
View File

@@ -169,6 +169,7 @@ func (s VendorComplianceReportService) GenerateFileURL(
presignedReq, err := presignClient.PresignGetObject(ctx, &s3.GetObjectInput{ presignedReq, err := presignClient.PresignGetObject(ctx, &s3.GetObjectInput{
Bucket: aws.String(s.svc.bucket), Bucket: aws.String(s.svc.bucket),
Key: aws.String(vendorComplianceReport.FileKey), Key: aws.String(vendorComplianceReport.FileKey),
ResponseCacheControl: aws.String("max-age=3600, public"),
ResponseContentDisposition: aws.String(contentDisposition), ResponseContentDisposition: aws.String(contentDisposition),
}, func(opts *s3.PresignOptions) { }, func(opts *s3.PresignOptions) {
opts.Expires = expiresIn opts.Expires = expiresIn