mirror of https://github.com/EasyDarwin/EasyDarwin
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
128 lines
3.6 KiB
128 lines
3.6 KiB
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)) |
|
}
|
|
|