package com.genersoft.iot.vmp.vmanager.streamProxy; 
 | 
  
 | 
import com.alibaba.fastjson.JSONObject; 
 | 
import com.genersoft.iot.vmp.common.StreamInfo; 
 | 
import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; 
 | 
import com.genersoft.iot.vmp.media.zlm.dto.StreamProxyItem; 
 | 
import com.genersoft.iot.vmp.service.IMediaServerService; 
 | 
import com.genersoft.iot.vmp.service.IMediaService; 
 | 
import com.genersoft.iot.vmp.storager.IRedisCatchStorage; 
 | 
import com.genersoft.iot.vmp.service.IStreamProxyService; 
 | 
import com.genersoft.iot.vmp.vmanager.bean.WVPResult; 
 | 
import com.github.pagehelper.PageInfo; 
 | 
import io.swagger.annotations.Api; 
 | 
import io.swagger.annotations.ApiImplicitParam; 
 | 
import io.swagger.annotations.ApiImplicitParams; 
 | 
import io.swagger.annotations.ApiOperation; 
 | 
import org.slf4j.Logger; 
 | 
import org.slf4j.LoggerFactory; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
import org.springframework.stereotype.Controller; 
 | 
import org.springframework.util.StringUtils; 
 | 
import org.springframework.web.bind.annotation.*; 
 | 
  
 | 
@SuppressWarnings("rawtypes") 
 | 
/** 
 | 
 * 拉流代理接口 
 | 
 */ 
 | 
@Api(tags = "拉流代理") 
 | 
@Controller 
 | 
@CrossOrigin 
 | 
@RequestMapping(value = "/api/proxy") 
 | 
public class StreamProxyController { 
 | 
  
 | 
    private final static Logger logger = LoggerFactory.getLogger(StreamProxyController.class); 
 | 
  
 | 
    @Autowired 
 | 
    private IRedisCatchStorage redisCatchStorage; 
 | 
  
 | 
  
 | 
    @Autowired 
 | 
    private IMediaServerService mediaServerService; 
 | 
  
 | 
    @Autowired 
 | 
    private IStreamProxyService streamProxyService; 
 | 
  
 | 
  
 | 
    @ApiOperation("分页查询流代理") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name="page", value = "当前页", required = true, dataTypeClass = Integer.class), 
 | 
            @ApiImplicitParam(name="count", value = "每页查询数量", required = true, dataTypeClass = Integer.class), 
 | 
            @ApiImplicitParam(name="query", value = "查询内容", dataTypeClass = String.class), 
 | 
            @ApiImplicitParam(name="online", value = "是否在线", dataTypeClass = Boolean.class), 
 | 
    }) 
 | 
    @GetMapping(value = "/list") 
 | 
    @ResponseBody 
 | 
    public PageInfo<StreamProxyItem> list(@RequestParam(required = false)Integer page, 
 | 
                                          @RequestParam(required = false)Integer count, 
 | 
                                          @RequestParam(required = false)String query, 
 | 
                                          @RequestParam(required = false)Boolean online ){ 
 | 
  
 | 
        return streamProxyService.getAll(page, count); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("保存代理") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name = "param", value = "代理参数", dataTypeClass = StreamProxyItem.class), 
 | 
    }) 
 | 
    @PostMapping(value = "/save") 
 | 
    @ResponseBody 
 | 
    public WVPResult save(@RequestBody StreamProxyItem param){ 
 | 
        logger.info("添加代理: " + JSONObject.toJSONString(param)); 
 | 
        if (StringUtils.isEmpty(param.getMediaServerId())) param.setMediaServerId("auto"); 
 | 
        if (StringUtils.isEmpty(param.getType())) param.setType("default"); 
 | 
        if (StringUtils.isEmpty(param.getGbId())) param.setGbId(null); 
 | 
        WVPResult<StreamInfo> result = streamProxyService.save(param); 
 | 
        return result; 
 | 
    } 
 | 
  
 | 
    @ApiOperation("获取ffmpeg.cmd模板") 
 | 
    @GetMapping(value = "/ffmpeg_cmd/list") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name = "mediaServerId", value = "流媒体ID", dataTypeClass = String.class), 
 | 
    }) 
 | 
    @ResponseBody 
 | 
    public WVPResult getFFmpegCMDs(@RequestParam String mediaServerId){ 
 | 
        logger.debug("获取节点[ {} ]ffmpeg.cmd模板", mediaServerId ); 
 | 
  
 | 
        MediaServerItem mediaServerItem = mediaServerService.getOne(mediaServerId); 
 | 
        JSONObject data = streamProxyService.getFFmpegCMDs(mediaServerItem); 
 | 
        WVPResult<JSONObject> result = new WVPResult<>(); 
 | 
        result.setCode(0); 
 | 
        result.setMsg("success"); 
 | 
        result.setData(data); 
 | 
        return result; 
 | 
    } 
 | 
  
 | 
    @ApiOperation("移除代理") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name = "app", value = "应用名", required = true, dataTypeClass = String.class), 
 | 
            @ApiImplicitParam(name = "stream", value = "流ID", required = true, dataTypeClass = String.class), 
 | 
    }) 
 | 
    @DeleteMapping(value = "/del") 
 | 
    @ResponseBody 
 | 
    public WVPResult del(@RequestParam String app, @RequestParam String stream){ 
 | 
        logger.info("移除代理: " + app + "/" + stream); 
 | 
        WVPResult<Object> result = new WVPResult<>(); 
 | 
        if (app == null || stream == null) { 
 | 
            result.setCode(400); 
 | 
            result.setMsg(app == null ?"app不能为null":"stream不能为null"); 
 | 
        }else { 
 | 
            streamProxyService.del(app, stream); 
 | 
            result.setCode(0); 
 | 
            result.setMsg("success"); 
 | 
        } 
 | 
        return result; 
 | 
    } 
 | 
  
 | 
    @ApiOperation("启用代理") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name = "app", value = "应用名", dataTypeClass = String.class), 
 | 
            @ApiImplicitParam(name = "stream", value = "流ID", dataTypeClass = String.class), 
 | 
    }) 
 | 
    @GetMapping(value = "/start") 
 | 
    @ResponseBody 
 | 
    public Object start(String app, String stream){ 
 | 
        logger.info("启用代理: " + app + "/" + stream); 
 | 
        boolean result = streamProxyService.start(app, stream); 
 | 
        return result?"success":"fail"; 
 | 
    } 
 | 
  
 | 
    @ApiOperation("停用代理") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name = "app", value = "应用名", dataTypeClass = String.class), 
 | 
            @ApiImplicitParam(name = "stream", value = "流ID", dataTypeClass = String.class), 
 | 
    }) 
 | 
    @GetMapping(value = "/stop") 
 | 
    @ResponseBody 
 | 
    public Object stop(String app, String stream){ 
 | 
        logger.info("停用代理: " + app + "/" + stream); 
 | 
        boolean result = streamProxyService.stop(app, stream); 
 | 
        return result?"success":"fail"; 
 | 
    } 
 | 
} 
 |