HBASE-23802 Remove unnecessary Configuration instantiation in LossyAccounting (#1127) (#1140)

Signed-off-by: stack <stack@apache.org>
This commit is contained in:
Nick Dimiduk 2020-02-07 08:56:11 -08:00 committed by GitHub
parent 68912c6bac
commit 9e4aec56c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 75 additions and 64 deletions

View File

@ -1,24 +1,29 @@
/** /*
* Licensed to the Apache Software Foundation (ASF) under one or more contributor license * Licensed to the Apache Software Foundation (ASF) under one
* agreements. See the NOTICE file distributed with this work for additional information regarding * or more contributor license agreements. See the NOTICE file
* copyright ownership. The ASF licenses this file to you under the Apache License, Version 2.0 (the * distributed with this work for additional information
* "License"); you may not use this file except in compliance with the License. You may obtain a * regarding copyright ownership. The ASF licenses this file
* copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable * to you under the Apache License, Version 2.0 (the
* law or agreed to in writing, software distributed under the License is distributed on an "AS IS" * "License"); you may not use this file except in compliance
* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License * with the License. You may obtain a copy of the License at
* for the specific language governing permissions and limitations under the License. *
* 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.coprocessor; package org.apache.hadoop.hbase.coprocessor;
import com.google.common.collect.ImmutableMap;
import java.io.IOException; import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.HashSet; import java.util.HashSet;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import org.apache.hadoop.conf.Configuration;
import com.google.common.collect.ImmutableMap;
import org.apache.hadoop.hbase.Cell; import org.apache.hadoop.hbase.Cell;
import org.apache.hadoop.hbase.CoprocessorEnvironment; import org.apache.hadoop.hbase.CoprocessorEnvironment;
import org.apache.hadoop.hbase.TableName; import org.apache.hadoop.hbase.TableName;
@ -31,6 +36,7 @@ import org.apache.hadoop.hbase.client.Row;
import org.apache.hadoop.hbase.ipc.RpcServer; import org.apache.hadoop.hbase.ipc.RpcServer;
import org.apache.hadoop.hbase.metrics.MetricRegistry; import org.apache.hadoop.hbase.metrics.MetricRegistry;
import org.apache.hadoop.hbase.regionserver.wal.WALEdit; import org.apache.hadoop.hbase.regionserver.wal.WALEdit;
import org.apache.hadoop.hbase.util.Bytes;
import org.apache.hadoop.hbase.util.LossyCounting; import org.apache.hadoop.hbase.util.LossyCounting;
/** /**
@ -48,15 +54,18 @@ public class MetaTableMetrics extends BaseRegionObserver {
private MetricRegistry registry; private MetricRegistry registry;
private LossyCounting<String> clientMetricsLossyCounting, regionMetricsLossyCounting; private LossyCounting<String> clientMetricsLossyCounting, regionMetricsLossyCounting;
private boolean active = false; private boolean active = false;
private Set<String> metrics = new HashSet<String>(); private Set<String> metrics = new HashSet<>();
enum MetaTableOps { enum MetaTableOps {
GET, PUT, DELETE; GET, PUT, DELETE,
} }
private ImmutableMap<Class<?>, MetaTableOps> opsNameMap = private ImmutableMap<Class<? extends Row>, MetaTableOps> opsNameMap =
ImmutableMap.<Class<?>, MetaTableOps>builder().put(Put.class, MetaTableOps.PUT) ImmutableMap.<Class<? extends Row>, MetaTableOps>builder()
.put(Get.class, MetaTableOps.GET).put(Delete.class, MetaTableOps.DELETE).build(); .put(Put.class, MetaTableOps.PUT)
.put(Get.class, MetaTableOps.GET)
.put(Delete.class, MetaTableOps.DELETE)
.build();
@Override @Override
public void preGetOp(ObserverContext<RegionCoprocessorEnvironment> e, Get get, List<Cell> results) public void preGetOp(ObserverContext<RegionCoprocessorEnvironment> e, Get get, List<Cell> results)
@ -93,13 +102,12 @@ public class MetaTableMetrics extends BaseRegionObserver {
* @param op such as get, put or delete. * @param op such as get, put or delete.
*/ */
private String getTableNameFromOp(Row op) { private String getTableNameFromOp(Row op) {
String tableName = null; final String tableRowKey = Bytes.toString(op.getRow());
String tableRowKey = new String(((Row) op).getRow(), StandardCharsets.UTF_8);
if (tableRowKey.isEmpty()) { if (tableRowKey.isEmpty()) {
return null; return null;
} }
tableName = tableRowKey.split(",").length > 0 ? tableRowKey.split(",")[0] : null; final String[] splits = tableRowKey.split(",");
return tableName; return splits.length > 0 ? splits[0] : null;
} }
/** /**
@ -108,13 +116,12 @@ public class MetaTableMetrics extends BaseRegionObserver {
* @param op such as get, put or delete. * @param op such as get, put or delete.
*/ */
private String getRegionIdFromOp(Row op) { private String getRegionIdFromOp(Row op) {
String regionId = null; final String tableRowKey = Bytes.toString(op.getRow());
String tableRowKey = new String(((Row) op).getRow(), StandardCharsets.UTF_8);
if (tableRowKey.isEmpty()) { if (tableRowKey.isEmpty()) {
return null; return null;
} }
regionId = tableRowKey.split(",").length > 2 ? tableRowKey.split(",")[2] : null; final String[] splits = tableRowKey.split(",");
return regionId; return splits.length > 2 ? splits[2] : null;
} }
private boolean isMetaTableOp(ObserverContext<RegionCoprocessorEnvironment> e) { private boolean isMetaTableOp(ObserverContext<RegionCoprocessorEnvironment> e) {
@ -258,8 +265,9 @@ public class MetaTableMetrics extends BaseRegionObserver {
metrics.remove(key); metrics.remove(key);
} }
}; };
clientMetricsLossyCounting = new LossyCounting<String>("clientMetaMetrics",listener); final Configuration conf = regionCoprocessorEnv.getConfiguration();
regionMetricsLossyCounting = new LossyCounting<String>("regionMetaMetrics",listener); clientMetricsLossyCounting = new LossyCounting<>("clientMetaMetrics", conf, listener);
regionMetricsLossyCounting = new LossyCounting<>("regionMetaMetrics", conf, listener);
// only be active mode when this region holds meta table. // only be active mode when this region holds meta table.
active = true; active = true;
} }

