Write filenames to shutdown prep bucket

glacials · Feb 8, 2025 · 9fc970c · 9fc970c
1 parent 5539303
commit 9fc970c
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 0 deletions.
diff --git a/app/controllers/api/v3/runs_controller.rb b/app/controllers/api/v3/runs_controller.rb
@@ -24,6 +24,14 @@ def create
       key:  "splits/#{filename}",
       body: params.require(:file)
     )
+    $s3_shutdown_prep_bucket.put_object(
+      key: @run.id,
+      body: filename,
+    )
+    $s3_shutdown_prep_bucket.put_object(
+      key: "extensions/#{@run.id}",
+      body: @run.program.file_extension,
+    )
 
     render status: :created, location: api_v3_run_url(@run), json: {
       status: 201,

diff --git a/app/controllers/api/v4/converts_controller.rb b/app/controllers/api/v4/converts_controller.rb
@@ -16,6 +16,11 @@ def create
       key:  "splits/#{filename}",
       body: params.require(:file)
     )
+    $s3_shutdown_prep_bucket.put_object(
+      key: @run.id,
+      body: filename,
+    )
+
 
     @run.parse_into_db
     if @run.program.nil?
@@ -25,6 +30,10 @@ def create
       }
       return
     end
+    $s3_shutdown_prep_bucket.put_object(
+      key: "extensions/#{@run.id}",
+      body: @run.program.file_extension,
+    )
 
     # Use a new find by to eager load the segment histories to prevent n+1 queries
     @run = Run.includes(segments: [:histories]).find(@run.id)

diff --git a/app/models/run.rb b/app/models/run.rb
@@ -416,6 +416,14 @@ def split(more: false, realtime_end_ms: nil, gametime_end_ms: nil) # rubocop:tod
         locals: {:@run => self},
       ),
     )
+    $s3_shutdown_prep_bucket.put_object(
+      key: id,
+      body: s3_filename,
+    )
+    $s3_shutdown_prep_bucket.put_object(
+      key: "extensions/#{@run.id}",
+      body: @run.program.file_extension,
+    )
 
     save
   end

diff --git a/config/initializers/s3.rb b/config/initializers/s3.rb
@@ -32,3 +32,5 @@
 
 $s3_bucket_internal = Aws::S3::Bucket.new(ENV['S3_BUCKET'], client: $s3_client_internal)
 $s3_bucket_external = Aws::S3::Bucket.new(ENV['S3_BUCKET'], client: $s3_client_external)
+
+$s3_shutdown_prep_bucket = Aws::S3::Bucket.new("splits.io-runid-to-s3filename", client: $s3_client_internal)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -32,3 +32,5 @@

		$s3_bucket_internal = Aws::S3::Bucket.new(ENV['S3_BUCKET'], client: $s3_client_internal)
		$s3_bucket_external = Aws::S3::Bucket.new(ENV['S3_BUCKET'], client: $s3_client_external)

		$s3_shutdown_prep_bucket = Aws::S3::Bucket.new("splits.io-runid-to-s3filename", client: $s3_client_internal)