apps/plugin-hdl
1
0
Fork 0
mirror of https://github.com/Monibuca/plugin-hdl.git synced 2025-10-05 08:47:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

对于远程连接返回非200时不再重连

Browse Source
This commit is contained in:
dexter
2022-12-31 22:01:59 +08:00
parent 7357cd8913
commit 1d6b18c48d
2 changed files with 36 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pull.go
View File

@@ -30,6 +30,9 @@ func (puller *HDLPuller) Connect() (err error) {
if strings.HasPrefix(puller.RemoteURL, "http") {
var res *http.Response
if res, err = http.Get(puller.RemoteURL); err == nil {
if res.StatusCode != http.StatusOK {
return io.EOF
}
puller.SetIO(res.Body)
}
} else {
@@ -38,14 +41,17 @@ func (puller *HDLPuller) Connect() (err error) {
puller.SetIO(res)
}
}
if err == nil {
head := puller.buf.SubBuf(0, len(codec.FLVHeader))
if _, err = io.ReadFull(puller, head); err == nil {
if head[0] != 'F' || head[1] != 'L' || head[2] != 'V' {
err = codec.ErrInvalidFLV
}
}
}
if err != nil {
HDLPlugin.Error("connect", zap.Error(err))
}
head := puller.buf.SubBuf(0, len(codec.FLVHeader))
_, err = io.ReadFull(puller, head)
if head[0] != 'F' || head[1] != 'L' || head[2] != 'V' {
err = codec.ErrInvalidFLV
}
return
}
@@ -71,6 +77,7 @@ func (puller *HDLPuller) Pull() (err error) {
return
}
puller.absTS = offsetTs + (timestamp - startTs)
// println(t, puller.absTS)
switch t {
case codec.FLV_TAG_TYPE_AUDIO:
puller.WriteAVCCAudio(puller.absTS, payload)