Mantiene la copertina YouTube se il telefono cade e avvia ffmpeg sul nodo, non via SSH.

Il relay HLS→RTMPS resta sul worker/agent del nodo assegnato, così tre dirette contemporanee restano in onda sul sito e sul canale della società senza schermo nero. Co-authored-by: Cursor <cursoragent@cursor.com>
2026-08-16 20:41:46 +02:00
parent bfb3115a64
commit baa6283a15
36 changed files with 1444 additions and 141 deletions
@@ -29,4 +29,126 @@ namespace :streams do
      puts Streams::DnsProviders::Lab.new.hosts_file_snippet
    end
  end
+
+  namespace :relay do
+    desc "Mostra coda e slug nodo per i worker ffmpeg YouTube"
+    task queues: :environment do
+      puts "local_slug=#{Streams::YoutubeRelay.local_node_slug} worker=#{Streams::YoutubeRelay.worker?}"
+      StreamNode.order(:slug).find_each do |node|
+        puts "node=#{node.slug} queue=youtube_relay_#{node.slug} status=#{node.status} relays=#{node.max_relays}"
+      end
+    end
+
+    desc "Simula un telefono RTMP e verifica che ffmpeg parta sul worker del nodo lab"
+    task simulate: :environment do
+      abort "Esegui questo task nel container Rails (docker compose exec rails ...)" unless File.exist?("/.dockerenv") || ENV["FORCE_RELAY_SIMULATE"] == "1"
+
+      home = Streams::NodeRegistry.ensure_home_from_env!
+      lab = StreamNode.find_by(slug: ENV.fetch("STREAM_NODE_SLUG", "ingest-lab-01"))
+      lab ||= Streams::NodeProvisioner.new.provision_lab!
+      user = User.find_by!(email: ENV.fetch("E2E_USER_EMAIL", "coach@matchlivetv.test"))
+      team = user.teams.first || user.clubs.first&.teams&.first || Team.first
+      abort "Nessun team per #{user.email}" unless team
+      match = team.matches.first || team.matches.create!(opponent_name: "Relay sim")
+
+      previous_max = home.max_publishers
+      filler = nil
+      session = nil
+      ffmpeg_pid = nil
+      begin
+        home.update!(max_publishers: 1, status: "ready")
+        filler = occupy_home!(home, user, match)
+        session = build_lab_youtube_session!(lab, user, match)
+        Mediamtx::Client.for_session(session).create_path(session)
+        session.begin_connect! if session.may_begin_connect?
+        session.update!(status: "connecting") unless session.status.in?(%w[connecting live reconnecting])
+
+        puts "filler_home=#{filler.id} lab_session=#{session.id} node=#{session.stream_node.slug} queue=#{Streams::YoutubeRelay.queue_for(session)}"
+        ffmpeg_pid = spawn_test_rtmp!(session)
+        wait_for!("publisher MediaMTX", 30) { Mediamtx::PublisherOnline.active?(session.reload) }
+
+        Streams::YoutubeRelay.enqueue_ensure!(session)
+        wait_for!("owner Redis sul nodo lab", 25) do
+          owner_for(session.id) == lab.slug
+        end
+
+        owner = owner_for(session.id)
+        home_owns = relay_redis.sismember(format(Streams::YoutubeRelay::OWNED_SET, "home"), session.id)
+        puts "owner=#{owner} home_owns=#{home_owns} queue=#{Streams::YoutubeRelay.queue_for(session)}"
+        abort "FAIL: owner atteso #{lab.slug}, trovato #{owner.inspect}" unless owner == lab.slug
+        abort "FAIL: il worker home ha reclamato la sessione lab" if home_owns
+        puts "OK — ffmpeg assegnato al worker #{lab.slug}, non a home"
+      ensure
+        begin
+          Process.kill("TERM", ffmpeg_pid) if ffmpeg_pid
+        rescue Errno::ESRCH
+          nil
+        end
+        session&.update!(status: "ended", ended_at: Time.current) if session&.persisted?
+        filler&.update!(status: "ended", ended_at: Time.current) if filler&.persisted?
+        home.update!(max_publishers: previous_max)
+      end
+    end
+  end
+end
+
+def occupy_home!(home, user, match)
+  StreamSession.create!(
+    match: match,
+    user: user,
+    platform: "matchlivetv",
+    status: "live",
+    privacy_status: "unlisted",
+    stream_node: home,
+    publish_token: SecureRandom.hex(8),
+    started_at: Time.current
+  )
+end
+
+def build_lab_youtube_session!(lab, user, match)
+  StreamSession.create!(
+    match: match,
+    user: user,
+    platform: "youtube",
+    status: "idle",
+    privacy_status: "unlisted",
+    stream_node: lab,
+    publish_token: SecureRandom.hex(8),
+    stream_key: "e2e-dummy-#{SecureRandom.hex(6)}"
+  )
+end
+
+def spawn_test_rtmp!(session)
+  intake = "#{session.mediamtx_internal_rtmp_url.chomp('/')}/#{session.mediamtx_path_name}"
+  cmd = [
+    "ffmpeg", "-hide_banner", "-loglevel", "error",
+    "-re", "-f", "lavfi", "-i", "testsrc=size=1280x720:rate=30",
+    "-f", "lavfi", "-i", "sine=frequency=440:sample_rate=48000",
+    "-t", "40",
+    "-c:v", "libx264", "-preset", "veryfast", "-pix_fmt", "yuv420p", "-b:v", "1500k",
+    "-c:a", "aac", "-ac", "1", "-ar", "48000", "-b:a", "96k",
+    "-f", "flv", intake
+  ]
+  pid = Process.spawn(*cmd, %i[out err] => "/tmp/relay_sim_ffmpeg.log")
+  Process.detach(pid)
+  puts "telefono simulato pid=#{pid} rtmp=#{intake}"
+  pid
+end
+
+def wait_for!(label, timeout_sec)
+  deadline = Time.now + timeout_sec
+  loop do
+    return true if yield
+    abort "FAIL timeout: #{label}" if Time.now >= deadline
+
+    sleep 1
+  end
+end
+
+def owner_for(session_id)
+  relay_redis.get(format(Streams::YoutubeRelay::OWNER_KEY, session_id))
+end
+
+def relay_redis
+  @relay_redis ||= Redis.new(url: ENV.fetch("REDIS_URL", "redis://redis:6379/0"))
 end