diff --git a/pkg/actor/state.go b/pkg/actor/state.go index 6303b30..be7c4ff 100644 --- a/pkg/actor/state.go +++ b/pkg/actor/state.go @@ -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 }