Skip to content

Commit c030c64

Browse files
authored
Add logs for migration related daily IT (#16175)
1 parent a0f31fe commit c030c64

2 files changed

Lines changed: 15 additions & 2 deletions

File tree

integration-test/src/test/java/org/apache/iotdb/confignode/it/regionmigration/IoTDBRegionOperationReliabilityITFramework.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -276,6 +276,7 @@ public void generalTestWithAllOptions(
276276
try {
277277
awaitUntilSuccess(
278278
client,
279+
selectedRegion,
279280
migrateRegionPredicate,
280281
Optional.of(destDataNode),
281282
Optional.of(originalDataNode));
@@ -563,6 +564,7 @@ private static void awaitUntilFlush(Statement statement, int originalDataNode) t
563564

564565
protected static void awaitUntilSuccess(
565566
SyncConfigNodeIServiceClient client,
567+
int selectedRegion,
566568
Predicate<TShowRegionResp> predicate,
567569
Optional<Integer> dataNodeExpectInRegionGroup,
568570
Optional<Integer> dataNodeExpectNotInRegionGroup) {
@@ -577,6 +579,7 @@ protected static void awaitUntilSuccess(
577579
() -> {
578580
try {
579581
TShowRegionResp resp = clientRef.get().showRegion(new TShowRegionReq());
582+
lastTimeDataNodes.set(getRegionMap(resp.getRegionInfoList()).get(selectedRegion));
580583
return predicate.test(resp);
581584
} catch (TException e) {
582585
clientRef.set(

integration-test/src/test/java/org/apache/iotdb/confignode/it/regionmigration/pass/commit/IoTDBRegionGroupExpandAndShrinkForIoTV1IT.java

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,12 @@ private void regionGroupExpand(
139139
return dataNodes.contains(targetDataNode);
140140
};
141141

142-
awaitUntilSuccess(client, expandRegionPredicate, Optional.of(targetDataNode), Optional.empty());
142+
awaitUntilSuccess(
143+
client,
144+
selectedRegion,
145+
expandRegionPredicate,
146+
Optional.of(targetDataNode),
147+
Optional.empty());
143148

144149
LOGGER.info("Region {} has expanded to DataNode {}", selectedRegion, targetDataNode);
145150
}
@@ -167,7 +172,12 @@ private void regionGroupShrink(
167172
return !dataNodes.contains(targetDataNode);
168173
};
169174

170-
awaitUntilSuccess(client, shrinkRegionPredicate, Optional.empty(), Optional.of(targetDataNode));
175+
awaitUntilSuccess(
176+
client,
177+
selectedRegion,
178+
shrinkRegionPredicate,
179+
Optional.empty(),
180+
Optional.of(targetDataNode));
171181

172182
LOGGER.info("Region {} has shrunk from DataNode {}", selectedRegion, targetDataNode);
173183
}

0 commit comments

Comments
 (0)