Vendor main dependencies.
This commit is contained in:
parent
49a09ab7dd
commit
dd5e3fba01
2738 changed files with 1045689 additions and 0 deletions
78
vendor/github.com/donovanhide/eventsource/decoder.go
generated
vendored
Normal file
78
vendor/github.com/donovanhide/eventsource/decoder.go
generated
vendored
Normal file
|
@ -0,0 +1,78 @@
|
|||
package eventsource
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type publication struct {
|
||||
id, event, data string
|
||||
retry int64
|
||||
}
|
||||
|
||||
func (s *publication) Id() string { return s.id }
|
||||
func (s *publication) Event() string { return s.event }
|
||||
func (s *publication) Data() string { return s.data }
|
||||
func (s *publication) Retry() int64 { return s.retry }
|
||||
|
||||
// A Decoder is capable of reading Events from a stream.
|
||||
type Decoder struct {
|
||||
*bufio.Reader
|
||||
}
|
||||
|
||||
// NewDecoder returns a new Decoder instance that reads events
|
||||
// with the given io.Reader.
|
||||
func NewDecoder(r io.Reader) *Decoder {
|
||||
dec := &Decoder{bufio.NewReader(newNormaliser(r))}
|
||||
return dec
|
||||
}
|
||||
|
||||
// Decode reads the next Event from a stream (and will block until one
|
||||
// comes in).
|
||||
// Graceful disconnects (between events) are indicated by an io.EOF error.
|
||||
// Any error occuring mid-event is considered non-graceful and will
|
||||
// show up as some other error (most likely io.ErrUnexpectedEOF).
|
||||
func (dec *Decoder) Decode() (Event, error) {
|
||||
|
||||
// peek ahead before we start a new event so we can return EOFs
|
||||
_, err := dec.Peek(1)
|
||||
if err == io.ErrUnexpectedEOF {
|
||||
err = io.EOF
|
||||
}
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
pub := new(publication)
|
||||
for {
|
||||
line, err := dec.ReadString('\n')
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if line == "\n" {
|
||||
break
|
||||
}
|
||||
line = strings.TrimSuffix(line, "\n")
|
||||
if strings.HasPrefix(line, ":") {
|
||||
continue
|
||||
}
|
||||
sections := strings.SplitN(line, ":", 2)
|
||||
field, value := sections[0], ""
|
||||
if len(sections) == 2 {
|
||||
value = strings.TrimPrefix(sections[1], " ")
|
||||
}
|
||||
switch field {
|
||||
case "event":
|
||||
pub.event = value
|
||||
case "data":
|
||||
pub.data += value + "\n"
|
||||
case "id":
|
||||
pub.id = value
|
||||
case "retry":
|
||||
pub.retry, _ = strconv.ParseInt(value, 10, 64)
|
||||
}
|
||||
}
|
||||
pub.data = strings.TrimSuffix(pub.data, "\n")
|
||||
return pub, nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue