96 lines
1.8 KiB
Go
96 lines
1.8 KiB
Go
package rotate
|
|
|
|
import (
|
|
"RedisShake/internal/log"
|
|
"RedisShake/internal/utils"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
type AOFReader struct {
|
|
name string
|
|
dir string
|
|
file *os.File
|
|
offset int64
|
|
pos int64
|
|
filepath string
|
|
}
|
|
|
|
func NewAOFReader(name string, dir string, offset int64) *AOFReader {
|
|
r := new(AOFReader)
|
|
r.name = name
|
|
r.dir = dir
|
|
r.openFile(offset)
|
|
return r
|
|
}
|
|
|
|
func (r *AOFReader) openFile(offset int64) {
|
|
r.filepath = fmt.Sprintf("%s/%d.aof", r.dir, r.offset)
|
|
var err error
|
|
r.file, err = os.OpenFile(r.filepath, os.O_RDONLY, 0644)
|
|
if err != nil {
|
|
log.Panicf(err.Error())
|
|
}
|
|
r.offset = offset
|
|
r.pos = 0
|
|
log.Debugf("[%s] open file for read. filename=[%s]", r.name, r.filepath)
|
|
}
|
|
|
|
func (r *AOFReader) readNextFile(offset int64) bool {
|
|
filepath := fmt.Sprintf("%s/%d.aof", r.dir, r.offset)
|
|
if r.filepath == filepath {
|
|
return false
|
|
}
|
|
if !utils.IsExist(filepath) {
|
|
return false
|
|
}
|
|
r.Close()
|
|
err := os.Remove(r.filepath)
|
|
if err != nil {
|
|
log.Panicf("[%s] remove file failed. filename=[%s], err=[%v]", r.name, r.filepath, err)
|
|
return false
|
|
}
|
|
r.openFile(offset)
|
|
return true
|
|
}
|
|
|
|
func (r *AOFReader) Read(buf []byte) (n int, err error) {
|
|
n, err = r.file.Read(buf)
|
|
if err == io.EOF {
|
|
if !r.readNextFile(r.offset) {
|
|
return n, io.EOF
|
|
}
|
|
_, err = r.file.Seek(0, 1)
|
|
if err != nil {
|
|
log.Panicf(err.Error())
|
|
}
|
|
n, err = r.file.Read(buf)
|
|
if err != nil {
|
|
return n, err
|
|
}
|
|
}
|
|
if err != nil {
|
|
log.Panicf(err.Error())
|
|
}
|
|
r.offset += int64(n)
|
|
r.pos += int64(n)
|
|
return n, nil
|
|
}
|
|
|
|
func (r *AOFReader) Offset() int64 {
|
|
return r.offset
|
|
}
|
|
|
|
func (r *AOFReader) Close() {
|
|
if r.file == nil {
|
|
return
|
|
}
|
|
err := r.file.Close()
|
|
if err != nil {
|
|
log.Panicf(err.Error())
|
|
}
|
|
r.file = nil
|
|
log.Debugf("[%s] close file. filename=[%s]", r.name, r.filepath)
|
|
}
|