dont read forever
This commit is contained in:
@@ -35,12 +35,15 @@ var ErrUnknownType = errors.New("unknown type")
|
||||
|
||||
func (s *StateStorage) Load(r io.Reader, onMessage func(msg proto.Message)) error {
|
||||
var err error
|
||||
var evt *StorageEvent
|
||||
for err == nil {
|
||||
evt, err := s.Read(r)
|
||||
evt, err = s.Read(r)
|
||||
if err == nil {
|
||||
onMessage(evt.Mutation)
|
||||
}
|
||||
|
||||
}
|
||||
if err == io.EOF {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user