hensei-api/app/services/preview_service/preview_generation_monitor.rb
Justin Edmund e3a44ca0d5
Implement embed images (#173)
* Add mini_magick and rufus-scheduler

* Expose attributes and add sigs to AwsService

* Get Party ready for preview state

* Added new fields for preview state and generated_at timestamp
* Add preview state enum to model
* Add preview_relevant_changes? after_commit hook

* Add jobs for generating and cleaning up party previews

* Add new endpoints to PartiesController

* `preview` shows the preview and queues it up for generation if it doesn't exist yet
* `regenerate_preview` allows the party owner to force regeneration of previews

* Schedule jobs

* Stalled jobs are checked every 5 minutes
* Failed jobs are retried every hour
* Old preview jobs are cleaned up daily

* Add the preview service

This is where the bulk of the work is. This service renders out the preview images bit by bit. Currently we render the party name, creator, job icon, and weapon grid.

This includes signatures and some fonts.
2025-01-18 09:08:15 -08:00

54 lines
1.7 KiB
Ruby

# app/services/preview_generation_monitor.rb
module PreviewService
class PreviewGenerationMonitor
class << self
def check_stalled_jobs
Party.where(preview_state: :queued)
.where('updated_at < ?', 10.minutes.ago)
.find_each do |party|
Rails.logger.warn("Found stalled preview generation for party #{party.id}")
# If no job is actually queued, reset the state
unless job_exists?(party)
party.update!(preview_state: :pending)
Rails.logger.info("Reset stalled party #{party.id} to pending state")
end
end
end
def retry_failed
Party.where(preview_state: :failed)
.where('updated_at < ?', 1.hour.ago)
.find_each do |party|
Rails.logger.info("Retrying failed preview generation for party #{party.id}")
GeneratePartyPreviewJob.perform_later(party.id)
end
end
def cleanup_old_previews
Party.where(preview_state: :generated)
.where('preview_generated_at < ?', 30.days.ago)
.find_each do |party|
PreviewService::Coordinator.new(party).delete_preview
end
end
private
def job_exists?(party)
# Implementation depends on your job backend
# For Sidekiq:
queue = Sidekiq::Queue.new('previews')
scheduled = Sidekiq::ScheduledSet.new
retrying = Sidekiq::RetrySet.new
[queue, scheduled, retrying].any? do |set|
set.any? do |job|
job.args.first == party.id &&
job.klass == 'GeneratePartyPreviewJob'
end
end
end
end
end
end