mirror of https://github.com/certusone/dc4bc.git
little ref
This commit is contained in:
parent
1a7bc55453
commit
cc688c48d3
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue