cubefs/remotecache/flashnode/flashnode_op.go
clinx d35a07e841 fix(flash): update read metric
with: #1000451308

Signed-off-by: clinx <chenlin1@oppo.com>
(cherry picked from commit 1581da2e1c)
2025-12-24 17:01:31 +08:00

1346 lines
41 KiB
Go

// Copyright 2023 The CubeFS Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
// implied. See the License for the specific language governing
// permissions and limitations under the License.
package flashnode
import (
"bytes"
"context"
"encoding/json"
"fmt"
"hash/crc32"
"io"
"net"
"strings"
"sync"
"sync/atomic"
"syscall"
"time"
"github.com/cubefs/cubefs/proto"
"github.com/cubefs/cubefs/remotecache/flashnode/cachengine"
"github.com/cubefs/cubefs/sdk/data/stream"
"github.com/cubefs/cubefs/sdk/meta"
"github.com/cubefs/cubefs/util"
"github.com/cubefs/cubefs/util/bytespool"
"github.com/cubefs/cubefs/util/errors"
"github.com/cubefs/cubefs/util/exporter"
"github.com/cubefs/cubefs/util/log"
"github.com/cubefs/cubefs/util/stat"
"github.com/google/uuid"
)
func (f *FlashNode) preHandle(conn net.Conn, p *proto.Packet) error {
if (p.Opcode == proto.OpFlashNodeCacheRead || p.Opcode == proto.OpFlashNodeCachePrepare || p.Opcode == 0xDC || p.Opcode == 0xDB) && !f.readLimiter.Allow() {
metric := exporter.NewTPCnt("NodeReqLimit")
metric.Set(nil)
err := errors.NewErrorf("%s", "remotecache read request was been limited")
log.LogWarnf("action[preHandle] %s, remote address:%s", err.Error(), conn.RemoteAddr())
return err
}
return nil
}
func (f *FlashNode) handlePacket(conn net.Conn, p *proto.Packet) (err error) {
switch p.Opcode {
case proto.OpFlashSDKHeartbeat:
err = f.opClientHeartbeat(conn, p)
case proto.OpFlashNodeHeartbeat:
err = f.opFlashNodeHeartbeat(conn, p)
case proto.OpFlashNodeCachePrepare:
err = f.opCachePrepare(conn, p)
case proto.OpFlashNodeCacheRead:
err = f.opCacheRead(conn, p)
case proto.OpFlashNodeCachePutBlock:
err = f.opCachePutBlock(conn, p)
case proto.OpFlashNodeCacheDelete:
err = f.opCacheDelete(conn, p)
case proto.OpFlashNodeCacheReadObject:
err = f.opCacheObjectGet(conn, p)
case proto.OpFlashNodeBatchReadObject:
err = f.opCacheBatchObjectGet(conn, p)
case proto.OpFlashNodeSetReadIOLimits:
err = f.opSetReadIOLimits(conn, p)
case proto.OpFlashNodeSetWriteIOLimits:
err = f.opSetWriteIOLimits(conn, p)
case proto.OpFlashNodeScan:
err = f.opFlashNodeScan(conn, p)
case proto.OpFlashNodeTaskCommand:
err = f.opFlashNodeTaskCommand(conn, p)
case proto.OpApplyWarmupMetaToken:
err = f.opApplyWarmupMetaToken(conn, p)
default:
// compatibility for historical opcodes
switch p.Opcode {
case 0xDB: // legacy OpFlashNodeCachePrepare
err = f.opCachePrepare(conn, p)
case 0xDC: // legacy OpFlashNodeCacheRead
err = f.opCacheRead(conn, p)
case 0xDA: // legacy OpFlashNodeHeartbeat
err = f.opFlashNodeHeartbeat(conn, p)
default:
err = fmt.Errorf(proto.ErrorUnknownOpcodeTpl, p.Opcode)
}
}
return
}
func (f *FlashNode) SetTimeout(handleReadTimeout int, readDataNodeTimeout int) {
if f.handleReadTimeout != handleReadTimeout && handleReadTimeout > 0 {
log.LogInfof("FlashNode set handleReadTimeout from %d(ms) to %d(ms)", f.handleReadTimeout, handleReadTimeout)
f.handleReadTimeout = handleReadTimeout
}
f.cacheEngine.SetReadDataNodeTimeout(readDataNodeTimeout)
}
func (f *FlashNode) opClientHeartbeat(conn net.Conn, p *proto.Packet) (err error) {
p.PacketOkReply()
if f.enableWarmUpPaths {
var warmUpPaths []*proto.WarmUpPathInfo
f.warmUpPaths.Range(func(key, value interface{}) bool {
info := value.(*proto.WarmUpPathInfo)
if time.Now().UnixNano() > info.Expiration {
f.warmUpPaths.Delete(key)
} else {
warmUpPaths = append(warmUpPaths, info)
}
return true
})
if len(warmUpPaths) != 0 {
p.Data, err = proto.MarshalBinaryWPSlice(warmUpPaths)
if err != nil {
log.LogWarnf("opClientHeartbeat MarshalBinaryWPSlice: %s", err.Error())
} else {
p.Size = uint32(len(p.Data))
}
}
}
if err = p.WriteToConn(conn); err != nil {
log.LogErrorf("opClientHeartbeat response: %s", err.Error())
}
if log.EnableInfo() {
log.LogInfof("opClientHeartbeat remoteaddr(%v)", conn.RemoteAddr())
}
return
}
func (f *FlashNode) opFlashNodeHeartbeat(conn net.Conn, p *proto.Packet) (err error) {
data := p.Data
go responseAckOKToMaster(conn, p)
req := &proto.HeartBeatRequest{}
resp := &proto.FlashNodeHeartbeatResponse{}
adminTask := &proto.AdminTask{
Request: req,
}
decode := json.NewDecoder(bytes.NewBuffer(data))
decode.UseNumber()
if err = decode.Decode(adminTask); err == nil {
f.SetTimeout(req.FlashNodeHandleReadTimeout, req.FlashNodeReadDataNodeTimeout)
if req.FlashHotKeyMissCount != 0 {
f.hotKeyMissCount = int32(req.FlashHotKeyMissCount)
}
if req.FlashKeyFlowLimit != f.keyLimiterFlow {
f.keyLimiterFlow = req.FlashKeyFlowLimit
f.cacheEngine.SetKeyLimiterFlow(req.FlashKeyFlowLimit)
}
if !f.localChangeWriteFlow && req.FlashWriteFlowLimit != int64(f.diskWriteFlow) {
f.diskWriteFlow = int(req.FlashWriteFlowLimit)
f.limitWrite.ResetFlow(f.diskWriteFlow)
}
if !f.localChangeReadFlow && req.FlashReadFlowLimit != int64(f.diskReadFlow) {
f.diskReadFlow = int(req.FlashReadFlowLimit)
f.limitRead.ResetFlow(f.diskReadFlow)
}
} else {
log.LogWarnf("decode HeartBeatRequest error: %s", err.Error())
resp.Status = proto.TaskFailed
resp.Result = fmt.Sprintf("remotecache(%v) heartbeat decode err(%v)", f.localAddr, err.Error())
goto end
}
resp.Stat = make([]*proto.FlashNodeDiskCacheStat, 0)
for _, cacheStat := range f.cacheEngine.GetHeartBeatCacheStat() {
cacheStat := &proto.FlashNodeDiskCacheStat{
DataPath: cacheStat.DataPath,
Medium: cacheStat.Medium,
Total: cacheStat.Total,
MaxAlloc: cacheStat.MaxAlloc,
HasAlloc: cacheStat.HasAlloc,
FreeSpace: cacheStat.MaxAlloc - cacheStat.HasAlloc,
HitRate: cacheStat.HitRate,
Evicts: cacheStat.Evicts,
ReadRps: f.readRps,
KeyNum: cacheStat.Num,
Status: cacheStat.Status,
}
resp.Stat = append(resp.Stat, cacheStat)
}
resp.LimiterStatus = &proto.FlashNodeLimiterStatusInfo{
WriteStatus: proto.FlashNodeLimiterStatus{Status: f.limitWrite.Status(true), DiskNum: len(f.disks), ReadTimeout: f.handleReadTimeout},
ReadStatus: proto.FlashNodeLimiterStatus{Status: f.limitRead.Status(true), DiskNum: len(f.disks), ReadTimeout: f.handleReadTimeout},
}
resp.FlashNodeTaskCountLimit = f.taskCountLimit
resp.ManualScanningTasks = make(map[string]*proto.FlashNodeManualTaskResponse)
f.manualScanners.Range(func(_, mScanner interface{}) bool {
scanner := mScanner.(*ManualScanner)
result := scanner.copyResponse()
resp.ManualScanningTasks[scanner.ID] = result
return true
})
resp.Status = proto.TaskSucceeds
end:
adminTask.Response = resp
f.respondToMaster(adminTask)
if log.EnableInfo() {
log.LogInfof("[opMasterHeartbeat] master:%s handleReadTimeout %v(ms) readDataNodeTimeout %v(ms) hotkeymisscount %v",
conn.RemoteAddr().String(), req.FlashNodeHandleReadTimeout, req.FlashNodeReadDataNodeTimeout, req.FlashHotKeyMissCount)
}
return
}
func (f *FlashNode) opCacheRead(conn net.Conn, p *proto.Packet) (err error) {
var volume string
bgTime := stat.BeginStat()
defer func() {
stat.EndStat("FlashNode:opCacheRead", err, bgTime, 1)
}()
defer func() {
if err != nil {
if !proto.IsFlashNodeLimitError(err) {
log.LogWarnf("action[opCacheRead] volume:[%s], logMsg:%s", volume,
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
}
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
if e := p.WriteToConn(conn); e != nil {
log.LogErrorf("action[opCacheRead] write to conn %v", e)
}
}
}()
ctx, ctxCancel := context.WithTimeout(context.Background(), time.Duration(f.handleReadTimeout)*time.Millisecond)
defer ctxCancel()
req := new(proto.CacheReadRequest)
if err = p.UnmarshalDataPb(req); err != nil {
return
}
if req.CacheRequest == nil {
err = proto.ErrorNoCacheReadRequest
return
}
volume = req.CacheRequest.Volume
cr := req.CacheRequest
f.updateSlotStat(cr.Slot)
block, err := f.cacheEngine.GetCacheBlockForRead(volume, cr.Inode, cr.FixedFileOffset, cr.Version, req.Size_)
if err != nil {
hitRateMap := f.cacheEngine.GetHitRate()
for dataPath, hitRate := range hitRateMap {
if hitRate < f.lowerHitRate {
log.LogDebugf("opCacheRead: remotecache %v dataPath(%v) is lower hitrate %v", f.localAddr, dataPath, hitRate)
errMetric := exporter.NewCounter("lowerHitRate")
errMetric.AddWithLabels(1, map[string]string{exporter.FlashNode: f.localAddr, exporter.Disk: dataPath, exporter.Err: "LowerHitRate"})
}
}
bgTime2 := stat.BeginStat()
missTaskDone := make(chan struct{})
// try to cache more miss data, but reply to client more quickly
reqSize := 0
for _, source := range req.CacheRequest.Sources {
reqSize += int(source.Size_)
}
if err = f.limitWrite.TryRunAsync(ctx, reqSize, f.waitForCacheBlock, func() {
if block2, err2 := f.cacheEngine.CreateBlock(cr, conn.RemoteAddr().String(), false); err2 != nil {
log.LogWarnf("opCacheRead: CreateBlock failed, req(%v) err(%v)", req, err2)
close(missTaskDone)
return
} else {
block2.InitOnceForCacheRead(f.cacheEngine, cr.Sources, missTaskDone)
}
}); err != nil {
stat.EndStat("MissCacheReadLimit", err, bgTime2, 1)
return
}
if !f.waitForCacheBlock {
stat.EndStat("MissCacheRead:Data is caching", nil, bgTime2, 1)
return proto.ErrorRequireDataIsCaching
}
select {
case <-ctx.Done():
stat.EndStat("MissCacheReadCancel", ctx.Err(), bgTime2, 1)
return ctx.Err()
case <-missTaskDone:
block, err = f.cacheEngine.GetCacheBlockForRead(volume, cr.Inode, cr.FixedFileOffset, cr.Version, req.Size_)
}
stat.EndStat("MissCacheRead", err, bgTime2, 1)
if err != nil {
return err
}
}
bgTime2 := stat.BeginStat()
// reply to client as quick as possible if hit cache
err2 := f.limitRead.RunNoWait(int(req.Size_), false, func() {
err = f.doStreamReadRequest(ctx, conn, req, p, block)
})
if err2 != nil {
err = err2
stat.EndStat("HitCacheRead", err, bgTime2, 1)
return
}
stat.EndStat("HitCacheRead", err, bgTime2, 1)
return
}
func (f *FlashNode) opCacheDelete(conn net.Conn, p *proto.Packet) (err error) {
data := p.Data
defer func() {
if err != nil {
log.LogWarnf("action[opCacheDelete] end, logMsg:%s",
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
} else {
p.PacketOkReply()
}
if e := p.WriteToConn(conn); e != nil {
log.LogErrorf("action[opCacheDelete] write to conn %v", e)
}
}()
if p.Size == 0 {
return proto.ErrorNoCacheDeleteRequest
}
uniKey := string(data)
if log.EnableDebug() {
log.LogDebugf("action[opCacheDelete] delete key(%v) from remote(%v)", uniKey, conn.RemoteAddr().String())
}
pDir := cachengine.MapKeyToDirectory(uniKey)
f.cacheEngine.DeleteCacheBlock(cachengine.GenCacheBlockKeyV2(pDir, uniKey))
return nil
}
func (f *FlashNode) opCachePutBlock(conn net.Conn, p *proto.Packet) (err error) {
bgTime := stat.BeginStat()
reqId := string(p.Arg)
var uniKey string
logPrefix := fmt.Sprintf("action[opCachePutBlock] reqId(%v)", reqId)
defer func() {
stat.EndStat("FlashNode:opCachePutBlock", err, bgTime, 1)
}()
defer func() {
if err != nil {
log.LogWarnf(logPrefix+" uniKey %v end, logMsg:%s", uniKey,
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
if e := p.WriteToConn(conn); e != nil {
log.LogWarnf(logPrefix+" uniKey %v write to conn %v", uniKey, e)
}
}
}()
if len(p.Data) == 0 {
return proto.ErrorUnableToBuildKeyFromPacket
}
req := new(proto.PutBlockHead)
if err = p.UnmarshalDataPb(req); err != nil {
return proto.ErrorParsingCacheWriteHead
}
uniKey = req.UniKey
var ep time.Time
pDir := cachengine.MapKeyToDirectory(uniKey)
blockKey := cachengine.GenCacheBlockKeyV2(pDir, uniKey)
eb, err1 := f.cacheEngine.GetCacheBlockForReadByKey(blockKey)
if err1 == nil {
if log.EnableDebug() {
log.LogDebug(logPrefix+" check block key:"+uniKey+" logMsg:",
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
}
if eb != nil {
ep = eb.GetExpiredTime()
}
err = fmt.Errorf(proto.ErrorBlockAlreadyExistsTpl, uniKey, ep)
return err
}
var allocSize int
if allocSize, err = cachengine.CalcAllocSizeV2(int(req.BlockLen)); err != nil {
return err
}
err1 = nil
if log.EnableDebug() {
log.LogDebug(logPrefix+" create block key:"+uniKey+" logMsg:",
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
}
defer f.missCache.Delete(uniKey)
if cb, err2, created, ci := f.cacheEngine.CreateBlockV2(pDir, uniKey, req.TTL, uint32(allocSize), conn.RemoteAddr().String()); err2 != nil || created {
if err2 != nil {
err = fmt.Errorf(proto.ErrorCreateBlockFailedTpl, cachengine.GenCacheBlockKeyV2(pDir, uniKey), err2.Error())
} else {
if cb != nil {
ep = cb.GetExpiredTime()
}
err = fmt.Errorf(proto.ErrorBlockAlreadyCreatedTpl, cachengine.GenCacheBlockKeyV2(pDir, uniKey), ep)
}
return
} else {
p.PacketOkReply()
defer func() {
if ci != nil {
ci.FreePreAllocatedSize(blockKey)
}
}()
if err1 = p.WriteToConn(conn); err1 != nil {
f.cacheEngine.DeleteCacheBlock(blockKey)
log.LogWarnf(logPrefix+" blockKey %v write to conn %v", blockKey, err1)
return
}
bgTime1 := stat.BeginStat()
defer func() {
stat.EndStat("CachePutBlock:Write", err1, bgTime1, 1)
}()
var (
totalWritten int64
n int
readSize int
)
writeLen := proto.CACHE_BLOCK_PACKET_SIZE + 4
buf := bytespool.Alloc(writeLen)
ch := &proto.CoonHandler{
Completed: make(chan struct{}),
WaitAckChan: make(chan struct{}, 32),
}
defer func() {
bytespool.Free(buf)
if !ch.WaitAckChanClosed {
ch.WaitAckChanClosed = true
close(ch.WaitAckChan)
}
}()
go f.replyPutDataOk(ch, conn, p, logPrefix)
for {
if ch.RemoteError != nil {
err1 = fmt.Errorf(proto.ErrorRemoteErrorTpl, ch.RemoteError)
break
}
readSize = proto.CACHE_BLOCK_PACKET_SIZE
missTaskDone := make(chan struct{})
if totalWritten+int64(readSize) > req.BlockLen {
readSize = int(req.BlockLen - totalWritten)
}
err = f.limitWrite.Run(readSize, true, func() {
defer func() {
close(missTaskDone)
}()
if n, err1 = io.ReadFull(conn, buf[:readSize+proto.CACHE_BLOCK_CRC_SIZE]); err1 != nil {
log.LogWarnf(logPrefix+" blockkey %v read data and crc conn %v", blockKey, err1)
return
}
if n != readSize+proto.CACHE_BLOCK_CRC_SIZE {
err1 = syscall.EBADMSG
return
}
err1 = cb.WriteAtV2(&proto.FlashWriteParam{
Offset: totalWritten,
Size: req.BlockLen,
Data: buf[:readSize],
Crc: buf[readSize : readSize+proto.CACHE_BLOCK_CRC_SIZE],
DataSize: int64(readSize),
})
if err1 != nil {
return
}
cachengine.UpdateWriteBytesMetric(uint64(readSize), cb.GetRootPath())
cachengine.UpdateWriteCountMetric(cb.GetRootPath())
err1 = cb.MaybeWriteCompleted(req.BlockLen)
if err1 != nil {
return
}
})
if err == nil {
<-missTaskDone
}
if err1 != nil || err != nil {
break
} else {
ch.WaitAckChan <- struct{}{}
}
totalWritten += int64(readSize)
if totalWritten == req.BlockLen {
if log.EnableDebug() {
log.LogDebugf(logPrefix+" blokeKey %v total write %v", blockKey, totalWritten)
}
break
}
}
if !ch.WaitAckChanClosed {
ch.WaitAckChanClosed = true
close(ch.WaitAckChan)
}
<-ch.Completed
if ch.RemoteError != nil {
err1 = fmt.Errorf(proto.ErrorRemoteErrorTpl, ch.RemoteError)
}
}
if err1 != nil {
f.cacheEngine.DeleteCacheBlock(blockKey)
err = err1
return
}
return err
}
func (f *FlashNode) replyPutDataOk(ch *proto.CoonHandler, conn net.Conn, p *proto.Packet, logPrefix string) {
defer close(ch.Completed)
replyPacket := proto.NewPacket()
replyPacket.ReqID = p.ReqID
replyPacket.Opcode = p.Opcode
replyPacket.StartT = p.StartT
replyPacket.PacketOkReply()
for range ch.WaitAckChan {
if ch.RemoteError = replyPacket.WriteToConn(conn); ch.RemoteError != nil {
log.LogWarnf(logPrefix+" reply to conn %v for write data", ch.RemoteError)
return
}
}
}
func (f *FlashNode) opCacheBatchObjectGet(conn net.Conn, p *proto.Packet) (err error) {
bgTime := stat.BeginStat()
defer func() {
if err != nil {
log.LogWarnf("action[opCacheBatchObjectGet] write to conn %v", err)
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
if e := p.WriteToConn(conn); e != nil {
log.LogWarnf("action[opCacheBatchObjectGet] write to conn %v", e)
}
}
stat.EndStat("FlashNode:opCacheBatchObjectGet", err, bgTime, 1)
}()
connAddr := conn.RemoteAddr().String()
req := new(proto.BatchReadReq)
if err = p.UnmarshalDataPb(req); err != nil {
return
}
resChan := make(chan *proto.ReadResult, len(req.Items))
bufChan := make(chan []byte, len(req.Items))
bg1 := stat.BeginStat()
var wg sync.WaitGroup
for _, item := range req.Items {
wg.Add(1)
_ = f.batchReadPool.AsyncRunNoHang(func(ri *proto.BatchReadItem) func() {
return func() {
defer wg.Done()
r, b := f.smallObjectGet(ri, connAddr, req.Deadline)
resChan <- r
bufChan <- b
}
}(item))
}
wg.Wait()
stat.EndStat("FlashNode:opCacheBatchComplete", err, bg1, 1)
close(resChan)
close(bufChan)
defer func() {
for b := range bufChan {
if b != nil {
bytespool.Free(b)
}
}
}()
results := make([]*proto.ReadResult, 0)
for a := range resChan {
results = append(results, a)
}
resp := new(proto.BatchReadResp)
resp.Results = results
p.ResultCode = proto.OpOk
if err = p.MarshalDataPb(resp); err == nil {
if e := p.WriteToConn(conn); e != nil {
log.LogWarnf("action[opCacheBatchObjectGet] write to conn %v", e)
}
}
return
}
func (f *FlashNode) smallObjectGet(req *proto.BatchReadItem, connAddr string, deadLine uint64) (result *proto.ReadResult, buf []byte) {
result = &proto.ReadResult{
ReqId: req.ReqId,
ResultCode: uint32(proto.OpOk),
}
var err error
bgTime := stat.BeginStat()
buf = bytespool.Alloc(proto.SMALL_OBJECT_BLOCK_SIZE)
reqID := req.Tid
uniKey := req.Key
if log.EnableDebug() {
log.LogDebugf("smallObjectGet req(%v) key(%v) id(%v) begin", req, uniKey, reqID)
}
defer func() {
if err != nil {
if proto.IsFlashNodeLimitError(err) || proto.IsCacheMissError(err) {
if log.EnableDebug() {
log.LogDebugf("action[smallObjectGet] req(%s) remoteAddr(%s) deadLine(%d) err:%v", req.String(), connAddr, deadLine, err)
}
} else {
log.LogWarnf("action[smallObjectGet] req(%s) remoteAddr(%s) deadLine(%d) err:%v", req.String(), connAddr, deadLine, err)
}
result.ResultCode = uint32(proto.OpErr)
result.Data = ([]byte)(err.Error())
}
stat.EndStat("FlashNode:smallObjectGet", err, bgTime, 1)
}()
pDir := cachengine.MapKeyToDirectory(uniKey)
blockKey := cachengine.GenCacheBlockKeyV2(pDir, uniKey)
f.updateSlotStat(req.Slot)
block, err := f.cacheEngine.GetCacheBlockForReadByKey(blockKey)
if err != nil {
// if not find block, check whether should cache
err = f.shouldCache(uniKey)
return
}
err = block.VerifyObjectReq(req.Offset, req.Size_)
stat.EndStat("LoadBlock", nil, bgTime, 1)
if err != nil {
createTime := block.GetCreateTime()
if time.Since(createTime) > 2*time.Minute {
f.cacheEngine.DeleteCacheBlock(blockKey)
err = f.shouldCache(uniKey)
}
return
}
ctx, ctxCancel := context.WithTimeout(context.Background(), time.Duration(f.handleReadTimeout)*time.Millisecond)
defer ctxCancel()
dataSize := block.LoadDataSize()
readDiskSize := util.Min(proto.CACHE_BLOCK_PACKET_SIZE, int(uint64(dataSize)-req.Offset))
var readDiskErr error
var readCRC uint32
var diskDataBuf []byte
if uint64(readDiskSize) <= req.Size_ {
diskDataBuf = buf
} else {
diskDataBuf = bytespool.Alloc(proto.CACHE_BLOCK_PACKET_SIZE)
}
if err = f.limitRead.AcquireDiskFlow(readDiskSize); err == nil {
readCRC, readDiskErr = block.Read(ctx, diskDataBuf[:readDiskSize], int64(req.Offset), int64(readDiskSize), f.waitForCacheBlock, true)
}
if readDiskErr != nil {
err = readDiskErr
}
if err == nil {
if uint64(readDiskSize) == req.Size_ {
result.CRC = readCRC
} else {
diskCrc := crc32.ChecksumIEEE(diskDataBuf[:readDiskSize])
if readCRC != diskCrc {
err = fmt.Errorf(proto.ErrorInconsistentCRCTpl, readCRC, diskCrc)
} else {
result.CRC = crc32.ChecksumIEEE(diskDataBuf[:req.Size_])
}
}
if err == nil {
if uint64(readDiskSize) > req.Size_ {
copy(buf[:req.Size_], diskDataBuf)
}
result.Data = buf[:req.Size_]
f.metrics.updateReadCountMetric(block.GetRootPath())
f.metrics.updateReadBytesMetric(req.Size_, block.GetRootPath())
}
}
if uint64(readDiskSize) > req.Size_ {
bytespool.Free(diskDataBuf)
}
return
}
func (f *FlashNode) opCacheObjectGet(conn net.Conn, p *proto.Packet) (err error) {
var uniKey string
bgTime := stat.BeginStat()
defer func() {
stat.EndStat("FlashNode:opCacheObjectGet", err, bgTime, 1)
}()
reqID := string(p.Arg)
defer func() {
if err != nil {
if proto.IsFlashNodeLimitError(err) || proto.IsCacheMissError(err) {
if log.EnableDebug() {
log.LogDebugf("action[opCacheObjectGet]reqID[%v] key:[%s], logMsg:%s", reqID, uniKey,
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
}
} else {
log.LogWarnf("action[opCacheObjectGet]reqID[%v] key:[%s], logMsg:%s", reqID, uniKey,
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
}
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
if e := p.WriteToConn(conn); e != nil {
log.LogWarnf("action[opCacheObjectGet] reqID[%v] key:[%s] write to conn %v", reqID, uniKey, e)
}
}
}()
req := new(proto.CacheReadRequestBase)
if err = p.UnmarshalDataPb(req); err != nil {
return
}
uniKey = req.Key
if log.EnableDebug() {
log.LogDebugf("opCacheObjectGet req(%v) key(%v) id(%v) begin", req, uniKey, reqID)
}
pDir := cachengine.MapKeyToDirectory(uniKey)
blockKey := cachengine.GenCacheBlockKeyV2(pDir, uniKey)
f.updateSlotStat(req.Slot)
block, err := f.cacheEngine.GetCacheBlockForReadByKey(blockKey)
if err != nil {
// if not find block, check whether should cache
err = f.shouldCache(uniKey)
return
}
err = block.VerifyObjectReq(req.Offset, req.Size_)
stat.EndStat("LoadBlock", nil, bgTime, 1)
if err != nil {
createTime := block.GetCreateTime()
if time.Since(createTime) > 2*time.Minute {
f.cacheEngine.DeleteCacheBlock(blockKey)
err = f.shouldCache(uniKey)
}
return
}
bgTime2 := stat.BeginStat()
err = block.CheckRateLimit(int(req.Size_), uint64(f.keyRateLimitThreshold))
if err != nil {
stat.EndStat("HitCacheRead:KeyLimit", err, bgTime2, 1)
return
}
ctx, ctxCancel := context.WithTimeout(context.Background(), time.Duration(f.handleReadTimeout)*time.Millisecond)
defer ctxCancel()
// reply to client as quick as possible if hit cache
err = f.doObjectReadRequest(ctx, conn, req, p, block, reqID)
if err != nil {
stat.EndStat("HitCacheRead", err, bgTime2, 1)
return
}
stat.EndStat("HitCacheRead", err, bgTime2, 1)
return
}
func (f *FlashNode) opFlashNodeScan(conn net.Conn, p *proto.Packet) (err error) {
data := p.Data
responseAckOKToMaster(conn, p)
var (
req = &proto.FlashNodeManualTaskRequest{}
resp = &proto.FlashNodeManualTaskResponse{
FlashNode: f.localAddr,
}
adminTask = &proto.AdminTask{
Request: req,
}
)
decoder := json.NewDecoder(bytes.NewBuffer(data))
decoder.UseNumber()
if err = decoder.Decode(adminTask); err != nil {
resp.Status = proto.TaskFailed
resp.Done = true
resp.StartErr = err.Error()
adminTask.Response = resp
f.respondToMaster(adminTask)
return
}
err = f.startTaskScan(adminTask)
f.respondToMaster(adminTask)
return
}
func (f *FlashNode) opFlashNodeTaskCommand(conn net.Conn, p *proto.Packet) error {
data := p.Data
var err error
defer func() {
if err != nil {
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
} else {
p.PacketOkReply()
}
if e := p.WriteToConn(conn); e != nil {
log.LogErrorf("action[opFlashNodeTaskCommand] write to conn %v", e)
}
}()
req := &proto.FlashNodeManualTaskCommand{}
adminTask := &proto.AdminTask{
Request: req,
}
decode := json.NewDecoder(bytes.NewBuffer(data))
decode.UseNumber()
if err = decode.Decode(adminTask); err != nil {
log.LogErrorf("decode FlashNodeManualTaskCommand error: %s", err.Error())
return err
}
mScanner, ok := f.manualScanners.Load(req.ID)
if !ok {
err = fmt.Errorf(proto.ErrorTaskIDNotExistTpl, req.ID)
return err
}
scanner := mScanner.(*ManualScanner)
scanner.processCommand(req.Command)
return nil
}
func (f *FlashNode) startTaskScan(adminTask *proto.AdminTask) (err error) {
request := adminTask.Request.(*proto.FlashNodeManualTaskRequest)
log.LogInfof("startTaskScan: scan task(%v) received!", request.Task)
resp := &proto.FlashNodeManualTaskResponse{}
adminTask.Response = resp
if _, ok := f.manualScanners.Load(request.Task.Id); ok {
log.LogInfof("startManualScan: scan task(%v) is already running!", request.Task)
return
}
var (
metaWrapper *meta.MetaWrapper
extentClient *stream.ExtentClient
)
metaWrapper, extentClient, err = f.getValidViewInfo(request)
if err != nil {
resp.ID = request.Task.Id
resp.Volume = request.Task.VolName
resp.Status = proto.TaskFailed
resp.Done = true
resp.StartErr = err.Error()
return
}
f.scannerMutex.Lock()
if _, ok := f.manualScanners.Load(request.Task.Id); ok {
log.LogInfof("startManualScan: scan task(%v) is already running!", request.Task)
f.scannerMutex.Unlock()
return
}
scanner := NewManualScanner(adminTask, f, metaWrapper, extentClient)
f.manualScanners.Store(scanner.ID, scanner)
f.scannerMutex.Unlock()
resp.Status = proto.TaskStart
if err = scanner.Start(); err != nil {
log.LogErrorf("start scanner[%v] failed err: %v", scanner.ID, err)
return
}
targetDir := scanner.manualTask.GetPathPrefix()
if targetDir == "" {
targetDir = "/"
}
wup := &proto.WarmUpPathInfo{
VolName: scanner.manualTask.VolName,
DirPath: targetDir,
}
wup.SetExpiration(time.Duration(scanner.manualTask.ManualTaskConfig.WarmUpPathExpire) * time.Second)
f.enableWarmUpPaths = true
f.warmUpPaths.Store(uuid.New().String(), wup)
return
}
func (f *FlashNode) getValidViewInfo(req *proto.FlashNodeManualTaskRequest) (metaWrapper *meta.MetaWrapper, extentClient *stream.ExtentClient, err error) {
task := req.Task
var volumeInfo *proto.SimpleVolView
volumeInfo, err = f.mc.AdminAPI().GetVolumeSimpleInfo(task.VolName)
if err != nil {
log.LogErrorf("NewVolume: get volume info from master failed: volume(%v) err(%v)", task.VolName, err)
return
}
if volumeInfo.Status == 1 {
log.LogWarnf("NewVolume: volume has been marked for deletion: volume(%v) status(%v - 0:normal/1:markDelete)",
task.VolName, volumeInfo.Status)
err = proto.ErrVolNotExists
return
}
metaConfig := &meta.MetaConfig{
Volume: task.VolName,
Masters: f.masters,
Authenticate: false,
ValidateOwner: false,
InnerReq: true,
MetaSendTimeout: 600,
}
if metaWrapper, err = meta.NewMetaWrapper(metaConfig); err != nil {
log.LogErrorf("NewMetaWrapper err: %v", err)
return
}
if task.Action == proto.FlashManualWarmupAction {
extentConfig := &stream.ExtentConfig{
Volume: task.VolName,
Masters: f.masters,
FollowerRead: false,
OnGetExtents: metaWrapper.GetExtents,
OnRenewalForbiddenMigration: metaWrapper.RenewalForbiddenMigration,
VolStorageClass: volumeInfo.VolStorageClass,
VolAllowedStorageClass: volumeInfo.AllowedStorageClass,
OnForbiddenMigration: metaWrapper.ForbiddenMigration,
MetaWrapper: metaWrapper,
NeedRemoteCache: true,
HeartBeatPing: true,
ForceRemoteCache: true,
}
log.LogInfof("[NewS3Scanner] extentConfig: vol(%v) volStorageClass(%v) allowedStorageClass(%v), followerRead(%v)",
extentConfig.Volume, extentConfig.VolStorageClass, extentConfig.VolAllowedStorageClass, extentConfig.FollowerRead)
if extentClient, err = stream.NewExtentClient(extentConfig); err != nil {
log.LogErrorf("NewExtentClient err: %v", err)
return
}
}
return
}
func (f *FlashNode) doStreamReadRequest(ctx context.Context, conn net.Conn, req *proto.CacheReadRequest, p *proto.Packet,
block *cachengine.CacheBlock,
) (err error) {
const action = "action[doStreamReadRequest]"
needReplySize := uint32(req.Size_)
offset := int64(req.Offset)
defer func() {
if err != nil {
// too many caching logs
if strings.Compare(err.Error(), "require data is caching") != 0 {
log.LogWarnf("%s cache block(%v) err:%v", action, block.String(), err)
}
} else {
f.metrics.updateReadCountMetric(block.GetRootPath())
f.metrics.updateReadBytesMetric(req.Size_, block.GetRootPath())
}
}()
for needReplySize > 0 {
err = nil
reply := proto.NewPacket()
reply.ReqID = p.ReqID
reply.StartT = p.StartT
currReadSize := uint32(util.Min(int(needReplySize), util.ReadBlockSize))
var bufOnce sync.Once
buf, bufErr := proto.Buffers.Get(util.ReadBlockSize)
bufRelease := func() {
bufOnce.Do(func() {
if bufErr == nil {
proto.Buffers.Put(reply.Data[:util.ReadBlockSize])
}
})
}
if bufErr != nil {
buf = make([]byte, currReadSize)
}
reply.Data = buf[:currReadSize]
reply.ExtentOffset = offset
p.Size = currReadSize
p.ExtentOffset = offset
reply.CRC, err = block.Read(ctx, reply.Data[:], offset, int64(currReadSize), f.waitForCacheBlock, false)
if err != nil {
bufRelease()
return
}
p.CRC = reply.CRC
reply.Size = currReadSize
reply.ResultCode = proto.OpOk
reply.Opcode = p.Opcode
p.ResultCode = proto.OpOk
bgTime := stat.BeginStat()
if err = reply.WriteToConn(conn); err != nil {
bufRelease()
log.LogErrorf("%s volume:[%s] %s", action, req.CacheRequest.Volume,
reply.LogMessage(reply.GetOpMsg(), conn.RemoteAddr().String(), reply.StartT, err))
return
}
stat.EndStat("HitCacheRead:ReplyToClient", err, bgTime, 1)
needReplySize -= currReadSize
offset += int64(currReadSize)
bufRelease()
if log.EnableInfo() {
log.LogInfof("%s ReqID[%d] volume:[%s] reply[%s] block[%s]", action, p.ReqID, req.CacheRequest.Volume,
reply.LogMessage(reply.GetOpMsg(), conn.RemoteAddr().String(), reply.StartT, err), block.String())
}
}
p.PacketOkReply()
return
}
func (f *FlashNode) doObjectReadRequest(ctx context.Context, conn net.Conn, req *proto.CacheReadRequestBase, p *proto.Packet,
block *cachengine.CacheBlock, reqId string,
) (err error) {
action := fmt.Sprintf("action[doObjectReadRequest] reqId(%v)", reqId)
offset := int64(req.Offset)
defer func() {
if err != nil {
// too many caching logs
if strings.Compare(err.Error(), "require data is caching") != 0 {
log.LogWarnf("%s cache block(%v) err:%v", action, block.GetBlockKey(), err)
}
} else {
f.metrics.updateReadCountMetric(block.GetRootPath())
f.metrics.updateReadBytesMetric(req.Size_, block.GetRootPath())
}
}()
// first reply to client
dataSize := block.LoadDataSize()
firstReply := proto.NewPacket()
firstReply.ReqID = p.ReqID
firstReply.Size = uint32(dataSize)
firstReply.ResultCode = proto.OpOk
firstReply.Opcode = p.Opcode
firstReply.StartT = p.StartT
end := offset + int64(req.Size_)
if err = firstReply.WriteToConn(conn); err != nil {
log.LogErrorf("action[doObjectReadRequest] key:[%s] %s", block.GetBlockKey(),
firstReply.LogMessage(firstReply.GetOpMsg(), conn.RemoteAddr().String(), firstReply.StartT, err))
return
}
// reply data to client
var errInner error
buf := bytespool.Alloc(proto.CACHE_BLOCK_PACKET_SIZE)
defer bytespool.Free(buf)
var alignedOffset int64
var readDiskSize uint32
readAndReply := func() {
reply := proto.NewPacket()
reply.ReqID = p.ReqID
reply.StartT = p.StartT
reply.Data = buf
reply.KernelOffset = uint64(offset)
reply.ExtentOffset = offset - alignedOffset
p.Size = proto.CACHE_BLOCK_PACKET_SIZE
p.ExtentOffset = offset
reply.CRC, errInner = block.Read(ctx, reply.Data[:readDiskSize], alignedOffset, int64(readDiskSize), f.waitForCacheBlock, true)
if errInner != nil {
return
}
p.CRC = reply.CRC
realNeedSize := uint32(util.Min(int(proto.CACHE_BLOCK_PACKET_SIZE-reply.ExtentOffset), int(end-offset)))
reply.Size = realNeedSize
reply.ResultCode = proto.OpOk
reply.Opcode = p.Opcode
p.ResultCode = proto.OpOk
bgTime := stat.BeginStat()
if errInner = reply.WriteToConnForOCS(conn, readDiskSize); errInner != nil {
log.LogWarnf("%s key:[%s] %s", action, block.GetBlockKey(),
reply.LogMessage(reply.GetOpMsg(), conn.RemoteAddr().String(), reply.StartT, errInner))
return
}
stat.EndStat("HitCacheRead:ReplyToClient", errInner, bgTime, 1)
offset = alignedOffset + int64(readDiskSize)
if log.EnableInfo() {
log.LogInfof("%s ReqID[%d] key:[%s] reply[%s] block[%s]", action, p.ReqID, block.GetBlockKey(),
reply.LogMessage(reply.GetOpMsg(), conn.RemoteAddr().String(), reply.StartT, errInner), block.GetBlockKey())
}
}
var keepAlive bool
for {
alignedOffset = offset / proto.CACHE_BLOCK_PACKET_SIZE * proto.CACHE_BLOCK_PACKET_SIZE
readDiskSize = uint32(util.Min(proto.CACHE_BLOCK_PACKET_SIZE, int(dataSize-alignedOffset)))
if !keepAlive {
err = f.limitRead.RunNoWait(int(readDiskSize), false, readAndReply)
keepAlive = true
} else {
err = f.limitRead.Run(int(readDiskSize), true, readAndReply)
}
if err != nil {
return
}
if errInner != nil {
err = errInner
return
}
if uint64(offset) >= req.Offset+req.Size_ {
break
}
}
p.PacketOkReply()
return
}
func (f *FlashNode) opCachePrepare(conn net.Conn, p *proto.Packet) (err error) {
action := "action[opCachePrepare]"
var volume string
bgTime := stat.BeginStat()
defer func() {
if err != nil {
log.LogErrorf("%s volume:[%s] %s", action, volume,
p.LogMessage(p.GetOpMsg(), conn.RemoteAddr().String(), p.StartT, err))
p.PacketErrorWithBody(proto.OpErr, ([]byte)(err.Error()))
if e := p.WriteToConn(conn); e != nil {
log.LogErrorf("%s write to conn %v", action, e)
}
}
stat.EndStat("FlashNode:opCachePrepare", err, bgTime, 1)
}()
req := new(proto.CachePrepareRequest)
if err = p.UnmarshalDataPb(req); err != nil {
return
}
if req.CacheRequest == nil {
err = proto.ErrorNoCachePrepareRequest
return
}
f.updateSlotStat(req.CacheRequest.Slot)
volume = req.CacheRequest.Volume
if err = f.cacheEngine.PrepareCache(p.ReqID, req.CacheRequest, conn.RemoteAddr().String()); err != nil {
log.LogErrorf("%s prepare %v", action, err)
return
}
p.PacketOkReply()
if e := p.WriteToConn(conn); e != nil {
log.LogErrorf("%s write to conn volume:%s %v", action, volume, e)
}
if len(req.FlashNodes) > 0 {
f.dispatchRequestToFollowers(req)
}
return
}
func (f *FlashNode) dispatchRequestToFollowers(request *proto.CachePrepareRequest) {
req := &proto.CachePrepareRequest{
CacheRequest: request.CacheRequest,
FlashNodes: make([]string, 0),
}
wg := sync.WaitGroup{}
for idx, addr := range request.FlashNodes {
if addr == f.localAddr {
continue
}
wg.Add(1)
log.LogDebugf("dispatchRequestToFollowers: try to prepare on addr:%s (%d/%d)", addr, idx, len(request.FlashNodes))
go func(addr string) {
defer wg.Done()
if err := f.sendPrepareRequest(addr, req); err != nil {
log.LogErrorf("dispatchRequestToFollowers: failed to distribute request to addr(%v) err(%v)", addr, err)
}
}(addr)
}
wg.Wait()
}
func (f *FlashNode) sendPrepareRequest(addr string, req *proto.CachePrepareRequest) (err error) {
action := "action[sendPrepareRequest]"
conn, err := f.connPool.GetConnect(addr)
if err != nil {
return err
}
defer func() {
f.connPool.PutConnect(conn, err != nil)
}()
log.LogDebugf("%s to addr:%s request:%v", action, addr, req)
followerPacket := proto.NewPacketReqID()
followerPacket.Opcode = proto.OpFlashNodeCachePrepare
if err = followerPacket.MarshalDataPb(req); err != nil {
log.LogWarnf("%s failed to MarshalDataPb (%+v) err(%v)", action, followerPacket, err)
return err
}
if err = followerPacket.WriteToNoDeadLineConn(conn); err != nil {
log.LogWarnf("%s failed to write to addr(%s) err(%v)", action, addr, err)
return err
}
reply := proto.NewPacket()
if err = reply.ReadFromConn(conn, 30); err != nil {
log.LogWarnf("%s read reply(%v) from addr(%s) err(%v)", action, reply, addr, err)
return err
}
if reply.ResultCode != proto.OpOk {
log.LogWarnf("%s reply(%v) from addr(%s) ResultCode(%d)", action, reply, addr, reply.ResultCode)
return fmt.Errorf(proto.ErrorResultCodeNOKTpl, reply.ResultCode)
}
return nil
}
func (f *FlashNode) opSetReadIOLimits(conn net.Conn, p *proto.Packet) (err error) {
data := p.Data
req := &proto.FlashNodeSetIOLimitsRequest{}
adminTask := &proto.AdminTask{
Request: req,
}
update := false
decode := json.NewDecoder(bytes.NewBuffer(data))
decode.UseNumber()
if err = decode.Decode(adminTask); err == nil {
log.LogDebugf("opSetReadIOLimits req: %v", req)
if req.Iocc != -1 {
f.diskReadIocc = req.Iocc
update = true
}
if req.Flow != -1 {
f.localChangeReadFlow = true
f.diskReadFlow = req.Flow
update = true
}
if req.Factor != -1 {
f.diskReadIoFactorFlow = req.Factor
update = true
}
if update {
f.limitRead.ResetIOEx(f.diskReadIocc*len(f.disks), f.diskReadIoFactorFlow, f.handleReadTimeout)
f.limitRead.ResetFlow(f.diskReadFlow)
}
} else {
log.LogErrorf("decode FlashNodeSetIOLimitsRequest error: %s", err.Error())
}
p.PacketOkReply()
return
}
func (f *FlashNode) opApplyWarmupMetaToken(conn net.Conn, p *proto.Packet) (err error) {
reqData := p.Data
var clientId string
if p.ArgLen != 0 {
clientId = string(p.Arg[:int(p.ArgLen)])
}
if len(reqData) == 0 {
p.PacketErrorWithBody(proto.OpErr, []byte("empty request data"))
if err = p.WriteToConn(conn); err != nil {
log.LogErrorf("opApplyWarmupMetaToken response: %s", err.Error())
}
return
}
requestType := reqData[0]
now := time.Now().Unix()
f.currentWarmUpWorkerMutex.Lock()
defer f.currentWarmUpWorkerMutex.Unlock()
switch requestType {
case proto.WarmupMetaTokenApply: // apply
if len(f.currentWarmUpWorkers) >= f.warmupMetaTotalToken {
p.PacketOkReply()
p.Data = []byte{0}
p.Size = 1
log.LogWarnf("opApplyWarmupMetaToken: client %s apply failed, current workers %d >= total token %d",
clientId, len(f.currentWarmUpWorkers), f.warmupMetaTotalToken)
} else {
f.currentWarmUpWorkers[clientId] = now
p.PacketOkReply()
p.Data = []byte{1}
p.Size = 1
log.LogInfof("opApplyWarmupMetaToken: client %s apply success, current workers %d",
clientId, len(f.currentWarmUpWorkers))
}
case proto.WarmupMetaTokenRenew: // renew
if _, exists := f.currentWarmUpWorkers[clientId]; exists {
f.currentWarmUpWorkers[clientId] = now
p.PacketOkReply()
p.Data = []byte{1}
p.Size = 1
log.LogDebugf("opApplyWarmupMetaToken: client %s renew success", clientId)
} else {
p.PacketOkReply()
p.Data = []byte{0}
p.Size = 1
log.LogWarnf("opApplyWarmupMetaToken: client %s renew failed, not found", clientId)
}
case proto.WarmupMetaTokenRelease: // release
if _, exists := f.currentWarmUpWorkers[clientId]; exists {
delete(f.currentWarmUpWorkers, clientId)
p.PacketOkReply()
p.Data = []byte{1}
p.Size = 1
log.LogInfof("opApplyWarmupMetaToken: client %s release success, current workers %d",
clientId, len(f.currentWarmUpWorkers))
} else {
p.PacketOkReply()
p.Data = []byte{0}
p.Size = 1
log.LogWarnf("opApplyWarmupMetaToken: client %s release failed, not found", clientId)
}
default:
p.PacketErrorWithBody(proto.OpErr, []byte("invalid request type"))
}
if err = p.WriteToConn(conn); err != nil {
log.LogErrorf("opApplyWarmupMetaToken response: %s", err.Error())
}
if log.EnableInfo() {
log.LogInfof("opApplyWarmupMetaToken remoteaddr(%v) clientId(%v) requestType(%d)",
conn.RemoteAddr(), clientId, requestType)
}
return
}
func (f *FlashNode) opSetWriteIOLimits(conn net.Conn, p *proto.Packet) (err error) {
data := p.Data
req := &proto.FlashNodeSetIOLimitsRequest{}
adminTask := &proto.AdminTask{
Request: req,
}
update := false
decode := json.NewDecoder(bytes.NewBuffer(data))
decode.UseNumber()
if err = decode.Decode(adminTask); err == nil {
log.LogDebugf("opSetReadIOLimits req: %v", req)
if req.Iocc != -1 {
f.diskWriteIocc = req.Iocc
update = true
}
if req.Flow != -1 {
f.localChangeWriteFlow = true
f.diskWriteFlow = req.Flow
update = true
}
if req.Factor != -1 {
f.diskWriteIoFactorFlow = req.Factor
update = true
}
if update {
f.limitWrite.ResetIOEx(f.diskWriteIocc*len(f.disks), f.diskWriteIoFactorFlow, f.handleReadTimeout)
f.limitWrite.ResetFlow(f.diskWriteFlow)
}
} else {
log.LogErrorf("decode opSetWriteIOLimits error: %s", err.Error())
}
p.PacketOkReply()
return
}
//nolint:unused // used for new read operation
func (f *FlashNode) shouldCache(key string) error {
var count int32
cm := f.missCache.Get(key)
if cm == nil {
count = f.missCache.Increment(key)
} else {
count = atomic.AddInt32(&cm.MissCount, 1)
}
if count == f.hotKeyMissCount {
return fmt.Errorf("%v, now hit %v", proto.ErrorNotExistShouldCache.Error(), count)
} else {
return fmt.Errorf("%v, now hit %v", proto.ErrorNotExistShouldNotCache.Error(), count)
}
}
func responseAckOKToMaster(conn net.Conn, p *proto.Packet) {
p.PacketOkReply()
if err := p.WriteToConn(conn); err != nil {
log.LogErrorf("ack master response: %s", err.Error())
}
}