package com.ard.zlm.api.domain;
|
|
import com.alibaba.fastjson2.JSONArray;
|
import com.alibaba.fastjson2.JSONObject;
|
import com.ard.zlm.api.hook.OnStreamArriveABLHookParam;
|
import com.ard.zlm.api.hook.OnStreamChangedHookParam;
|
import com.ard.zlm.api.hook.OriginType;
|
import com.ard.zlm.api.utils.MediaServerUtils;
|
import lombok.Data;
|
import org.springframework.util.ObjectUtils;
|
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 视频信息
|
*/
|
@Data
|
public class MediaInfo {
|
/**应用名**/
|
private String app;
|
/**流ID**/
|
private String stream;
|
/**流媒体节点**/
|
private ZlmMediaServer mediaServer;
|
/**协议**/
|
private String schema;
|
|
/**观看人数**/
|
private Integer readerCount;
|
/**视频编码类型**/
|
private String videoCodec;
|
/**视频宽度**/
|
private Integer width;
|
/**视频高度**/
|
private Integer height;
|
/**FPS**/
|
private Integer fps;
|
/**丢包率**/
|
private Integer loss;
|
/**音频编码类型**/
|
private String audioCodec;
|
/**音频通道数**/
|
private Integer audioChannels;
|
/**音频采样率**/
|
private Integer audioSampleRate;
|
/**时长**/
|
private Long duration;
|
/**在线**/
|
private Boolean online;
|
/**unknown = 0,rtmp_push=1,rtsp_push=2,rtp_push=3,pull=4,ffmpeg_pull=5,mp4_vod=6,device_chn=7,rtc_push=8**/
|
private Integer originType;
|
/**originType的文本描述**/
|
private String originTypeStr;
|
/**产生流的源流地址**/
|
private String originUrl;
|
/**存活时间，单位秒**/
|
private Long aliveSecond;
|
/**数据产生速度，单位byte/s**/
|
private Long bytesSpeed;
|
/**鉴权参数**/
|
private String callId;
|
/**额外参数**/
|
private Map<String, String> paramMap;
|
/**服务ID**/
|
private String serverId;
|
|
|
public static MediaInfo getInstance(JSONObject jsonObject, ZlmMediaServer mediaServer, String serverId) {
|
MediaInfo mediaInfo = new MediaInfo();
|
mediaInfo.setMediaServer(mediaServer);
|
mediaInfo.setServerId(serverId);
|
String app = jsonObject.getString("app");
|
mediaInfo.setApp(app);
|
String stream = jsonObject.getString("stream");
|
mediaInfo.setStream(stream);
|
String schema = jsonObject.getString("schema");
|
mediaInfo.setSchema(schema);
|
Integer totalReaderCount = jsonObject.getInteger("totalReaderCount");
|
Boolean online = jsonObject.getBoolean("online");
|
Integer originType = jsonObject.getInteger("originType");
|
String originUrl = jsonObject.getString("originUrl");
|
String originTypeStr = jsonObject.getString("originTypeStr");
|
Long aliveSecond = jsonObject.getLong("aliveSecond");
|
String params = jsonObject.getString("params");
|
Long bytesSpeed = jsonObject.getLong("bytesSpeed");
|
if (totalReaderCount != null) {
|
mediaInfo.setReaderCount(totalReaderCount);
|
} else {
|
mediaInfo.setReaderCount(0);
|
}
|
if (online != null) {
|
mediaInfo.setOnline(online);
|
}
|
if (originType != null) {
|
mediaInfo.setOriginType(originType);
|
}
|
if (originTypeStr != null) {
|
mediaInfo.setOriginTypeStr(originTypeStr);
|
}
|
|
if (aliveSecond != null) {
|
mediaInfo.setAliveSecond(aliveSecond);
|
}
|
if (bytesSpeed != null) {
|
mediaInfo.setBytesSpeed(bytesSpeed);
|
}
|
if (params != null) {
|
mediaInfo.setParamMap(MediaServerUtils.urlParamToMap(params));
|
if (mediaInfo.getCallId() == null) {
|
mediaInfo.setCallId(mediaInfo.getParamMap().get("callId"));
|
}
|
}
|
JSONArray jsonArray = jsonObject.getJSONArray("tracks");
|
if (!ObjectUtils.isEmpty(jsonArray)) {
|
for (int i = 0; i < jsonArray.size(); i++) {
|
JSONObject trackJson = jsonArray.getJSONObject(i);
|
Integer channels = trackJson.getInteger("channels");
|
Integer codecId = trackJson.getInteger("codec_id");
|
Integer codecType = trackJson.getInteger("codec_type");
|
Integer sampleRate = trackJson.getInteger("sample_rate");
|
Integer height = trackJson.getInteger("height");
|
Integer width = trackJson.getInteger("width");
|
Integer fps = trackJson.getInteger("fps");
|
Integer loss = trackJson.getInteger("loss");
|
Integer frames = trackJson.getInteger("frames");
|
Long keyFrames = trackJson.getLongValue("key_frames");
|
Integer gop_interval_ms = trackJson.getInteger("gop_interval_ms");
|
Long gop_size = trackJson.getLongValue("gop_size");
|
|
Long duration = trackJson.getLongValue("duration");
|
if (channels != null) {
|
mediaInfo.setAudioChannels(channels);
|
}
|
if (sampleRate != null) {
|
mediaInfo.setAudioSampleRate(sampleRate);
|
}
|
if (height != null) {
|
mediaInfo.setHeight(height);
|
}
|
if (width != null) {
|
mediaInfo.setWidth(width);
|
}
|
if (fps != null) {
|
mediaInfo.setFps(fps);
|
}
|
if (loss != null) {
|
mediaInfo.setLoss(loss);
|
}
|
if (duration > 0L) {
|
mediaInfo.setDuration(duration);
|
}
|
if (codecId != null) {
|
switch (codecId) {
|
case 0:
|
mediaInfo.setVideoCodec("H264");
|
break;
|
case 1:
|
mediaInfo.setVideoCodec("H265");
|
break;
|
case 2:
|
mediaInfo.setAudioCodec("AAC");
|
break;
|
case 3:
|
mediaInfo.setAudioCodec("G711A");
|
break;
|
case 4:
|
mediaInfo.setAudioCodec("G711U");
|
break;
|
}
|
}
|
}
|
}
|
return mediaInfo;
|
}
|
|
public static MediaInfo getInstance(OnStreamChangedHookParam param, ZlmMediaServer mediaServer, String serverId) {
|
|
MediaInfo mediaInfo = new MediaInfo();
|
mediaInfo.setApp(param.getApp());
|
mediaInfo.setStream(param.getStream());
|
mediaInfo.setSchema(param.getSchema());
|
mediaInfo.setMediaServer(mediaServer);
|
mediaInfo.setReaderCount(param.getTotalReaderCount());
|
mediaInfo.setOnline(param.isRegist());
|
mediaInfo.setOriginType(param.getOriginType());
|
mediaInfo.setOriginTypeStr(param.getOriginTypeStr());
|
mediaInfo.setOriginUrl(param.getOriginUrl());
|
mediaInfo.setOriginUrl(param.getOriginUrl());
|
mediaInfo.setAliveSecond(param.getAliveSecond());
|
mediaInfo.setBytesSpeed(param.getBytesSpeed());
|
mediaInfo.setParamMap(param.getParamMap());
|
if (mediaInfo.getCallId() == null) {
|
mediaInfo.setCallId(param.getParamMap().get("callId"));
|
}
|
mediaInfo.setServerId(serverId);
|
List<OnStreamChangedHookParam.MediaTrack> tracks = param.getTracks();
|
if (tracks == null || tracks.isEmpty()) {
|
return mediaInfo;
|
}
|
for (OnStreamChangedHookParam.MediaTrack mediaTrack : tracks) {
|
switch (mediaTrack.getCodec_id()) {
|
case 0:
|
mediaInfo.setVideoCodec("H264");
|
break;
|
case 1:
|
mediaInfo.setVideoCodec("H265");
|
break;
|
case 2:
|
mediaInfo.setAudioCodec("AAC");
|
break;
|
case 3:
|
mediaInfo.setAudioCodec("G711A");
|
break;
|
case 4:
|
mediaInfo.setAudioCodec("G711U");
|
break;
|
}
|
if (mediaTrack.getSample_rate() > 0) {
|
mediaInfo.setAudioSampleRate(mediaTrack.getSample_rate());
|
}
|
if (mediaTrack.getChannels() > 0) {
|
mediaInfo.setAudioChannels(mediaTrack.getChannels());
|
}
|
if (mediaTrack.getHeight() > 0) {
|
mediaInfo.setHeight(mediaTrack.getHeight());
|
}
|
if (mediaTrack.getWidth() > 0) {
|
mediaInfo.setWidth(mediaTrack.getWidth());
|
}
|
}
|
return mediaInfo;
|
}
|
|
public static MediaInfo getInstance(OnStreamArriveABLHookParam param, ZlmMediaServer mediaServer) {
|
|
MediaInfo mediaInfo = new MediaInfo();
|
mediaInfo.setApp(param.getApp());
|
mediaInfo.setStream(param.getStream());
|
mediaInfo.setMediaServer(mediaServer);
|
mediaInfo.setReaderCount(param.getReaderCount());
|
mediaInfo.setOnline(true);
|
mediaInfo.setVideoCodec(param.getVideoCodec());
|
switch (param.getNetworkType()) {
|
case 21:
|
mediaInfo.setOriginType(OriginType.RTMP_PUSH.ordinal());
|
break;
|
case 23:
|
mediaInfo.setOriginType(OriginType.RTSP_PUSH.ordinal());
|
break;
|
case 30:
|
case 31:
|
case 32:
|
case 33:
|
mediaInfo.setOriginType(OriginType.PULL.ordinal());
|
break;
|
default:
|
mediaInfo.setOriginType(OriginType.UNKNOWN.ordinal());
|
break;
|
|
}
|
mediaInfo.setWidth(param.getWidth());
|
mediaInfo.setHeight(param.getHeight());
|
mediaInfo.setAudioCodec(param.getAudioCodec());
|
mediaInfo.setAudioChannels(param.getAudioChannels());
|
mediaInfo.setAudioSampleRate(param.getAudioSampleRate());
|
|
return mediaInfo;
|
}
|
}
|
