24 changed files with 641 additions and 77 deletions
@ -0,0 +1,88 @@ |
|||||
|
package org.dromara.business.config; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import org.springframework.boot.context.properties.ConfigurationProperties; |
||||
|
import org.springframework.context.annotation.Bean; |
||||
|
import org.springframework.context.annotation.Configuration; |
||||
|
import org.springframework.web.client.RestTemplate; |
||||
|
|
||||
|
/** |
||||
|
* 视频流配置类 |
||||
|
*/ |
||||
|
@Data |
||||
|
@Configuration |
||||
|
@ConfigurationProperties(prefix = "zlm") |
||||
|
public class ZlmConfig { |
||||
|
|
||||
|
/** |
||||
|
* 请求url |
||||
|
*/ |
||||
|
private String apiUrl; |
||||
|
/** |
||||
|
*应用名 |
||||
|
*/ |
||||
|
private String app; |
||||
|
/** |
||||
|
* 流 id |
||||
|
*/ |
||||
|
private String stream; |
||||
|
|
||||
|
/** |
||||
|
* 0 为 hls，1 为 mp4 |
||||
|
*/ |
||||
|
private String type; |
||||
|
/** |
||||
|
* api 操作密钥 |
||||
|
*/ |
||||
|
private String secret; |
||||
|
/** |
||||
|
* 虚拟主机 |
||||
|
*/ |
||||
|
private String vhost; |
||||
|
|
||||
|
/** |
||||
|
* mp4 录像切片时间大小,单位秒，置 0 则采用配置项 |
||||
|
*/ |
||||
|
private Integer maxSecond; |
||||
|
|
||||
|
/** |
||||
|
* 截图需要的url |
||||
|
*/ |
||||
|
private String rtmp; |
||||
|
|
||||
|
/** |
||||
|
* 开始录制url |
||||
|
*/ |
||||
|
private String startRecordUrl; |
||||
|
|
||||
|
/** |
||||
|
* 停止录制url |
||||
|
*/ |
||||
|
private String stopRecordUrl; |
||||
|
|
||||
|
/** |
||||
|
* 录制状态url |
||||
|
*/ |
||||
|
private String isRecordUrl; |
||||
|
|
||||
|
/** |
||||
|
* 设置录像速度url |
||||
|
*/ |
||||
|
private String recordSpeedUrl; |
||||
|
|
||||
|
/** |
||||
|
* 获取截图url |
||||
|
*/ |
||||
|
private String snapUrl; |
||||
|
|
||||
|
/** |
||||
|
* 设置录像流播放位置 |
||||
|
*/ |
||||
|
private String seekRecordStampUrl; |
||||
|
|
||||
|
@Bean |
||||
|
public RestTemplate restTemplate() { |
||||
|
return new RestTemplate(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
@ -0,0 +1,90 @@ |
|||||
|
package org.dromara.business.controller; |
||||
|
|
||||
|
|
||||
|
import io.swagger.v3.oas.annotations.Operation; |
||||
|
import io.swagger.v3.oas.annotations.Parameter; |
||||
|
import io.swagger.v3.oas.annotations.tags.Tag; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import org.dromara.business.service.BusinessVideoService; |
||||
|
import org.dromara.common.core.domain.R; |
||||
|
import org.springframework.validation.annotation.Validated; |
||||
|
import org.springframework.web.bind.annotation.GetMapping; |
||||
|
import org.springframework.web.bind.annotation.RequestMapping; |
||||
|
import org.springframework.web.bind.annotation.RequestParam; |
||||
|
import org.springframework.web.bind.annotation.RestController; |
||||
|
|
||||
|
/** |
||||
|
* rtmp默认端口1935, rtsp默认端口554 |
||||
|
*/ |
||||
|
@Validated |
||||
|
@RequiredArgsConstructor |
||||
|
@RestController |
||||
|
@RequestMapping("/video") |
||||
|
@Tag(name = "ZLMediaKit视频流相关操作") |
||||
|
public class BusinessVideoController { |
||||
|
|
||||
|
private final BusinessVideoService businessVideoService; |
||||
|
|
||||
|
|
||||
|
/** |
||||
|
* 开始录制 |
||||
|
*/ |
||||
|
@Operation(summary ="开始录制",description = "开始录制") |
||||
|
@GetMapping("/startRecord") |
||||
|
public R<?> startRecording() { |
||||
|
return businessVideoService.startRecording()?R.ok("启动成功!"):R.fail("启动失败!"); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 停止录制 |
||||
|
*/ |
||||
|
@Operation(summary ="停止录制",description = "停止录制") |
||||
|
@GetMapping("/stopRecord") |
||||
|
public R<?> stopRecording() { |
||||
|
return businessVideoService.stopRecording()?R.ok("停止成功!"):R.fail("停止失败!"); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 获取录制状态 |
||||
|
*/ |
||||
|
@Operation(summary ="获取录制状态",description = "获取录制状态") |
||||
|
@GetMapping("/isRecording") |
||||
|
public R<?> isRecording() { |
||||
|
return R.ok("获取录制状态!",businessVideoService.isRecording()); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 获取截图 |
||||
|
*/ |
||||
|
@Operation(summary ="获取截图",description = "获取截图",parameters = { |
||||
|
@Parameter(name = "timeoutSec",description = "截图失败超时时间"), |
||||
|
@Parameter(name = "expireSec",description = "截图的过期时间，该时间内产生的截图都会作为缓存返回") |
||||
|
}) |
||||
|
@GetMapping("/getSnap") |
||||
|
public R<?> getSnap(@RequestParam("timeoutSec") Integer timeoutSec,@RequestParam("expireSec") Integer expireSec) { |
||||
|
return R.ok("截图成功!",businessVideoService.getSnap(timeoutSec,expireSec)); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
/** |
||||
|
* 设置录像流播放位置(官方提供的接口有问题)暂时不用 |
||||
|
*/ |
||||
|
@Operation(summary ="设置录像流播放位置",description = "设置录像流播放位置",parameters = { |
||||
|
@Parameter(name = "stamp",description = "要设置的录像播放位置") |
||||
|
}) |
||||
|
@GetMapping("/seekRecordStamp") |
||||
|
public R<?> seekRecordStamp(@RequestParam("stamp") Integer stamp) { |
||||
|
return businessVideoService.seekRecordStamp(stamp)?R.ok("设置成功!"):R.fail("设置失败!"); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 设置录像速度(官方提供的接口有问题)暂时不用 |
||||
|
*/ |
||||
|
@Operation(summary ="设置录像速度",description = "设置录像速度",parameters = { |
||||
|
@Parameter(name = "speed",description = "要设置的录像倍速") |
||||
|
}) |
||||
|
@GetMapping("/setRecordSpeed") |
||||
|
public R<?> setRecordSpeed(@RequestParam("speed") Double speed) { |
||||
|
return businessVideoService.setRecordSpeed(speed)?R.ok("获取成功!"):R.fail("获取失败!"); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,16 @@ |
|||||
|
package org.dromara.business.service; |
||||
|
|
||||
|
|
||||
|
public interface BusinessVideoService { |
||||
|
Boolean startRecording(); |
||||
|
|
||||
|
Boolean stopRecording(); |
||||
|
|
||||
|
String isRecording(); |
||||
|
|
||||
|
Boolean setRecordSpeed(Double speed); |
||||
|
|
||||
|
String getSnap(Integer timeoutSec, Integer expireSec); |
||||
|
|
||||
|
Boolean seekRecordStamp(Integer stamp); |
||||
|
} |
||||
@ -0,0 +1,188 @@ |
|||||
|
package org.dromara.business.service.impl; |
||||
|
|
||||
|
import cn.hutool.core.util.ObjectUtil; |
||||
|
import com.fasterxml.jackson.databind.ObjectMapper; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.dromara.business.config.ZlmConfig; |
||||
|
import org.dromara.business.service.BusinessVideoService; |
||||
|
import org.springframework.http.ResponseEntity; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.springframework.web.client.RestTemplate; |
||||
|
import java.net.URLEncoder; |
||||
|
import java.nio.charset.StandardCharsets; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
public class BusinessVideoServiceImpl implements BusinessVideoService { |
||||
|
|
||||
|
private final ZlmConfig zlmConfig; |
||||
|
private static final ObjectMapper objectMapper = new ObjectMapper(); |
||||
|
private final RestTemplate restTemplate; |
||||
|
|
||||
|
|
||||
|
@Override |
||||
|
public Boolean startRecording() { |
||||
|
|
||||
|
String url = String.format("%s%s?app=%s&stream=%s&type=%s&vhost=%s&secret=%s", |
||||
|
zlmConfig.getApiUrl(), |
||||
|
zlmConfig.getStartRecordUrl(), |
||||
|
zlmConfig.getApp(), |
||||
|
zlmConfig.getStream(), |
||||
|
zlmConfig.getType(), |
||||
|
zlmConfig.getVhost(), |
||||
|
zlmConfig.getSecret()); |
||||
|
|
||||
|
//是否需要切片视频
|
||||
|
if (ObjectUtil.isNotEmpty(zlmConfig.getMaxSecond())){ |
||||
|
url = url + "&max_second=" + zlmConfig.getMaxSecond(); |
||||
|
} |
||||
|
|
||||
|
try { |
||||
|
ResponseEntity<String> response = restTemplate.getForEntity(url, String.class); |
||||
|
if (response.getStatusCode().is2xxSuccessful()) { |
||||
|
Map<String, Object> result = objectMapper.readValue(response.getBody(), Map.class); |
||||
|
log.info("录制启动成功: {}", result); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("请求异常: {}", e.getMessage(), e); |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Boolean stopRecording() { |
||||
|
String url = String.format("%s%s?app=%s&stream=%s&type=%s&vhost=%s&secret=%s", |
||||
|
zlmConfig.getApiUrl(), |
||||
|
zlmConfig.getStopRecordUrl(), |
||||
|
zlmConfig.getApp(), |
||||
|
zlmConfig.getStream(), |
||||
|
zlmConfig.getType(), |
||||
|
zlmConfig.getVhost(), |
||||
|
zlmConfig.getSecret()); |
||||
|
|
||||
|
try { |
||||
|
ResponseEntity<String> response = restTemplate.getForEntity(url, String.class); |
||||
|
if (response.getStatusCode().is2xxSuccessful()) { |
||||
|
Map<String, Object> result = objectMapper.readValue(response.getBody(), Map.class); |
||||
|
log.info("录制停止成功: {}", result); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("请求异常: {}", e.getMessage(), e); |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String isRecording() { |
||||
|
String result = null; |
||||
|
|
||||
|
String url = String.format("%s%s?app=%s&stream=%s&type=%s&vhost=%s&secret=%s", |
||||
|
zlmConfig.getApiUrl(), |
||||
|
zlmConfig.getIsRecordUrl(), |
||||
|
zlmConfig.getApp(), |
||||
|
zlmConfig.getStream(), |
||||
|
zlmConfig.getType(), |
||||
|
zlmConfig.getVhost(), |
||||
|
zlmConfig.getSecret()); |
||||
|
|
||||
|
try { |
||||
|
ResponseEntity<String> response = restTemplate.getForEntity(url, String.class); |
||||
|
if (response.getStatusCode().is2xxSuccessful()) { |
||||
|
Map<String, Object> resultMap = objectMapper.readValue(response.getBody(), Map.class); |
||||
|
result = resultMap.get("status") + ""; |
||||
|
log.info("获取录制状态成功: {}", resultMap); |
||||
|
} |
||||
|
|
||||
|
return result; |
||||
|
} catch (Exception e) { |
||||
|
log.error("请求异常: {}", e.getMessage(), e); |
||||
|
} |
||||
|
|
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Boolean setRecordSpeed(Double speed) { |
||||
|
String url = String.format("%s%s?app=%s&stream=%s&type=%s&vhost=%s&secret=%s&speed=%s", |
||||
|
zlmConfig.getApiUrl(), |
||||
|
zlmConfig.getRecordSpeedUrl(), |
||||
|
zlmConfig.getApp(), |
||||
|
zlmConfig.getStream(), |
||||
|
zlmConfig.getType(), |
||||
|
zlmConfig.getVhost(), |
||||
|
zlmConfig.getSecret(), |
||||
|
speed); |
||||
|
|
||||
|
try { |
||||
|
ResponseEntity<String> response = restTemplate.getForEntity(url, String.class); |
||||
|
if (response.getStatusCode().is2xxSuccessful()) { |
||||
|
Map<String, Object> result = objectMapper.readValue(response.getBody(), Map.class); |
||||
|
log.info("设置录制速度成功: {}", result); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("请求异常: {}", e.getMessage(), e); |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String getSnap(Integer timeoutSec, Integer expireSec) { |
||||
|
String url = String.format("rtmp://%s/%s/%s",zlmConfig.getRtmp(),zlmConfig.getApp(),zlmConfig.getStream()); |
||||
|
String apiUrl = String.format("%s%s?secret=%s&url=%s&timeout_sec=%s&expire_sec=%s", |
||||
|
zlmConfig.getApiUrl(), |
||||
|
zlmConfig.getSnapUrl(), |
||||
|
zlmConfig.getSecret(), |
||||
|
url, |
||||
|
timeoutSec, |
||||
|
expireSec); |
||||
|
|
||||
|
try { |
||||
|
ResponseEntity<String> response = restTemplate.getForEntity(apiUrl, String.class); |
||||
|
if (response.getStatusCode().is2xxSuccessful()) { |
||||
|
log.info("获取截图成功: {}", response.getBody()); |
||||
|
} |
||||
|
|
||||
|
return response.getBody(); |
||||
|
} catch (Exception e) { |
||||
|
log.error("请求异常: {}", e.getMessage(), e); |
||||
|
} |
||||
|
|
||||
|
return ""; |
||||
|
} |
||||
|
|
||||
|
|
||||
|
@Override |
||||
|
public Boolean seekRecordStamp(Integer stamp) { |
||||
|
String url = String.format("%s%s?app=%s&stream=%s&vhost=%s&secret=%s&stamp=%s", |
||||
|
zlmConfig.getApiUrl(), |
||||
|
zlmConfig.getSeekRecordStampUrl(), |
||||
|
zlmConfig.getApp(), |
||||
|
zlmConfig.getStream(), |
||||
|
zlmConfig.getVhost(), |
||||
|
zlmConfig.getSecret(), |
||||
|
stamp); |
||||
|
|
||||
|
try { |
||||
|
ResponseEntity<String> response = restTemplate.getForEntity(url, String.class); |
||||
|
if (response.getStatusCode().is2xxSuccessful()) { |
||||
|
Map<String, Object> result = objectMapper.readValue(response.getBody(), Map.class); |
||||
|
log.info("设置录像流播放位置成功: {}", result); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("请求异常: {}", e.getMessage(), e); |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
return true; |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue