little ref

This commit is contained in:
programmer10110 2020-07-29 12:26:25 +03:00
parent 1a7bc55453
commit cc688c48d3
1 changed files with 7 additions and 3 deletions

View File

@ -16,7 +16,11 @@ type FileStorage struct {
reader *bufio.Reader reader *bufio.Reader
} }
func InitFileStorage(filename string) (*FileStorage, error) { const (
EOL = '\n'
)
func InitFileStorage(filename string) (Storage, error) {
var ( var (
fs FileStorage fs FileStorage
err error err error
@ -36,7 +40,7 @@ func (fs *FileStorage) Post(m Message) error {
if err != nil { if err != nil {
return err return err
} }
data = append(data, '\n') data = append(data, EOL)
_, err = fs.file.Write(data) _, err = fs.file.Write(data)
return err return err
} }
@ -55,7 +59,7 @@ func (fs *FileStorage) GetMessages(offset int) ([]Message, error) {
return nil, err return nil, err
} }
for { for {
row, err = fs.reader.ReadBytes('\n') row, err = fs.reader.ReadBytes(EOL)
if err != nil { if err != nil {
break break
} }