View File

@ -1,4 +1,4 @@
/** /*
* Licensed to the Apache Software Foundation (ASF) under one * Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file * or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information * distributed with this work for additional information
@ -20,7 +20,6 @@ package org.apache.hadoop.hbase.regionserver;
import com.google.common.annotations.VisibleForTesting; import com.google.common.annotations.VisibleForTesting;
import java.io.IOException; import java.io.IOException;
import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.CompatibilitySingletonFactory; import org.apache.hadoop.hbase.CompatibilitySingletonFactory;
import org.apache.hadoop.hbase.classification.InterfaceAudience; import org.apache.hadoop.hbase.classification.InterfaceAudience;
@ -41,7 +40,7 @@ public class MetricsUserAggregateImpl implements MetricsUserAggregate{
public MetricsUserAggregateImpl(Configuration conf) { public MetricsUserAggregateImpl(Configuration conf) {
source = CompatibilitySingletonFactory.getInstance(MetricsRegionServerSourceFactory.class) source = CompatibilitySingletonFactory.getInstance(MetricsRegionServerSourceFactory.class)
.getUserAggregate(); .getUserAggregate();
userMetricLossyCounting = new LossyCounting<MetricsUserSource>("userMetrics", userMetricLossyCounting = new LossyCounting<>("userMetrics", conf,
new LossyCounting.LossyCountingListener<MetricsUserSource>() { new LossyCounting.LossyCountingListener<MetricsUserSource>() {
@Override @Override
public void sweep(MetricsUserSource key) { public void sweep(MetricsUserSource key) {

View File

@ -1,4 +1,4 @@
/** /*
* *
* Licensed to the Apache Software Foundation (ASF) under one * Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file * or more contributor license agreements. See the NOTICE file
@ -21,7 +21,6 @@ package org.apache.hadoop.hbase.util;
import com.google.common.annotations.VisibleForTesting; import com.google.common.annotations.VisibleForTesting;
import com.google.common.util.concurrent.ThreadFactoryBuilder; import com.google.common.util.concurrent.ThreadFactoryBuilder;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
@ -29,8 +28,7 @@ import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors; import java.util.concurrent.Executors;
import java.util.concurrent.Future; import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicReference; import java.util.concurrent.atomic.AtomicReference;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.HConstants; import org.apache.hadoop.hbase.HConstants;
import org.apache.hadoop.hbase.classification.InterfaceAudience; import org.apache.hadoop.hbase.classification.InterfaceAudience;
import org.slf4j.Logger; import org.slf4j.Logger;
@ -46,7 +44,6 @@ import org.slf4j.LoggerFactory;
* Based on paper: * Based on paper:
* http://www.vldb.org/conf/2002/S10P03.pdf * http://www.vldb.org/conf/2002/S10P03.pdf
*/ */
@InterfaceAudience.Private @InterfaceAudience.Private
public class LossyCounting<T> { public class LossyCounting<T> {
private static final Logger LOG = LoggerFactory.getLogger(LossyCounting.class); private static final Logger LOG = LoggerFactory.getLogger(LossyCounting.class);
@ -56,14 +53,18 @@ public class LossyCounting<T> {
private Map<T, Integer> data; private Map<T, Integer> data;
private long totalDataCount; private long totalDataCount;
private final String name; private final String name;
private LossyCountingListener listener; private LossyCountingListener<T> listener;
private static AtomicReference<Future> fut = new AtomicReference<>(null); private static AtomicReference<Future<?>> fut = new AtomicReference<>(null);
public interface LossyCountingListener<T> { public interface LossyCountingListener<T> {
void sweep(T key); void sweep(T key);
} }
public LossyCounting(double errorRate, String name, LossyCountingListener listener) { LossyCounting(String name, double errorRate) {
this(name, errorRate, null);
}
public LossyCounting(String name, double errorRate, LossyCountingListener<T> listener) {
if (errorRate < 0.0 || errorRate > 1.0) { if (errorRate < 0.0 || errorRate > 1.0) {
throw new IllegalArgumentException(" Lossy Counting error rate should be within range [0,1]"); throw new IllegalArgumentException(" Lossy Counting error rate should be within range [0,1]");
} }
@ -78,9 +79,12 @@ public class LossyCounting<T> {
new ThreadFactoryBuilder().setDaemon(true).setNameFormat("lossy-count-%d").build()); new ThreadFactoryBuilder().setDaemon(true).setNameFormat("lossy-count-%d").build());
} }
public LossyCounting(String name, LossyCountingListener listener) { LossyCounting(String name, Configuration conf) {
this(HBaseConfiguration.create().getDouble(HConstants.DEFAULT_LOSSY_COUNTING_ERROR_RATE, 0.02), this(name, conf, null);
name, listener); }
public LossyCounting(String name, Configuration conf, LossyCountingListener<T> listener) {
this(name, conf.getDouble(HConstants.DEFAULT_LOSSY_COUNTING_ERROR_RATE, 0.02), listener);
} }
private void addByOne(T key) { private void addByOne(T key) {
@ -101,7 +105,7 @@ public class LossyCounting<T> {
if(totalDataCount % bucketSize == 0) { if(totalDataCount % bucketSize == 0) {
//sweep the entries at bucket boundaries //sweep the entries at bucket boundaries
//run Sweep //run Sweep
Future future = fut.get(); Future<?> future = fut.get();
if (future != null && !future.isDone()){ if (future != null && !future.isDone()){
return; return;
} }
@ -166,7 +170,7 @@ public class LossyCounting<T> {
} }
} }
@VisibleForTesting public Future getSweepFuture() { @VisibleForTesting public Future<?> getSweepFuture() {
return fut.get(); return fut.get();
} }
} }

View File

@ -1,4 +1,4 @@
/** /*
* *
* Licensed to the Apache Software Foundation (ASF) under one * Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file * or more contributor license agreements. See the NOTICE file
@ -20,7 +20,9 @@
package org.apache.hadoop.hbase.util; package org.apache.hadoop.hbase.util;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.testclassification.MiscTests; import org.apache.hadoop.hbase.testclassification.MiscTests;
import org.apache.hadoop.hbase.testclassification.SmallTests; import org.apache.hadoop.hbase.testclassification.SmallTests;
import org.junit.Test; import org.junit.Test;
@ -29,31 +31,33 @@ import org.junit.experimental.categories.Category;
@Category({MiscTests.class, SmallTests.class}) @Category({MiscTests.class, SmallTests.class})
public class TestLossyCounting { public class TestLossyCounting {
private final Configuration conf = HBaseConfiguration.create();
@Test @Test
public void testBucketSize() { public void testBucketSize() {
LossyCounting lossyCounting = new LossyCounting(0.01, "testBucketSize", null); LossyCounting<?> lossyCounting = new LossyCounting<>("testBucketSize", 0.01);
assertEquals(100L, lossyCounting.getBucketSize()); assertEquals(100L, lossyCounting.getBucketSize());
LossyCounting lossyCounting2 = new LossyCounting("testBucketSize2", null); LossyCounting<?> lossyCounting2 = new LossyCounting<>("testBucketSize2", conf);
assertEquals(50L, lossyCounting2.getBucketSize()); assertEquals(50L, lossyCounting2.getBucketSize());
} }
@Test @Test
public void testAddByOne() { public void testAddByOne() {
LossyCounting lossyCounting = new LossyCounting(0.01, "testAddByOne", null); LossyCounting<String> lossyCounting = new LossyCounting<>("testAddByOne", 0.01);
for(int i = 0; i < 100; i++){ for (int i = 0; i < 100; i++) {
String key = "" + i; String key = "" + i;
lossyCounting.add(key); lossyCounting.add(key);
} }
assertEquals(100L, lossyCounting.getDataSize()); assertEquals(100L, lossyCounting.getDataSize());
for(int i = 0; i < 100; i++){ for (int i = 0; i < 100; i++) {
String key = "" + i; String key = "" + i;
assertEquals(true, lossyCounting.contains(key)); assertTrue(lossyCounting.contains(key));
} }
} }
@Test @Test
public void testSweep1() { public void testSweep1() throws Exception {
LossyCounting lossyCounting = new LossyCounting(0.01, "testSweep1", null); LossyCounting<String> lossyCounting = new LossyCounting<>("testSweep1", 0.01);
for(int i = 0; i < 400; i++){ for(int i = 0; i < 400; i++){
String key = "" + i; String key = "" + i;
lossyCounting.add(key); lossyCounting.add(key);
@ -66,22 +70,19 @@ public class TestLossyCounting {
assertEquals(lossyCounting.getBucketSize() - 1, lossyCounting.getDataSize()); assertEquals(lossyCounting.getBucketSize() - 1, lossyCounting.getDataSize());
} }
private void waitForSweep(LossyCounting<Object> lossyCounting) { private void waitForSweep(LossyCounting<?> lossyCounting) throws InterruptedException {
//wait for sweep thread to complete //wait for sweep thread to complete
int retry = 0; int retry = 0;
while (!lossyCounting.getSweepFuture().isDone() && retry < 10) { while (!lossyCounting.getSweepFuture().isDone() && retry < 10) {
try {
Thread.sleep(100); Thread.sleep(100);
} catch (InterruptedException e) {
}
retry++; retry++;
} }
} }
@Test @Test
public void testSweep2() { public void testSweep2() throws Exception {
LossyCounting lossyCounting = new LossyCounting(0.1, "testSweep2", null); LossyCounting<String> lossyCounting = new LossyCounting<>("testSweep2", 0.1);
for(int i = 0; i < 10; i++){ for (int i = 0; i < 10; i++) {
String key = "" + i; String key = "" + i;
lossyCounting.add(key); lossyCounting.add(key);
} }
@ -94,5 +95,4 @@ public class TestLossyCounting {
waitForSweep(lossyCounting); waitForSweep(lossyCounting);
assertEquals(1L, lossyCounting.getDataSize()); assertEquals(1L, lossyCounting.getDataSize());
} }
} }