| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 org.apache.hadoop.hbase.client; |
| |
| import static org.apache.hadoop.hbase.client.ConnectionUtils.filterCells; |
| |
| import java.io.IOException; |
| import java.util.Arrays; |
| |
| import org.apache.hadoop.hbase.Cell; |
| import org.apache.hadoop.hbase.CellUtil; |
| import org.apache.yetus.audience.InterfaceAudience; |
| |
| /** |
| * A ScanResultCache that may return partial result. |
| * <p> |
| * As we can only scan from the starting of a row when error, so here we also implement the logic |
| * that skips the cells that have already been returned. |
| */ |
| @InterfaceAudience.Private |
| class AllowPartialScanResultCache implements ScanResultCache { |
| |
| // used to filter out the cells that already returned to user as we always start from the |
| // beginning of a row when retry. |
| private Cell lastCell; |
| |
| private boolean lastResultPartial; |
| |
| private int numberOfCompleteRows; |
| |
| private void recordLastResult(Result result) { |
| lastCell = result.rawCells()[result.rawCells().length - 1]; |
| lastResultPartial = result.mayHaveMoreCellsInRow(); |
| } |
| |
| @Override |
| public Result[] addAndGet(Result[] results, boolean isHeartbeatMessage) throws IOException { |
| if (results.length == 0) { |
| if (!isHeartbeatMessage && lastResultPartial) { |
| // An empty non heartbeat result indicate that there must be a row change. So if the |
| // lastResultPartial is true then we need to increase numberOfCompleteRows. |
| numberOfCompleteRows++; |
| } |
| return EMPTY_RESULT_ARRAY; |
| } |
| int i; |
| for (i = 0; i < results.length; i++) { |
| Result r = filterCells(results[i], lastCell); |
| if (r != null) { |
| results[i] = r; |
| break; |
| } |
| } |
| if (i == results.length) { |
| return EMPTY_RESULT_ARRAY; |
| } |
| if (lastResultPartial && !CellUtil.matchingRows(lastCell, results[0].getRow())) { |
| // there is a row change, so increase numberOfCompleteRows |
| numberOfCompleteRows++; |
| } |
| recordLastResult(results[results.length - 1]); |
| if (i > 0) { |
| results = Arrays.copyOfRange(results, i, results.length); |
| } |
| for (Result result : results) { |
| if (!result.mayHaveMoreCellsInRow()) { |
| numberOfCompleteRows++; |
| } |
| } |
| return results; |
| } |
| |
| @Override |
| public void clear() { |
| // we do not cache anything |
| } |
| |
| @Override |
| public int numberOfCompleteRows() { |
| return numberOfCompleteRows; |
| } |
| } |