78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
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
|
|
}
|