hbase-server/src/test/java/org/apache/hadoop/hbase/client/SimpleScanResultConsumer.java

   1 /**
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements.  See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership.  The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License.  You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18 package org.apache.hadoop.hbase.client;
  19
  20 import org.apache.hbase.thirdparty.com.google.common.base.Throwables;
  21
  22 import java.util.ArrayList;
  23 import java.util.List;
  24
  25 import org.apache.hadoop.hbase.client.metrics.ScanMetrics;
  26
  27 final class SimpleScanResultConsumer implements ScanResultConsumer {
  28
  29   private ScanMetrics scanMetrics;
  30
  31   private final List<Result> results = new ArrayList<>();
  32
  33   private Throwable error;
  34
  35   private boolean finished = false;
  36
  37   @Override
  38   public void onScanMetricsCreated(ScanMetrics scanMetrics) {
  39     this.scanMetrics = scanMetrics;
  40   }
  41
  42   @Override
  43   public synchronized boolean onNext(Result result) {
  44     results.add(result);
  45     return true;
  46   }
  47
  48   @Override
  49   public synchronized void onError(Throwable error) {
  50     this.error = error;
  51     finished = true;
  52     notifyAll();
  53   }
  54
  55   @Override
  56   public synchronized void onComplete() {
  57     finished = true;
  58     notifyAll();
  59   }
  60
  61   public synchronized List<Result> getAll() throws Exception {
  62     while (!finished) {
  63       wait();
  64     }
  65     if (error != null) {
  66       Throwables.propagateIfPossible(error, Exception.class);
  67       throw new Exception(error);
  68     }
  69     return results;
  70   }
  71
  72   public ScanMetrics getScanMetrics() {
  73     return scanMetrics;
  74   }
  75 }