klsince commented on a change in pull request #7319: URL: https://github.com/apache/pinot/pull/7319#discussion_r704593059
########## File path: pinot-core/src/main/java/org/apache/pinot/core/data/manager/BaseTableDataManager.java ########## @@ -256,4 +267,147 @@ public void addSegmentError(String segmentName, SegmentErrorInfo segmentErrorInf .collect(Collectors.toMap(map -> map.getKey().getSecond(), Map.Entry::getValue)); } } + + @Override + public void addOrReplaceSegment(String segmentName, IndexLoadingConfig indexLoadingConfig, + SegmentMetadata localMetadata, SegmentZKMetadata zkMetadata, boolean forceDownload) + throws Exception { + if (!forceDownload && !isNewSegment(zkMetadata, localMetadata)) { + LOGGER.info("Segment: {} of table: {} has crc: {} same as before, already loaded, do nothing", segmentName, + _tableNameWithType, localMetadata.getCrc()); + return; + } + + // If not forced to download, then try to recover if no local metadata is provided. + if (!forceDownload && localMetadata == null) { + LOGGER.info("Segment: {} of table: {} is not loaded, checking disk", segmentName, _tableNameWithType); + localMetadata = recoverSegmentQuietly(segmentName); + if (!isNewSegment(zkMetadata, localMetadata)) { + LOGGER.info("Segment: {} of table {} has crc: {} same as before, loading", segmentName, _tableNameWithType, + localMetadata.getCrc()); + if (loadSegmentQuietly(segmentName, indexLoadingConfig)) { Review comment: basically, method with `quietly` suffix takes care of exceptions and may return a boolean to indicate whether the operation is done or not. -- 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: commits-unsubscr...@pinot.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org For additional commands, e-mail: commits-h...@pinot.apache.org