迪奥布兰度正在挑战fgo 小说_god eater resurrection

// 监听aof通道并写入文件
func (handler *Handler) handleAof()  { 

// 序列化执行
handler.currentDB = 0
// 遍历通道中的命令
for p := range handler.aofChan { 

// 阻止其它协程暂停AOF
handler.pausingAof.RLock()
// 如果管道中的命令不属于当前选择的数据库，还要加上选择数据库的命令
if p.dbIndex != handler.currentDB { 

data := RESP.MakeMultiBulkReply(Utils.ToCmdLine("SELECT", strconv.Itoa(p.dbIndex))).ToBytes()
// 将选择数据库的指令写入aof文件
_, err := handler.aofFile.Write(data)
if err != nil { 

log.Error(err)
continue
}
handler.currentDB = p.dbIndex
}
// 将命令转化为字节数组
data := RESP.MakeMultiBulkReply(p.cmdLine).ToBytes()
_, err := handler.aofFile.Write(data)
if err != nil { 

log.Error(err)
}
handler.pausingAof.RUnlock()
}
// 将msg发送到主协程
handler.aofFinished <- struct{ 
}{ 
}
}

// LoadAof 将文件中的数据加载到内存
func (handler *Handler) LoadAof(maxBytes int) { 

// 删除aofChan以防再次写入
aofChan := handler.aofChan
handler.aofChan = nil
defer func(aofChan chan *payload) { 

handler.aofChan = aofChan
}(aofChan)
// 打开aof文件
file, err := os.Open(handler.aofFilename)
if err != nil { 

// 如果是路径有错，就直接结束
if _, ok := err.(*os.PathError); ok { 

return
}
log.Error(err)
return
}
defer file.Close()
// 开始读文件
var reader io.Reader
if maxBytes > 0 { 

reader = io.LimitReader(file, int64(maxBytes))
} else { 

reader = file
}
// 从 reader 读取数据并通过通道发送payload
ch := RESP.ParseStream(reader)
// 创建一个伪客户端
fakeConn := &connection.FakeConn{ 
}
for p := range ch { 

if p.Err != nil { 

// 文件读取完毕就退出
if p.Err == io.EOF { 

break
}
log.Error("parse error: " + p.Err.Error())
continue
}
if p.Data == nil { 

log.Error("empty payload")
continue
}
// 这里的p.Data必须是一个字符串list
r, ok := p.Data.(*RESP.MultiBulkReply)
if !ok { 

log.Error("require multi bulk reply")
continue
}
// 使用伪客户端发送命令
ret := handler.db.Exec(fakeConn, r.Args)
// 检查格式是否正确
if ret.ToBytes()[0] == '-' { 

log.Error("exec err", err)
}
}
}

// StartRewrite 为重写做准备
// 返回一个RewriteCtx， 包含了一个临时文件指针、文件大小以及当前数据库的编号
func (handler *Handler) StartRewrite() (*RewriteCtx, error) { 

// 暂停aof写入，数据会在aofChan中暂时堆积
handler.pausingAof.Lock()
defer handler.pausingAof.Unlock()
// 调用 fsync 将缓冲区中的数据落盘，防止 aof 文件不完整造成错误
err := handler.aofFile.Sync()
if err != nil { 

log.Error("fsync failed")
return nil, err
}
// 获得当前 aof 文件大小，用于判断哪些数据是 aof 重写过程中产生的
// handleAof 会保证每次写入完整的一条指令
fileInfo, _ := os.Stat(handler.aofFilename)
filesize := fileInfo.Size()
// 创建临时文件
// 系统会自动将*号替换成随机的字符
file, err := ioutil.TempFile("", "*.aof")
if err != nil { 

log.Error("tmp file create failed")
return nil, err
}
return &RewriteCtx{ 

tmpFile:  file,
fileSize: filesize,
dbIdx:    handler.currentDB,   // 重写开始时 aof 文件选中的数据库
}, nil
}

