panlinlin
2021-01-05 730a64be077b51add311b9d880bb23f5162253ec
src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRunner.java
@@ -4,6 +4,7 @@
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.genersoft.iot.vmp.conf.MediaServerConfig;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import com.genersoft.iot.vmp.storager.IVideoManagerStorager;
import okhttp3.*;
import org.slf4j.Logger;
@@ -13,6 +14,7 @@
import org.springframework.boot.CommandLineRunner;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
@@ -29,8 +31,17 @@
    @Autowired
    private IVideoManagerStorager storager;
    @Autowired
    private IRedisCatchStorage redisCatchStorage;
    @Value("${media.ip}")
    private String mediaIp;
    @Value("${media.wanIp}")
    private String mediaWanIp;
    @Value("${media.hookIp}")
    private String mediaHookIp;
    @Value("${media.port}")
    private int mediaPort;
@@ -38,11 +49,17 @@
    @Value("${media.secret}")
    private String mediaSecret;
    @Value("${media.streamNoneReaderDelayMS}")
    private String streamNoneReaderDelayMS;
    @Value("${sip.ip}")
    private String sipIP;
    @Value("${server.port}")
    private String serverPort;
    @Value("${media.autoConfig}")
    private boolean autoConfig;
    @Autowired
    private ZLMRESTfulUtils zlmresTfulUtils;
@@ -54,14 +71,11 @@
        MediaServerConfig mediaServerConfig = getMediaServerConfig();
        if (mediaServerConfig != null) {
            logger.info("zlm接入成功...");
            storager.updateMediaInfo(mediaServerConfig);
            logger.info("设置zlm...");
            saveZLMConfig();
            if (autoConfig) saveZLMConfig();
            mediaServerConfig = getMediaServerConfig();
            redisCatchStorage.updateMediaInfo(mediaServerConfig);
        }
    }
    public MediaServerConfig getMediaServerConfig() {
        JSONObject responseJSON = zlmresTfulUtils.getMediaServerConfig();
@@ -71,6 +85,7 @@
            if (data != null && data.size() > 0) {
                mediaServerConfig = JSON.parseObject(JSON.toJSONString(data.get(0)), MediaServerConfig.class);
                mediaServerConfig.setLocalIP(mediaIp);
                mediaServerConfig.setWanIp(StringUtils.isEmpty(mediaWanIp)? mediaIp: mediaWanIp);
            }
        } else {
            logger.error("getMediaServerConfig失败, 1s后重试");
@@ -79,22 +94,24 @@
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            getMediaServerConfig();
            mediaServerConfig = getMediaServerConfig();
        }
        return mediaServerConfig;
    }
    private void saveZLMConfig() {
        String hookIP = sipIP;
        if (mediaIp.equals(sipIP)) {
            hookIP = "127.0.0.1";
        logger.info("设置zlm...");
        if (StringUtils.isEmpty(mediaHookIp)) {
            mediaHookIp = sipIP;
        }
        String hookPrex = String.format("http://%s:%s/index/hook", hookIP, serverPort);
        String hookPrex = String.format("http://%s:%s/index/hook", mediaHookIp, serverPort);
        Map<String, Object> param = new HashMap<>();
        param.put("secret",mediaSecret);
        param.put("api.secret",mediaSecret); // -profile:v Baseline
        param.put("ffmpeg.cmd","%s -fflags nobuffer -rtsp_transport tcp -i %s -c:a aac -strict -2 -ar 44100 -ab 48k -c:v libx264  -f flv %s");
        param.put("hook.enable","1");
        param.put("hook.on_flow_report","");
        param.put("hook.on_play","");
        param.put("hook.on_http_access","");
        param.put("hook.on_publish",String.format("%s/on_publish", hookPrex));
        param.put("hook.on_record_mp4","");
@@ -103,9 +120,11 @@
        param.put("hook.on_rtsp_realm","");
        param.put("hook.on_server_started",String.format("%s/on_server_started", hookPrex));
        param.put("hook.on_shell_login",String.format("%s/on_shell_login", hookPrex));
        param.put("hook.on_stream_changed",String.format("%s/on_stream_changed", hookPrex));
        param.put("hook.on_stream_none_reader",String.format("%s/on_stream_none_reader", hookPrex));
        param.put("hook.on_stream_not_found",String.format("%s/on_stream_not_found", hookPrex));
        param.put("hook.timeoutSec","20");
        param.put("general.streamNoneReaderDelayMS",streamNoneReaderDelayMS);
        JSONObject responseJSON = zlmresTfulUtils.setServerConfig(param);