62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
package wal
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
"git.crumpington.com/public/jldb/lib/errs"
|
|
)
|
|
|
|
func (wal *WAL) Recv(conn net.Conn, timeout time.Duration) error {
|
|
defer conn.Close()
|
|
|
|
var (
|
|
rec Record
|
|
msgType = make([]byte, 1)
|
|
)
|
|
|
|
// Send sequence number.
|
|
seqNum := wal.Info().LastSeqNum + 1
|
|
conn.SetWriteDeadline(time.Now().Add(timeout))
|
|
if err := binary.Write(conn, binary.LittleEndian, seqNum); err != nil {
|
|
return errs.IO.WithErr(err)
|
|
}
|
|
conn.SetWriteDeadline(time.Time{})
|
|
|
|
for {
|
|
conn.SetReadDeadline(time.Now().Add(timeout))
|
|
|
|
if _, err := io.ReadFull(conn, msgType); err != nil {
|
|
return errs.IO.WithErr(err)
|
|
}
|
|
|
|
switch msgType[0] {
|
|
|
|
case msgTypeHeartbeat:
|
|
// Nothing to do.
|
|
|
|
case msgTypeError:
|
|
e := &errs.Error{}
|
|
if err := e.Read(conn); err != nil {
|
|
return err
|
|
}
|
|
|
|
return e
|
|
|
|
case msgTypeRec:
|
|
if err := rec.readFrom(conn); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, _, err := wal.appendRecord(rec); err != nil {
|
|
return err
|
|
}
|
|
|
|
default:
|
|
return errs.Unexpected.WithMsg("Unknown message type: %d", msgType[0])
|
|
}
|
|
}
|
|
}
|