mirror of https://github.com/EasyDarwin/EasyDarwin
129 lines
3.6 KiB
Go
129 lines
3.6 KiB
Go
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)
|
||
// 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))
|
||
}
|