org.jetlinks.zlmedia.restful.ZLMediaConfigs Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zlmedia4j-restful Show documentation
The newest version!
package org.jetlinks.zlmedia.restful;

import lombok.Getter;
import lombok.Setter;
import org.jetlinks.zlmedia.ZLMediaOperations;
import org.springframework.util.StringUtils;

import java.io.File;
import java.util.*;

@Getter
@Setter
public class ZLMediaConfigs {
    private static final List ALL_HOOKS =
        Arrays.asList(
            "on_flow_report",
            "on_http_access",
            "on_play",
            "on_publish",
            "on_record_mp4",
            "on_record_ts",
            "on_rtsp_auth",
            "on_rtsp_realm",
            "on_shell_login",
            "on_stream_changed",
            "on_stream_none_reader",
            "on_server_started",
            "on_server_exited",
            "on_server_keepalive",
            "on_send_rtp_stopped",
            "on_rtp_server_timeout",
            "on_stream_not_found"
        );

    /**
     * 数据存储路径,截图,录像等数据存储到此目录.
     */
    private String dataPath = "data/zlmedia";

    /**
     * hook 通知的根地址,如: http://127.0.0.1:8080/zlmedia/hook/
     *
     * @see ZLMediaOperations#opsForHook()
     * @see org.jetlinks.zlmedia.restful.RestfulHookOperations#fireEvent(String, String)
     */
    private String hookBasePath;

    /**
     * hook通知参数
     */
    private String hookParams = "";

    /**
     * hook通知的事件类型,默认全部
     */
    private Set hooks = new HashSet<>(ALL_HOOKS);

    /**
     * 服务ID,用于区分不同的服务,在集群通知时可根据参数中的serverId进行集群调用.
     */
    private String serverId;

    /**
     * 端口信息
     */
    private Ports ports = new Ports();

    /**
     * 其他自定义配置
     */
    private Map others;

    /**
     * 自定义密钥
     */
    private String secret = UUID.randomUUID().toString().replace("-", "");

    public Map createConfigs() {
        Map configs = new HashMap<>();

        configs.put("api.secret", secret);

        if (others != null) {
            configs.putAll(others);
        }
        //服务ID
        if (serverId != null) {
            configs.put("general.mediaServerId", serverId);
        }
        //hook配置
        if (hookBasePath != null) {
            configs.put("hook.enable", "1");
            String prefix = hookBasePath;
            if (!prefix.endsWith("/")) {
                prefix = prefix + "/";
            }
            for (String hook : hooks) {
                String url = prefix + hook;
                if (StringUtils.hasText(hookParams)) {
                    url = url + "?" + hookParams;
                }
                configs.put("hook." + hook, url);
            }
        }

        ports.applyConfig(configs);

        //存储相关配置
        if (dataPath != null) {
            File file = new File(dataPath);
            String path = file.getAbsolutePath();
            configs.put("api.downloadRoot", path);
            configs.put("http.rootPath", path);
            configs.put("api.snapRoot", path + "/snapshot/");
            configs.put("api.defaultSnap", path + "/snapshot.png");
            configs.put("protocol.mp4_save_path", path);
            configs.put("protocol.hls_save_path", path);
        }

        return configs;
    }

    public void setConfig(String key, String value) {
        if (ports.setConfig(key, value)) {
            return;
        }
        switch (key) {
            case "api.secret":
                secret = value;
                break;
            case "general.mediaServerId":
                serverId = value;
                break;
            default:
                if (others != null) {
                    others.put(key, value);
                }
        }
    }

    @Getter
    @Setter
    public static class Ports {
        private int http = 8188;
        private int rtp = 10000;
        private int[] rtpRange = new int[]{30000, 35000};
        private int rtsp = 554;
        private int rtmp = 1935;
        private int rtc = 8000;
        private int srt = 9000;

        public boolean setConfig(String key, String value) {
            switch (key) {
                case "rtsp.port":
                    setRtsp(Integer.parseInt(value));
                    return true;
                case "rtmp.port":
                    setRtmp(Integer.parseInt(value));
                    return true;
                case "http.port":
                    setHttp(Integer.parseInt(value));
                    return true;
                case "rtp_proxy.port":
                    setRtp(Integer.parseInt(value));
                    return true;
                case "rtp_proxy.port_range":
                    String[] range = value.split("-");
                    setRtpRange(new int[]{Integer.parseInt(range[0]), Integer.parseInt(range[1])});
                    return true;
                case "rtc.port":
                    setRtc(Integer.parseInt(value));
                    return true;
                case "srt.port":
                    setSrt(Integer.parseInt(value));
                    return true;
            }
            return false;
        }

        public void applyConfig(Map config) {
            config.put("http.port", String.valueOf(http));

            config.put("rtp_proxy.port", String.valueOf(rtp));
            config.put("rtp_proxy.port_range", rtpRange[0] + "-" + rtpRange[1]);

            config.put("rtsp.port", String.valueOf(rtsp));
            config.put("rtmp.port", String.valueOf(rtmp));

            config.put("rtc.port", String.valueOf(rtc));
            config.put("rtc.tcpPort", String.valueOf(rtc));

            config.put("srt.port", String.valueOf(srt));
        }
    }

}