648540858
4 years ago
13 changed files with 234 additions and 23 deletions
@ -0,0 +1,152 @@ |
|||
package com.genersoft.iot.vmp.media.zlm; |
|||
|
|||
import com.alibaba.fastjson.JSON; |
|||
import com.alibaba.fastjson.JSONArray; |
|||
import com.alibaba.fastjson.JSONObject; |
|||
import com.genersoft.iot.vmp.conf.MediaServerConfig; |
|||
import com.genersoft.iot.vmp.storager.IVideoManagerStorager; |
|||
import okhttp3.*; |
|||
import org.slf4j.Logger; |
|||
import org.slf4j.LoggerFactory; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.beans.factory.annotation.Value; |
|||
import org.springframework.boot.CommandLineRunner; |
|||
import org.springframework.core.annotation.Order; |
|||
import org.springframework.stereotype.Component; |
|||
|
|||
import java.io.IOException; |
|||
import java.io.UnsupportedEncodingException; |
|||
import java.net.URLEncoder; |
|||
|
|||
@Component |
|||
@Order(value=1) |
|||
public class ZLMRunner implements CommandLineRunner { |
|||
|
|||
private final static Logger logger = LoggerFactory.getLogger(ZLMRunner.class); |
|||
|
|||
@Autowired |
|||
private IVideoManagerStorager storager; |
|||
|
|||
@Value("${media.ip}") |
|||
private String mediaIp; |
|||
|
|||
@Value("${media.port}") |
|||
private int mediaPort; |
|||
|
|||
@Value("${media.secret}") |
|||
private String mediaSecret; |
|||
|
|||
@Value("${sip.ip}") |
|||
private String sipIP; |
|||
|
|||
@Value("${server.port}") |
|||
private String serverPort; |
|||
|
|||
@Override |
|||
public void run(String... strings) throws Exception { |
|||
// 获取zlm信息
|
|||
logger.info("等待zlm接入..."); |
|||
MediaServerConfig mediaServerConfig = getMediaServerConfig(); |
|||
if (mediaServerConfig != null) { |
|||
logger.info("zlm接入成功..."); |
|||
storager.updateMediaInfo(mediaServerConfig); |
|||
logger.info("设置zlm..."); |
|||
saveZLMConfig(); |
|||
|
|||
} |
|||
} |
|||
|
|||
|
|||
|
|||
public MediaServerConfig getMediaServerConfig() { |
|||
MediaServerConfig mediaServerConfig = null; |
|||
OkHttpClient client = new OkHttpClient(); |
|||
String url = String.format("http://%s:%s/index/api/getServerConfig?secret=%s", mediaIp, mediaPort, mediaSecret); |
|||
//创建一个Request
|
|||
Request request = new Request.Builder() |
|||
.get() |
|||
.url(url) |
|||
.build(); |
|||
//通过client发起请求
|
|||
final Call call = client.newCall(request); |
|||
//执行同步请求，获取Response对象
|
|||
Response response = null; |
|||
try { |
|||
response = call.execute(); |
|||
if (response.isSuccessful()) { |
|||
String responseStr = response.body().string(); |
|||
if (responseStr != null) { |
|||
JSONObject responseJSON = JSON.parseObject(responseStr); |
|||
JSONArray data = responseJSON.getJSONArray("data"); |
|||
if (data != null && data.size() > 0) { |
|||
mediaServerConfig = JSON.parseObject(JSON.toJSONString(data.get(0)), MediaServerConfig.class); |
|||
mediaServerConfig.setLocalIP(mediaIp); |
|||
} |
|||
} |
|||
}else { |
|||
logger.error("getMediaServerConfig失败, 1s后重试"); |
|||
Thread.sleep(1000); |
|||
getMediaServerConfig(); |
|||
} |
|||
} catch (IOException e) { |
|||
e.printStackTrace(); |
|||
} catch (InterruptedException e) { |
|||
e.printStackTrace(); |
|||
} |
|||
|
|||
return mediaServerConfig; |
|||
} |
|||
|
|||
private void saveZLMConfig() { |
|||
String hookIP = sipIP; |
|||
if (mediaIp.equals(sipIP)) { |
|||
hookIP = "127.0.0.1"; |
|||
} |
|||
OkHttpClient client = new OkHttpClient(); |
|||
String url = String.format("http://%s:%s/index/api/setServerConfig", mediaIp, mediaPort); |
|||
String hookPrex = String.format("http://%s:%s/index/hook", hookIP, serverPort); |
|||
|
|||
RequestBody body = new FormBody.Builder() |
|||
.add("secret",mediaSecret) |
|||
.add("hook.enable","1") |
|||
.add("hook.on_flow_report","") |
|||
.add("hook.on_http_access","") |
|||
.add("hook.on_publish",String.format("%s/on_publish", hookPrex)) |
|||
.add("hook.on_record_mp4","") |
|||
.add("hook.on_record_ts","") |
|||
.add("hook.on_rtsp_auth","") |
|||
.add("hook.on_rtsp_realm","") |
|||
.add("hook.on_server_started",String.format("%s/on_server_started", hookPrex)) |
|||
.add("hook.on_shell_login",String.format("%s/on_shell_login", hookPrex)) |
|||
.add("hook.on_stream_none_reader",String.format("%s/on_stream_none_reader", hookPrex)) |
|||
.add("hook.on_stream_not_found",String.format("%s/on_stream_not_found", hookPrex)) |
|||
.add("hook.timeoutSec","20") |
|||
.build(); |
|||
|
|||
Request request = new Request.Builder() |
|||
.post(body) |
|||
.url(url) |
|||
.build(); |
|||
client.newCall(request).enqueue(new Callback() { |
|||
@Override |
|||
public void onFailure(Call call, IOException e) { |
|||
logger.error("saveZLMConfig ",e); |
|||
} |
|||
@Override |
|||
public void onResponse(Call call, Response response) throws IOException { |
|||
if (response.isSuccessful()) { |
|||
String responseStr = response.body().string(); |
|||
if (responseStr != null) { |
|||
JSONObject responseJSON = JSON.parseObject(responseStr); |
|||
if (responseJSON.getInteger("code") == 0) { |
|||
logger.info("设置zlm成功"); |
|||
}else { |
|||
logger.info("设置zlm失败: " + responseJSON.getString("msg")); |
|||
} |
|||
} |
|||
} |
|||
|
|||
} |
|||
}); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue