eminuxandCursor
4c5af99efa
Senza questo il telefono (mono 48k) veniva rifiutato sui nodi Hetzner e lo warm spare rubava le sessioni a home. Co-authored-by: Cursor <cursoragent@cursor.com>
67 lines
2.2 KiB
Ruby
67 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Streams
|
|
# Assegna un StreamNode a una nuova sessione.
|
|
# Preferisce home finché ha slot; poi least-loaded tra i nodi overflow ready.
|
|
# Garantisce il nodo "home" derivato dagli ENV MediaMTX attuali.
|
|
class NodeRegistry
|
|
class NoCapacityError < StandardError; end
|
|
|
|
HOME_SLUG = "home"
|
|
|
|
class << self
|
|
def ensure_home_from_env!
|
|
StreamNode.find_or_initialize_by(slug: HOME_SLUG).tap do |node|
|
|
attrs = {
|
|
hostname: home_hostname,
|
|
role: "home",
|
|
provider: "local",
|
|
rtmp_base_url: MatchLiveTv.mediamtx_rtmp_url,
|
|
hls_base_url: MatchLiveTv.hls_public_url,
|
|
api_base_url: MatchLiveTv.mediamtx_api_url,
|
|
internal_rtmp_url: ENV.fetch("MEDIAMTX_INTERNAL_RTMP_URL", "rtmp://mediamtx:1935"),
|
|
internal_hls_url: ENV.fetch("MEDIAMTX_HLS_URL", "http://mediamtx:8888"),
|
|
max_publishers: ENV.fetch("STREAM_NODE_HOME_MAX_PUBLISHERS", "6").to_i,
|
|
max_relays: ENV.fetch("STREAM_NODE_HOME_MAX_RELAYS", "6").to_i
|
|
}
|
|
# Non sovrascrivere drain/offline/error (ops) a ogni allocate!
|
|
attrs[:status] = "ready" if node.new_record? || !%w[draining offline error].include?(node.status)
|
|
node.assign_attributes(attrs)
|
|
node.save!
|
|
end
|
|
end
|
|
|
|
def allocate!
|
|
ensure_home_from_env!
|
|
|
|
# Overflow: riempi prima home; i nodi cloud/lab sono solo quando home è pieno
|
|
# (altrimenti lo warm spare ruberebbe tutte le sessioni).
|
|
home = StreamNode.find_by(slug: HOME_SLUG)
|
|
return home if home&.allocatable?
|
|
|
|
node = StreamNode.ready
|
|
.where.not(slug: HOME_SLUG)
|
|
.to_a
|
|
.select(&:allocatable?)
|
|
.min_by { |n| [n.active_publishers, n.slug] }
|
|
|
|
raise NoCapacityError, "Nessun nodo streaming con slot liberi" if node.nil?
|
|
|
|
node
|
|
end
|
|
|
|
private
|
|
|
|
def home_hostname
|
|
ENV["STREAM_NODE_HOME_HOSTNAME"].presence ||
|
|
begin
|
|
uri = URI.parse(MatchLiveTv.mediamtx_rtmp_url.sub(/\Artmps?:\/\//, "http://"))
|
|
uri.host.presence
|
|
rescue URI::InvalidURIError
|
|
nil
|
|
end || "home"
|
|
end
|
|
end
|
|
end
|
|
end
|