9e319ed27c
* feat(backend): use channel of changed sessions instead of sync.Map * feat(backend): avoid memory alloc for message body in message iterator * feat(backend): removed unnecessary locks in file syncer * feat(backend): sync.Map with prev updates * feat(backend): improved write algorith (added bufio.Writer) * feat(backend): session writer refactoring * feat(backend): removed unnecessary type definition * feat(backend): added write retrier to avoid data losing * feat(backend): refactoring * feat(backend): added session file implementation
57 lines
946 B
Go
57 lines
946 B
Go
package sessionwriter
|
|
|
|
import (
|
|
"bufio"
|
|
"os"
|
|
)
|
|
|
|
type File struct {
|
|
file *os.File
|
|
buffer *bufio.Writer
|
|
updated bool
|
|
}
|
|
|
|
func NewFile(path string, bufSize int) (*File, error) {
|
|
file, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &File{
|
|
file: file,
|
|
buffer: bufio.NewWriterSize(file, bufSize),
|
|
updated: false,
|
|
}, nil
|
|
}
|
|
|
|
func (f *File) Write(data []byte) error {
|
|
leftToWrite := len(data)
|
|
for leftToWrite > 0 {
|
|
writtenDown, err := f.buffer.Write(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
leftToWrite -= writtenDown
|
|
}
|
|
f.updated = true
|
|
return nil
|
|
}
|
|
|
|
func (f *File) Sync() error {
|
|
if !f.updated {
|
|
return nil
|
|
}
|
|
if err := f.buffer.Flush(); err != nil {
|
|
return err
|
|
}
|
|
if err := f.file.Sync(); err != nil {
|
|
return err
|
|
}
|
|
f.updated = false
|
|
return nil
|
|
}
|
|
|
|
func (f *File) Close() error {
|
|
_ = f.buffer.Flush()
|
|
_ = f.file.Sync()
|
|
return f.file.Close()
|
|
}
|