You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@accumulo.apache.org by ec...@apache.org on 2013/10/24 00:08:01 UTC
git commit: ACCUMULO-1451 add functional test for configurable
compactions
Updated Branches:
refs/heads/master f3e3869b9 -> e00e39591
ACCUMULO-1451 add functional test for configurable compactions
Project: http://git-wip-us.apache.org/repos/asf/accumulo/repo
Commit: http://git-wip-us.apache.org/repos/asf/accumulo/commit/e00e3959
Tree: http://git-wip-us.apache.org/repos/asf/accumulo/tree/e00e3959
Diff: http://git-wip-us.apache.org/repos/asf/accumulo/diff/e00e3959
Branch: refs/heads/master
Commit: e00e395919198cc5f6478d553f5da3319defaa45
Parents: f3e3869
Author: Eric Newton <er...@gmail.com>
Authored: Wed Oct 23 18:08:22 2013 -0400
Committer: Eric Newton <er...@gmail.com>
Committed: Wed Oct 23 18:08:22 2013 -0400
----------------------------------------------------------------------
.../functional/ConfigurableCompactionIT.java | 129 +++++++++++++++++++
1 file changed, 129 insertions(+)
----------------------------------------------------------------------
http://git-wip-us.apache.org/repos/asf/accumulo/blob/e00e3959/test/src/test/java/org/apache/accumulo/test/functional/ConfigurableCompactionIT.java
----------------------------------------------------------------------
diff --git a/test/src/test/java/org/apache/accumulo/test/functional/ConfigurableCompactionIT.java b/test/src/test/java/org/apache/accumulo/test/functional/ConfigurableCompactionIT.java
new file mode 100644
index 0000000..8dd4bee
--- /dev/null
+++ b/test/src/test/java/org/apache/accumulo/test/functional/ConfigurableCompactionIT.java
@@ -0,0 +1,129 @@
+/*
+ * 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.accumulo.test.functional;
+
+import static org.junit.Assert.assertTrue;
+
+import java.io.IOException;
+import java.util.Collections;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Random;
+
+import org.apache.accumulo.core.client.BatchWriter;
+import org.apache.accumulo.core.client.BatchWriterConfig;
+import org.apache.accumulo.core.client.Connector;
+import org.apache.accumulo.core.client.Scanner;
+import org.apache.accumulo.core.conf.Property;
+import org.apache.accumulo.core.data.Key;
+import org.apache.accumulo.core.data.Mutation;
+import org.apache.accumulo.core.data.Value;
+import org.apache.accumulo.core.metadata.MetadataTable;
+import org.apache.accumulo.core.metadata.schema.MetadataSchema;
+import org.apache.accumulo.core.security.Authorizations;
+import org.apache.accumulo.fate.util.UtilWaitThread;
+import org.apache.accumulo.minicluster.MiniAccumuloConfig;
+import org.apache.accumulo.server.tabletserver.compaction.CompactionPlan;
+import org.apache.accumulo.server.tabletserver.compaction.CompactionStrategy;
+import org.apache.accumulo.server.tabletserver.compaction.MajorCompactionRequest;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class ConfigurableCompactionIT extends ConfigurableMacIT {
+
+ @Override
+ public void configure(MiniAccumuloConfig cfg) {
+ cfg.setSiteConfig(Collections.singletonMap(Property.TSERV_MAJC_DELAY.getKey(), "1s"));
+ }
+
+ public static class SimpleCompactionStrategy extends CompactionStrategy {
+
+ @Override
+ public void init(Map<String,String> options) {
+ String countString = options.get("count");
+ if (countString != null)
+ count = Integer.parseInt(countString);
+ }
+
+ int count = 3;
+
+ @Override
+ public boolean shouldCompact(MajorCompactionRequest request) throws IOException {
+ return request.getFiles().size() == count;
+
+ }
+
+ @Override
+ public CompactionPlan getCompactionPlan(MajorCompactionRequest request) throws IOException {
+ CompactionPlan result = new CompactionPlan();
+ result.inputFiles.addAll(request.getFiles().keySet());
+ return result;
+ }
+
+ }
+
+ @Test
+ public void test() throws Exception {
+ final Connector c = getConnector();
+ final String tableName = getTableNames(1)[0];
+ c.tableOperations().create(tableName);
+ c.tableOperations().setProperty(tableName, Property.TABLE_COMPACTION_STRATEGY.getKey(), SimpleCompactionStrategy.class.getName());
+ runTest(c, tableName, 3);
+ c.tableOperations().setProperty(tableName, Property.TABLE_COMPACTION_STRATEGY_PREFIX.getKey() + "count", "" + 5);
+ runTest(c, tableName, 5);
+ }
+
+ final static Random r = new Random();
+
+ private void makeFile(Connector conn, String tablename) throws Exception {
+ BatchWriter bw = conn.createBatchWriter(tablename, new BatchWriterConfig());
+ byte[] empty = {};
+ byte[] row = new byte[10];
+ r.nextBytes(row);
+ Mutation m = new Mutation(row, 0, 10);
+ m.put(empty, empty, empty);
+ bw.addMutation(m);
+ bw.flush();
+ bw.close();
+ conn.tableOperations().flush(tablename, null, null, true);
+ }
+
+ private void runTest(final Connector c, final String tableName, final int n) throws Exception {
+ for (int i = countFiles(c, tableName); i < n - 1; i++)
+ makeFile(c, tableName);
+ Assert.assertEquals(n - 1, countFiles(c, tableName));
+ makeFile(c, tableName);
+ for (int i = 0; i < 10; i++) {
+ int count = countFiles(c, tableName);
+ assertTrue(count == 1 || count == n);
+ if (count == 1)
+ break;
+ UtilWaitThread.sleep(1000);
+ }
+ }
+
+ private int countFiles(Connector c, String tableName) throws Exception {
+ Scanner s = c.createScanner(MetadataTable.NAME, Authorizations.EMPTY);
+ s.fetchColumnFamily(MetadataSchema.TabletsSection.DataFileColumnFamily.NAME);
+ int count = 0;
+ for (@SuppressWarnings("unused") Entry<Key,Value> entry : s) {
+ count++;
+ }
+ return count;
+ }
+
+}