Replace master branch with local files
This commit is contained in:
commit
875a53f499
60 changed files with 21637 additions and 0 deletions
84
openresty/lua/stream_monitor.lua
Normal file
84
openresty/lua/stream_monitor.lua
Normal file
|
|
@ -0,0 +1,84 @@
|
|||
|
||||
local redis_helper = require "redis_helper"
|
||||
local cjson = require "cjson"
|
||||
|
||||
-- Safely load the http module
|
||||
local has_http, http = pcall(require, "resty.http")
|
||||
|
||||
local function disconnect_stream(stream_key)
|
||||
if not has_http then
|
||||
ngx.log(ngx.WARN, "resty.http module not available, skipping stream disconnection")
|
||||
return
|
||||
end
|
||||
|
||||
local httpc = http.new()
|
||||
local ome_url = os.getenv("OME_URL") or "http://ovenmediaengine:8081"
|
||||
local ome_token = os.getenv("OME_API_TOKEN") or "your-api-token"
|
||||
|
||||
-- Get active streams from OME
|
||||
local res, err = httpc:request_uri(ome_url .. "/v1/vhosts/default/apps/app/streams", {
|
||||
method = "GET",
|
||||
headers = {
|
||||
["Authorization"] = "Bearer " .. ome_token,
|
||||
["Content-Type"] = "application/json"
|
||||
}
|
||||
})
|
||||
|
||||
if not res then
|
||||
ngx.log(ngx.ERR, "Failed to get streams: ", err)
|
||||
return
|
||||
end
|
||||
|
||||
local ok, data = pcall(cjson.decode, res.body)
|
||||
if ok and data and data.response and data.response.streams then
|
||||
for _, stream in ipairs(data.response.streams) do
|
||||
if stream.name == stream_key then
|
||||
-- Disconnect the stream
|
||||
local del_res, del_err = httpc:request_uri(
|
||||
ome_url .. "/v1/vhosts/default/apps/app/streams/" .. stream.name,
|
||||
{
|
||||
method = "DELETE",
|
||||
headers = {
|
||||
["Authorization"] = "Bearer " .. ome_token
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
if del_res and del_res.status == 200 then
|
||||
ngx.log(ngx.INFO, "Disconnected stream: ", stream_key)
|
||||
else
|
||||
ngx.log(ngx.ERR, "Failed to disconnect stream: ", stream_key)
|
||||
end
|
||||
|
||||
break
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function monitor_streams()
|
||||
-- Check if Redis is available
|
||||
local ok, keys = pcall(redis_helper.get_streams_to_disconnect)
|
||||
if not ok then
|
||||
ngx.log(ngx.WARN, "Redis not available yet")
|
||||
return
|
||||
end
|
||||
|
||||
for _, key in ipairs(keys or {}) do
|
||||
disconnect_stream(key)
|
||||
redis_helper.remove_stream_from_disconnect(key)
|
||||
end
|
||||
end
|
||||
|
||||
-- Start monitoring in a timer with error handling
|
||||
local function start_monitoring()
|
||||
local ok, err = ngx.timer.every(1, monitor_streams)
|
||||
if not ok then
|
||||
ngx.log(ngx.ERR, "Failed to create timer: ", err)
|
||||
else
|
||||
ngx.log(ngx.INFO, "Stream monitor started")
|
||||
end
|
||||
end
|
||||
|
||||
-- Delay start to ensure services are ready
|
||||
ngx.timer.at(5, start_monitoring)
|
||||
Loading…
Add table
Add a link
Reference in a new issue