james/yarr
4
0
Fork 0
mirror of https://github.com/nkanaev/yarr.git synced 2025-05-24 00:33:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix feed sniff reader

Browse Source
This commit is contained in:
Nazar Kanaev 2021-03-24 12:18:46 +00:00
parent a2bfd1682b
commit e0e6166cdf
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/parser/feed.go
View File

@ -44,8 +44,15 @@ func sniff(lookup string) (string, processor) {
func Parse(r io.Reader) (*Feed, error) { func Parse(r io.Reader) (*Feed, error) {
lookup := make([]byte, 1024) lookup := make([]byte, 1024)
if _, err := r.Read(lookup); err != nil { n, err := io.ReadFull(r, lookup)
return nil, fmt.Errorf("Failed to read input: %s", err) switch {
case err != nil:
return nil, err
case err == io.ErrUnexpectedEOF:
lookup = lookup[:n]
r = bytes.NewReader(lookup)
default:
r = io.MultiReader(bytes.NewReader(lookup), r)
} }
_, callback := sniff(string(lookup)) _, callback := sniff(string(lookup))
@ -53,7 +60,7 @@ func Parse(r io.Reader) (*Feed, error) {
return nil, UnknownFormat return nil, UnknownFormat
} }
feed, err := callback(io.MultiReader(bytes.NewReader(lookup), r)) feed, err := callback(r)
if feed != nil { if feed != nil {
feed.cleanup() feed.cleanup()
} }