|
|
package routers
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"log"
|
|
|
"net/http"
|
|
|
"strings"
|
|
|
"time"
|
|
|
|
|
|
"github.com/EasyDarwin/EasyDarwin/models"
|
|
|
"github.com/penggy/EasyGoLib/db"
|
|
|
|
|
|
"github.com/EasyDarwin/EasyDarwin/rtsp"
|
|
|
"github.com/gin-gonic/gin"
|
|
|
)
|
|
|
|
|
|
/**
|
|
|
* @apiDefine stream 流管理
|
|
|
*/
|
|
|
|
|
|
/**
|
|
|
* @api {get} /api/v1/stream/start 启动拉转推
|
|
|
* @apiGroup stream
|
|
|
* @apiName StreamStart
|
|
|
* @apiParam {String} url RTSP源地址
|
|
|
* @apiParam {String} [customPath] 转推时的推送PATH
|
|
|
* @apiParam {String=TCP,UDP} [transType=TCP] 拉流传输模式
|
|
|
* @apiParam {Number} [idleTimeout] 拉流时的超时时间
|
|
|
* @apiParam {Number} [heartbeatInterval] 拉流时的心跳间隔，毫秒为单位。如果心跳间隔不为0，那拉流时会向源地址以该间隔发送OPTION请求用来心跳保活
|
|
|
* @apiSuccess (200) {String} ID 拉流的ID。后续可以通过该ID来停止拉流
|
|
|
*/
|
|
|
func (h *APIHandler) StreamStart(c *gin.Context) {
|
|
|
type Form struct {
|
|
|
URL string `form:"url" binding:"required"`
|
|
|
CustomPath string `form:"customPath"`
|
|
|
TransType string `form:"transType"`
|
|
|
IdleTimeout int `form:"idleTimeout"`
|
|
|
HeartbeatInterval int `form:"heartbeatInterval"`
|
|
|
}
|
|
|
var form Form
|
|
|
err := c.Bind(&form)
|
|
|
if err != nil {
|
|
|
log.Printf("Pull to push err:%v", err)
|
|
|
return
|
|
|
}
|
|
|
agent := fmt.Sprintf("EasyDarwinGo/%s", BuildVersion)
|
|
|
if BuildDateTime != "" {
|
|
|
agent = fmt.Sprintf("%s(%s)", agent, BuildDateTime)
|
|
|
}
|
|
|
client, err := rtsp.NewRTSPClient(rtsp.GetServer(), form.URL, int64(form.HeartbeatInterval)*1000, agent)
|
|
|
if err != nil {
|
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, err.Error())
|
|
|
return
|
|
|
}
|
|
|
if form.CustomPath != "" && !strings.HasPrefix(form.CustomPath, "/") {
|
|
|
form.CustomPath = "/" + form.CustomPath
|
|
|
}
|
|
|
client.CustomPath = form.CustomPath
|
|
|
switch strings.ToLower(form.TransType) {
|
|
|
case "udp":
|
|
|
client.TransType = rtsp.TRANS_TYPE_UDP
|
|
|
case "tcp":
|
|
|
fallthrough
|
|
|
default:
|
|
|
client.TransType = rtsp.TRANS_TYPE_TCP
|
|
|
}
|
|
|
|
|
|
pusher := rtsp.NewClientPusher(client)
|
|
|
if rtsp.GetServer().GetPusher(pusher.Path()) != nil {
|
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, fmt.Sprintf("Path %s already exists", client.Path))
|
|
|
return
|
|
|
}
|
|
|
err = client.Start(time.Duration(form.IdleTimeout) * time.Second)
|
|
|
if err != nil {
|
|
|
log.Printf("Pull stream err :%v", err)
|
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, fmt.Sprintf("Pull stream err: %v", err))
|
|
|
return
|
|
|
}
|
|
|
log.Printf("Pull to push %v success ", form)
|
|
|
rtsp.GetServer().AddPusher(pusher, false)
|
|
|
// save to db.
|
|
|
var stream = models.Stream{
|
|
|
URL: form.URL,
|
|
|
CustomPath: form.CustomPath,
|
|
|
IdleTimeout: form.IdleTimeout,
|
|
|
HeartbeatInterval: form.HeartbeatInterval,
|
|
|
}
|
|
|
if db.SQLite.Where(&models.Stream{URL: form.URL}).First(&models.Stream{}).RecordNotFound() {
|
|
|
db.SQLite.Create(&stream)
|
|
|
} else {
|
|
|
db.SQLite.Save(&stream)
|
|
|
}
|
|
|
c.IndentedJSON(200, pusher.ID())
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
* @api {get} /api/v1/stream/stop 停止推流
|
|
|
* @apiGroup stream
|
|
|
* @apiName StreamStop
|
|
|
* @apiParam {String} id 拉流的ID
|
|
|
* @apiUse simpleSuccess
|
|
|
*/
|
|
|
func (h *APIHandler) StreamStop(c *gin.Context) {
|
|
|
type Form struct {
|
|
|
ID string `form:"id" binding:"required"`
|
|
|
}
|
|
|
var form Form
|
|
|
err := c.Bind(&form)
|
|
|
if err != nil {
|
|
|
log.Printf("stop pull to push err:%v", err)
|
|
|
return
|
|
|
}
|
|
|
pushers := rtsp.GetServer().GetPushers()
|
|
|
for _, v := range pushers {
|
|
|
if v.ID() == form.ID {
|
|
|
v.Stop()
|
|
|
c.IndentedJSON(200, "OK")
|
|
|
log.Printf("Stop %v success ", v)
|
|
|
if v.RTSPClient != nil {
|
|
|
var stream models.Stream
|
|
|
stream.URL = v.RTSPClient.URL
|
|
|
db.SQLite.Delete(stream)
|
|
|
}
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
c.AbortWithStatusJSON(http.StatusBadRequest, fmt.Sprintf("Pusher[%s] not found", form.ID))
|
|
|
}
|
