xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

chore: more logic for migrate with speed limited

This commit is contained in:
Cary Xu 2022-09-25 22:30:48 +08:00
parent 132425b194
commit b101e425c3
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
source/dnode/vnode/src/tsdb/tsdbReaderWriter.c
View File

@ -622,13 +622,17 @@ static int64_t tsdbFSendFile(TdFilePtr pOutFD, TdFilePtr pInFD, int64_t size, in
} }
int64_t offset = 0; int64_t offset = 0;
int64_t tBytes = 0;
int64_t nBytes = 0; int64_t nBytes = 0;
int64_t startMs = 0; int64_t startMs = 0;
int64_t cost = 0; int64_t cost = 0;
while ((offset + speed) < size) { while ((offset + speed) < size) {
startMs = taosGetTimestampMs(); startMs = taosGetTimestampMs();
nBytes += taosFSendFile(pOutFD, pInFD, &offset, speed); if ((nBytes = taosFSendFile(pOutFD, pInFD, &offset, speed)) < 0) {
return nBytes;
}
cost = taosGetTimestampMs() - startMs; cost = taosGetTimestampMs() - startMs;
tBytes += nBytes;
if (cost < 0) { if (cost < 0) {
taosMsleep(1000); taosMsleep(1000);
@ -637,9 +641,12 @@ static int64_t tsdbFSendFile(TdFilePtr pOutFD, TdFilePtr pInFD, int64_t size, in
} }
} }
if (offset < size) { if (offset < size) {
nBytes += taosFSendFile(pOutFD, pInFD, &offset, size - offset); if ((nBytes = taosFSendFile(pOutFD, pInFD, &offset, size - offset)) < 0) {
return nBytes;
}
tBytes += nBytes;
} }
return nBytes; return tBytes;
} }
int32_t tsdbDFileSetCopy(STsdb *pTsdb, SDFileSet *pSetFrom, SDFileSet *pSetTo, int32_t maxSpeed) { int32_t tsdbDFileSetCopy(STsdb *pTsdb, SDFileSet *pSetFrom, SDFileSet *pSetTo, int32_t maxSpeed) {