// DoRewrite 实际上重写的是aof文件
// 外部调用请使用 Rewrite 函数
func (handler *Handler) DoRewrite(ctx *RewriteCtx) error { 

tmpFile := ctx.tmpFile
// 将重写开始前的数据加载到内存
tmpAof := handler.newRewriteHandler()
tmpAof.LoadAof(int(ctx.fileSize))
// 将内存中的数据写入临时文件
for i := 0; i < Properties.Databases; i++ { 

// 选择数据库
data := RESP.MakeMultiBulkReply(Utils.ToCmdLine("SELECT", strconv.Itoa(i))).ToBytes()
// 写入文件
_, err := tmpFile.Write(data)
if err != nil { 

return err
}
// dump db
tmpAof.db.ForEach(i, func(key string, entity *database.DataEntity, expiration *time.Time) bool { 

// 序列化为命令
cmd := EntityToCmd(key, entity)
if cmd != nil { 

// 将命令写入到临时文件中
_, _ = tmpFile.Write(cmd.ToBytes())
}
// 如果命令已经过期，那就为命令添加上过期的标签并设置过期时间
if expiration != nil { 

cmd := MakeExpireCmd(key, *expiration)
if cmd != nil { 

_, _ = tmpFile.Write(cmd.ToBytes())
}
}
return true
})
}
return nil
}

// FinishRewrite 进行重写的善后操作
func (handler *Handler) FinishRewrite(ctx *RewriteCtx) { 

// 同样暂停 handleAof 的写入
handler.pausingAof.Lock()
defer handler.pausingAof.Unlock()
// 打开线上 aof 文件并 seek 到重写开始的位置
tmpFile := ctx.tmpFile
src, err := os.Open(handler.aofFilename)
if err != nil { 

log.Error("open aofFilename failed:" + err.Error())
return
}
defer func() { 

_ = src.Close()
}()
_, err = src.Seek(ctx.fileSize, 0)
if err != nil { 

log.Error("seek failed: " + err.Error())
return
}
// 写入一条 Select 命令，使 tmpAof 选中重写开始时刻线上 aof 文件选中的数据库
data := RESP.MakeMultiBulkReply(Utils.ToCmdLine("SELECT", strconv.Itoa(ctx.dbIdx))).ToBytes()
_, err = tmpFile.Write(data)
if err != nil { 

log.Error("tmp file rewrite failed: " + err.Error())
return
}
// 对齐数据库后就可以把重写过程中产生的数据复制到 tmpAof 文件了
_, err = io.Copy(tmpFile, src)
if err != nil { 

log.Error("copy aof filed failed: " + err.Error())
return
}
// 使用 mv 命令用 tmpAof 代替线上 aof 文件
_ = handler.aofFile.Close()
_ = os.Rename(tmpFile.Name(), handler.aofFilename)
// 重新打开线上 aof
aofFile, err := os.OpenFile(handler.aofFilename, os.O_APPEND|os.O_CREATE|os.O_RDWR, 0600)
if err != nil { 

panic(err)
}
handler.aofFile = aofFile
// reset selected db 重新写入一次 select 指令保证 aof 中的数据库与 handler.currentDB 一致
data = RESP.MakeMultiBulkReply(Utils.ToCmdLine("SELECT", strconv.Itoa(handler.currentDB))).ToBytes()
_, err = handler.aofFile.Write(data)
if err != nil { 

panic(err)
}
}

// EntityToCmd 将数据实体序列化为 redis 命令
// DataEntity 存储绑定到一个键的数据，包括字符串、列表、哈希、集合等
func EntityToCmd(key string, entity *database.DataEntity) *RESP.MultiBulkReply { 

if entity == nil { 

return nil
}
var cmd *RESP.MultiBulkReply
// 用类型断言来判断键的类型，然后序列化为对应的命令并返回
switch val := entity.Data.(type) { 

case []byte:
cmd = stringToCmd(key, val)
case *list.LinkedList:
cmd = listToCmd(key, val)
case *set.Set:
cmd = setToCmd(key, val)
case dict.Dict:
cmd = hashToCmd(key, val)
case *sortedset.SortedSet:
cmd = zSetToCmd(key, val)
}
return cmd
}

// AddAof 通过通道向 aof goroutine 发送命令
func (handler *Handler) AddAof(dbIndex int, cmdLine CmdLine)  { 

// 如果持久化选项开启 且 aof通道被初始化
if Properties.AppendOnly == true && handler.aofChan != nil { 

handler.aofChan <- &payload{ 

cmdLine: cmdLine,
dbIndex: dbIndex,
}
}
}