SteveYurongSu commented on code in PR #14717:
URL: https://github.com/apache/iotdb/pull/14717#discussion_r1952454235
##########
iotdb-core/datanode/src/main/java/org/apache/iotdb/db/storageengine/StorageEngine.java:
##########
@@ -910,17 +910,36 @@ public TSStatus writeLoadTsFileNode(
LoadTsFileRateLimiter.getInstance().acquire(pieceNode.getDataSize());
+ final DataRegion dataRegion = getDataRegion(dataRegionId);
+ if (dataRegion == null) {
+ LOGGER.warn(
+ "DataRegion {} not found on this DataNode when writing piece node"
+ + "of TsFile {} (maybe due to region migration), will skip.",
+ dataRegionId,
+ pieceNode.getTsFile());
+ return RpcUtils.SUCCESS_STATUS;
+ }
+
try {
- loadTsFileManager.writeToDataRegion(getDataRegion(dataRegionId),
pieceNode, uuid);
+ loadTsFileManager.writeToDataRegion(dataRegion, pieceNode, uuid);
} catch (IOException e) {
- LOGGER.error(
+ LOGGER.warn(
"IO error when writing piece node of TsFile {} to DataRegion {}.",
pieceNode.getTsFile(),
dataRegionId,
e);
status.setCode(TSStatusCode.LOAD_FILE_ERROR.getStatusCode());
status.setMessage(e.getMessage());
return status;
+ } catch (Exception e) {
+ LOGGER.warn(
+ "Exception occurred when writing piece node of TsFile {} to
DataRegion {}.",
+ pieceNode.getTsFile(),
+ dataRegionId,
+ e);
+ status.setCode(TSStatusCode.INTERNAL_SERVER_ERROR.getStatusCode());
Review Comment:
```suggestion
status.setCode(TSStatusCode.LOAD_FILE_ERROR.getStatusCode());
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
