|
|
|
package reader
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/alibaba/RedisShake/internal/entry"
|
|
|
|
"github.com/alibaba/RedisShake/internal/log"
|
|
|
|
"github.com/alibaba/RedisShake/internal/rdb"
|
|
|
|
"github.com/alibaba/RedisShake/internal/statistics"
|
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
)
|
|
|
|
|
|
|
|
type rdbReader struct {
|
|
|
|
path string
|
|
|
|
ch chan *entry.Entry
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewRDBReader(path string) Reader {
|
|
|
|
log.Infof("NewRDBReader: path=[%s]", path)
|
|
|
|
absolutePath, err := filepath.Abs(path)
|
|
|
|
if err != nil {
|
|
|
|
log.Panicf("NewRDBReader: filepath.Abs error: %s", err.Error())
|
|
|
|
}
|
|
|
|
log.Infof("NewRDBReader: absolute path=[%s]", absolutePath)
|
|
|
|
r := new(rdbReader)
|
|
|
|
r.path = absolutePath
|
|
|
|
return r
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rdbReader) StartRead() chan *entry.Entry {
|
|
|
|
r.ch = make(chan *entry.Entry, 1024)
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
// start parse rdb
|
|
|
|
log.Infof("start send RDB. path=[%s]", r.path)
|
|
|
|
fi, err := os.Stat(r.path)
|
|
|
|
if err != nil {
|
|
|
|
log.Panicf("NewRDBReader: os.Stat error: %s", err.Error())
|
|
|
|
}
|
|
|
|
statistics.Metrics.RdbFileSize = uint64(fi.Size())
|
|
|
|
statistics.Metrics.RdbReceivedSize = uint64(fi.Size())
|
|
|
|
rdbLoader := rdb.NewLoader(r.path, r.ch)
|
|
|
|
_ = rdbLoader.ParseRDB()
|
|
|
|
log.Infof("send RDB finished. path=[%s]", r.path)
|
|
|
|
close(r.ch)
|
|
|
|
}()
|
|
|
|
|
|
|
|
return r.ch
|
|
|
|
}
|