more data
This commit is contained in:
38
message.go
38
message.go
@@ -42,42 +42,6 @@ func GetData(fn func(w io.Writer) error) ([]byte, error) {
|
||||
return b, nil
|
||||
}
|
||||
|
||||
// func (w *MessageWriter) WriteUint64(value uint64) (int, error) {
|
||||
// bytes := make([]byte, 8)
|
||||
// binary.LittleEndian.PutUint64(bytes, value)
|
||||
// return w.Write(bytes)
|
||||
// }
|
||||
|
||||
// func (w *MessageWriter) WriteInt64(value int64) (int, error) {
|
||||
// return w.WriteUint64(uint64(value))
|
||||
// }
|
||||
|
||||
// func (w *MessageWriter) WriteMessage(m *Message) (int, error) {
|
||||
// var i, l int
|
||||
// var err error
|
||||
// i, err = w.WriteUint64(m.Type)
|
||||
// l += i
|
||||
// i, err = w.WriteInt64(*m.TimeStamp)
|
||||
// l += i
|
||||
// var messageBytes []byte
|
||||
// var err error
|
||||
// if m.Type == AddRequestType {
|
||||
// messageBytes, err = proto.Marshal(m.Content.(*messages.AddRequest))
|
||||
// } else if m.Type == AddItemType {
|
||||
// messageBytes, err = proto.Marshal(m.Content.(*messages.AddItem))
|
||||
// } else {
|
||||
// return fmt.Errorf("unknown message type")
|
||||
// }
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
// if err := w.WriteUint64(uint64(len(messageBytes))); err != nil {
|
||||
// return err
|
||||
// }
|
||||
// _, err = w.Write(messageBytes)
|
||||
// return err
|
||||
// }
|
||||
|
||||
func (m Message) Write(w io.Writer) error {
|
||||
data, err := GetData(func(wr io.Writer) error {
|
||||
if m.Type == AddRequestType {
|
||||
@@ -140,6 +104,8 @@ func MessageFromReader(reader io.Reader, m *Message) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
m.Type = header.Type
|
||||
m.TimeStamp = &header.TimeStamp
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user