Java tutorial
/* * 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.merkle.cli; import java.io.ByteArrayOutputStream; import java.io.DataOutputStream; import java.io.File; import java.io.FileNotFoundException; import java.nio.charset.StandardCharsets; import java.security.MessageDigest; import java.security.NoSuchAlgorithmException; import java.util.ArrayList; import java.util.Collection; import java.util.Collections; import java.util.List; import java.util.Map.Entry; import java.util.TreeSet; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import org.apache.accumulo.core.cli.BatchWriterOpts; import org.apache.accumulo.core.cli.ClientOnRequiredTable; import org.apache.accumulo.core.client.AccumuloException; import org.apache.accumulo.core.client.AccumuloSecurityException; 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.IteratorSetting; import org.apache.accumulo.core.client.MutationsRejectedException; import org.apache.accumulo.core.client.Scanner; import org.apache.accumulo.core.client.TableNotFoundException; import org.apache.accumulo.core.data.Key; import org.apache.accumulo.core.data.Mutation; import org.apache.accumulo.core.data.Range; import org.apache.accumulo.core.data.Value; import org.apache.accumulo.core.security.Authorizations; import org.apache.accumulo.test.merkle.RangeSerialization; import org.apache.accumulo.test.merkle.skvi.DigestIterator; import org.apache.commons.codec.binary.Hex; import org.apache.hadoop.io.Text; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import com.beust.jcommander.Parameter; import com.google.common.collect.Iterables; import com.google.common.collect.Lists; /** * */ public class GenerateHashes { private static final Logger log = LoggerFactory.getLogger(GenerateHashes.class); public static class GenerateHashesOpts extends ClientOnRequiredTable { @Parameter(names = { "-hash", "--hash" }, required = true, description = "type of hash to use") private String hashName; @Parameter(names = { "-o", "--output" }, required = true, description = "output table name, expected to exist and be writable") private String outputTableName; @Parameter(names = { "-nt", "--numThreads" }, required = false, description = "number of concurrent threads calculating digests") private int numThreads = 4; @Parameter(names = { "-iter", "--iterator" }, required = false, description = "Should we push down logic with an iterator") private boolean iteratorPushdown = false; @Parameter(names = { "-s", "--splits" }, required = false, description = "File of splits to use for merkle tree") private String splitsFile = null; public String getHashName() { return hashName; } public void setHashName(String hashName) { this.hashName = hashName; } public String getOutputTableName() { return outputTableName; } public void setOutputTableName(String outputTableName) { this.outputTableName = outputTableName; } public int getNumThreads() { return numThreads; } public void setNumThreads(int numThreads) { this.numThreads = numThreads; } public boolean isIteratorPushdown() { return iteratorPushdown; } public void setIteratorPushdown(boolean iteratorPushdown) { this.iteratorPushdown = iteratorPushdown; } public String getSplitsFile() { return splitsFile; } public void setSplitsFile(String splitsFile) { this.splitsFile = splitsFile; } } public Collection<Range> getRanges(Connector conn, String tableName, String splitsFile) throws TableNotFoundException, AccumuloSecurityException, AccumuloException, FileNotFoundException { if (null == splitsFile) { Collection<Text> endRows = conn.tableOperations().listSplits(tableName); return endRowsToRanges(endRows); } else { ArrayList<Text> splits = new ArrayList<Text>(); String line; java.util.Scanner file = new java.util.Scanner(new File(splitsFile), StandardCharsets.UTF_8.name()); List<Text> result = Lists.newArrayList(); try { while (file.hasNextLine()) { line = file.nextLine(); if (!line.isEmpty()) { result.add(new Text(line)); } } } finally { file.close(); } Collections.sort(splits); return endRowsToRanges(splits); } } public void run(GenerateHashesOpts opts) throws TableNotFoundException, AccumuloSecurityException, AccumuloException, NoSuchAlgorithmException, FileNotFoundException { Collection<Range> ranges = getRanges(opts.getConnector(), opts.getTableName(), opts.getSplitsFile()); run(opts.getConnector(), opts.getTableName(), opts.getOutputTableName(), opts.getHashName(), opts.getNumThreads(), opts.isIteratorPushdown(), ranges); } public void run(final Connector conn, final String inputTableName, final String outputTableName, final String digestName, int numThreads, final boolean iteratorPushdown, final Collection<Range> ranges) throws TableNotFoundException, AccumuloSecurityException, AccumuloException, NoSuchAlgorithmException { if (!conn.tableOperations().exists(outputTableName)) { throw new IllegalArgumentException(outputTableName + " does not exist, please create it"); } ExecutorService svc = Executors.newFixedThreadPool(numThreads); final BatchWriter bw = conn.createBatchWriter(outputTableName, new BatchWriterConfig()); try { for (final Range range : ranges) { final MessageDigest digest = getDigestAlgorithm(digestName); svc.execute(new Runnable() { @Override public void run() { Scanner s; try { s = conn.createScanner(inputTableName, Authorizations.EMPTY); } catch (Exception e) { log.error("Could not get scanner for " + inputTableName, e); throw new RuntimeException(e); } s.setRange(range); Value v = null; Mutation m = null; if (iteratorPushdown) { IteratorSetting cfg = new IteratorSetting(50, DigestIterator.class); cfg.addOption(DigestIterator.HASH_NAME_KEY, digestName); s.addScanIterator(cfg); // The scanner should only ever return us one Key-Value, otherwise this approach won't work Entry<Key, Value> entry = Iterables.getOnlyElement(s); v = entry.getValue(); m = RangeSerialization.toMutation(range, v); } else { ByteArrayOutputStream baos = new ByteArrayOutputStream(); for (Entry<Key, Value> entry : s) { DataOutputStream out = new DataOutputStream(baos); try { entry.getKey().write(out); entry.getValue().write(out); } catch (Exception e) { log.error("Error writing {}", entry, e); throw new RuntimeException(e); } digest.update(baos.toByteArray()); baos.reset(); } v = new Value(digest.digest()); m = RangeSerialization.toMutation(range, v); } // Log some progress log.info("{} computed digest for {} of {}", Thread.currentThread().getName(), range, Hex.encodeHexString(v.get())); try { bw.addMutation(m); } catch (MutationsRejectedException e) { log.error("Could not write mutation", e); throw new RuntimeException(e); } } }); } svc.shutdown(); // Wait indefinitely for the scans to complete while (!svc.isTerminated()) { try { Thread.sleep(1000); } catch (InterruptedException e) { log.error("Interrupted while waiting for executor service to gracefully complete. Exiting now"); svc.shutdownNow(); return; } } } finally { // We can only safely close this when we're exiting or we've completely all tasks bw.close(); } } public TreeSet<Range> endRowsToRanges(Collection<Text> endRows) { ArrayList<Text> sortedEndRows = new ArrayList<Text>(endRows); Collections.sort(sortedEndRows); Text prevEndRow = null; TreeSet<Range> ranges = new TreeSet<>(); for (Text endRow : sortedEndRows) { if (null == prevEndRow) { ranges.add(new Range(null, false, endRow, true)); } else { ranges.add(new Range(prevEndRow, false, endRow, true)); } prevEndRow = endRow; } ranges.add(new Range(prevEndRow, false, null, false)); return ranges; } protected MessageDigest getDigestAlgorithm(String digestName) throws NoSuchAlgorithmException { return MessageDigest.getInstance(digestName); } public static void main(String[] args) throws Exception { GenerateHashesOpts opts = new GenerateHashesOpts(); BatchWriterOpts bwOpts = new BatchWriterOpts(); opts.parseArgs(GenerateHashes.class.getName(), args, bwOpts); if (opts.isIteratorPushdown() && null != opts.getSplitsFile()) { throw new IllegalArgumentException( "Cannot use iterator pushdown with anything other than table split points"); } GenerateHashes generate = new GenerateHashes(); generate.run(opts); } }