You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@accumulo.apache.org by ct...@apache.org on 2016/03/03 23:00:01 UTC
[36/61] [abbrv] [partial] accumulo git commit: ACCUMULO-722 put trunk
in my sandbox
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/InputFormatBase.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/InputFormatBase.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/InputFormatBase.java
new file mode 100644
index 0000000..704e7e6
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/InputFormatBase.java
@@ -0,0 +1,1242 @@
+/*
+ * 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.core.client.mapreduce;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.DataInput;
+import java.io.DataInputStream;
+import java.io.DataOutput;
+import java.io.DataOutputStream;
+import java.io.IOException;
+import java.io.UnsupportedEncodingException;
+import java.math.BigInteger;
+import java.net.InetAddress;
+import java.net.URLDecoder;
+import java.net.URLEncoder;
+import java.nio.ByteBuffer;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.Set;
+import java.util.StringTokenizer;
+
+import org.apache.accumulo.core.Constants;
+import org.apache.accumulo.core.client.AccumuloException;
+import org.apache.accumulo.core.client.AccumuloSecurityException;
+import org.apache.accumulo.core.client.ClientSideIteratorScanner;
+import org.apache.accumulo.core.client.Connector;
+import org.apache.accumulo.core.client.Instance;
+import org.apache.accumulo.core.client.IsolatedScanner;
+import org.apache.accumulo.core.client.IteratorSetting;
+import org.apache.accumulo.core.client.RowIterator;
+import org.apache.accumulo.core.client.Scanner;
+import org.apache.accumulo.core.client.TableDeletedException;
+import org.apache.accumulo.core.client.TableNotFoundException;
+import org.apache.accumulo.core.client.TableOfflineException;
+import org.apache.accumulo.core.client.ZooKeeperInstance;
+import org.apache.accumulo.core.client.impl.OfflineScanner;
+import org.apache.accumulo.core.client.impl.Tables;
+import org.apache.accumulo.core.client.impl.TabletLocator;
+import org.apache.accumulo.core.client.mock.MockInstance;
+import org.apache.accumulo.core.client.mock.MockTabletLocator;
+import org.apache.accumulo.core.data.ByteSequence;
+import org.apache.accumulo.core.data.Key;
+import org.apache.accumulo.core.data.KeyExtent;
+import org.apache.accumulo.core.data.PartialKey;
+import org.apache.accumulo.core.data.Range;
+import org.apache.accumulo.core.data.Value;
+import org.apache.accumulo.core.iterators.SortedKeyValueIterator;
+import org.apache.accumulo.core.iterators.user.VersioningIterator;
+import org.apache.accumulo.core.master.state.tables.TableState;
+import org.apache.accumulo.core.security.Authorizations;
+import org.apache.accumulo.core.security.TablePermission;
+import org.apache.accumulo.core.security.thrift.AuthInfo;
+import org.apache.accumulo.core.util.ArgumentChecker;
+import org.apache.accumulo.core.util.Pair;
+import org.apache.accumulo.core.util.TextUtil;
+import org.apache.accumulo.core.util.UtilWaitThread;
+import org.apache.commons.codec.binary.Base64;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.filecache.DistributedCache;
+import org.apache.hadoop.fs.FSDataInputStream;
+import org.apache.hadoop.fs.FSDataOutputStream;
+import org.apache.hadoop.fs.FileSystem;
+import org.apache.hadoop.fs.Path;
+import org.apache.hadoop.fs.permission.FsAction;
+import org.apache.hadoop.fs.permission.FsPermission;
+import org.apache.hadoop.io.Text;
+import org.apache.hadoop.io.Writable;
+import org.apache.hadoop.mapreduce.InputFormat;
+import org.apache.hadoop.mapreduce.InputSplit;
+import org.apache.hadoop.mapreduce.JobContext;
+import org.apache.hadoop.mapreduce.RecordReader;
+import org.apache.hadoop.mapreduce.TaskAttemptContext;
+import org.apache.log4j.Level;
+import org.apache.log4j.Logger;
+
+/**
+ * This class allows MapReduce jobs to use Accumulo as the source of data. This input format provides keys and values of type K and V to the Map() and Reduce()
+ * functions.
+ *
+ * Subclasses must implement the following method: public RecordReader<K, V> createRecordReader(InputSplit split, TaskAttemptContext context) throws
+ * IOException, InterruptedException
+ *
+ * This class includes a static class that can be used to create a RecordReader: protected abstract static class RecordReaderBase<K,V> extends RecordReader<K,V>
+ *
+ * Subclasses of RecordReaderBase must implement the following method: public boolean nextKeyValue() throws IOException, InterruptedException This method should
+ * set the following variables: K currentK V currentV Key currentKey (used for progress reporting) int numKeysRead (used for progress reporting)
+ *
+ * See AccumuloInputFormat for an example implementation.
+ *
+ * Other static methods are optional
+ */
+
+public abstract class InputFormatBase<K,V> extends InputFormat<K,V> {
+ protected static final Logger log = Logger.getLogger(InputFormatBase.class);
+
+ private static final String PREFIX = AccumuloInputFormat.class.getSimpleName();
+ private static final String INPUT_INFO_HAS_BEEN_SET = PREFIX + ".configured";
+ private static final String INSTANCE_HAS_BEEN_SET = PREFIX + ".instanceConfigured";
+ private static final String USERNAME = PREFIX + ".username";
+ private static final String PASSWORD_PATH = PREFIX + ".password";
+ private static final String TABLE_NAME = PREFIX + ".tablename";
+ private static final String AUTHORIZATIONS = PREFIX + ".authorizations";
+
+ private static final String INSTANCE_NAME = PREFIX + ".instanceName";
+ private static final String ZOOKEEPERS = PREFIX + ".zooKeepers";
+ private static final String MOCK = ".useMockInstance";
+
+ private static final String RANGES = PREFIX + ".ranges";
+ private static final String AUTO_ADJUST_RANGES = PREFIX + ".ranges.autoAdjust";
+
+ private static final String COLUMNS = PREFIX + ".columns";
+ private static final String LOGLEVEL = PREFIX + ".loglevel";
+
+ private static final String ISOLATED = PREFIX + ".isolated";
+
+ private static final String LOCAL_ITERATORS = PREFIX + ".localiters";
+
+ // Used to specify the maximum # of versions of an Accumulo cell value to return
+ private static final String MAX_VERSIONS = PREFIX + ".maxVersions";
+
+ // Used for specifying the iterators to be applied
+ private static final String ITERATORS = PREFIX + ".iterators";
+ private static final String ITERATORS_OPTIONS = PREFIX + ".iterators.options";
+ private static final String ITERATORS_DELIM = ",";
+
+ private static final String READ_OFFLINE = PREFIX + ".read.offline";
+
+ /**
+ * Enable or disable use of the {@link IsolatedScanner} in this configuration object. By default it is not enabled.
+ *
+ * @param conf
+ * The Hadoop configuration object
+ * @param enable
+ * if true, enable usage of the IsolatedScanner. Otherwise, disable.
+ */
+ public static void setIsolated(Configuration conf, boolean enable) {
+ conf.setBoolean(ISOLATED, enable);
+ }
+
+ /**
+ * Enable or disable use of the {@link ClientSideIteratorScanner} in this configuration object. By default it is not enabled.
+ *
+ * @param conf
+ * The Hadoop configuration object
+ * @param enable
+ * if true, enable usage of the ClientSideInteratorScanner. Otherwise, disable.
+ */
+ public static void setLocalIterators(Configuration conf, boolean enable) {
+ conf.setBoolean(LOCAL_ITERATORS, enable);
+ }
+
+ /**
+ * Initialize the user, table, and authorization information for the configuration object that will be used with an Accumulo InputFormat.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param user
+ * a valid accumulo user
+ * @param passwd
+ * the user's password
+ * @param table
+ * the table to read
+ * @param auths
+ * the authorizations used to restrict data read
+ */
+ public static void setInputInfo(Configuration conf, String user, byte[] passwd, String table, Authorizations auths) {
+ if (conf.getBoolean(INPUT_INFO_HAS_BEEN_SET, false))
+ throw new IllegalStateException("Input info can only be set once per job");
+ conf.setBoolean(INPUT_INFO_HAS_BEEN_SET, true);
+
+ ArgumentChecker.notNull(user, passwd, table);
+ conf.set(USERNAME, user);
+ conf.set(TABLE_NAME, table);
+ if (auths != null && !auths.isEmpty())
+ conf.set(AUTHORIZATIONS, auths.serialize());
+
+ try {
+ FileSystem fs = FileSystem.get(conf);
+ Path file = new Path(fs.getWorkingDirectory(), conf.get("mapred.job.name") + System.currentTimeMillis() + ".pw");
+ conf.set(PASSWORD_PATH, file.toString());
+ FSDataOutputStream fos = fs.create(file, false);
+ fs.setPermission(file, new FsPermission(FsAction.ALL, FsAction.NONE, FsAction.NONE));
+ fs.deleteOnExit(file);
+
+ byte[] encodedPw = Base64.encodeBase64(passwd);
+ fos.writeInt(encodedPw.length);
+ fos.write(encodedPw);
+ fos.close();
+
+ DistributedCache.addCacheFile(file.toUri(), conf);
+ } catch (IOException ioe) {
+ throw new RuntimeException(ioe);
+ }
+
+ }
+
+ /**
+ * Configure a {@link ZooKeeperInstance} for this configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param instanceName
+ * the accumulo instance name
+ * @param zooKeepers
+ * a comma-separated list of zookeeper servers
+ */
+ public static void setZooKeeperInstance(Configuration conf, String instanceName, String zooKeepers) {
+ if (conf.getBoolean(INSTANCE_HAS_BEEN_SET, false))
+ throw new IllegalStateException("Instance info can only be set once per job");
+ conf.setBoolean(INSTANCE_HAS_BEEN_SET, true);
+
+ ArgumentChecker.notNull(instanceName, zooKeepers);
+ conf.set(INSTANCE_NAME, instanceName);
+ conf.set(ZOOKEEPERS, zooKeepers);
+ }
+
+ /**
+ * Configure a {@link MockInstance} for this configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param instanceName
+ * the accumulo instance name
+ */
+ public static void setMockInstance(Configuration conf, String instanceName) {
+ conf.setBoolean(INSTANCE_HAS_BEEN_SET, true);
+ conf.setBoolean(MOCK, true);
+ conf.set(INSTANCE_NAME, instanceName);
+ }
+
+ /**
+ * Set the ranges to map over for this configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param ranges
+ * the ranges that will be mapped over
+ */
+ public static void setRanges(Configuration conf, Collection<Range> ranges) {
+ ArgumentChecker.notNull(ranges);
+ ArrayList<String> rangeStrings = new ArrayList<String>(ranges.size());
+ try {
+ for (Range r : ranges) {
+ ByteArrayOutputStream baos = new ByteArrayOutputStream();
+ r.write(new DataOutputStream(baos));
+ rangeStrings.add(new String(Base64.encodeBase64(baos.toByteArray())));
+ }
+ } catch (IOException ex) {
+ throw new IllegalArgumentException("Unable to encode ranges to Base64", ex);
+ }
+ conf.setStrings(RANGES, rangeStrings.toArray(new String[0]));
+ }
+
+ /**
+ * Disables the adjustment of ranges for this configuration object. By default, overlapping ranges will be merged and ranges will be fit to existing tablet
+ * boundaries. Disabling this adjustment will cause there to be exactly one mapper per range set using {@link #setRanges(Configuration, Collection)}.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ */
+ public static void disableAutoAdjustRanges(Configuration conf) {
+ conf.setBoolean(AUTO_ADJUST_RANGES, false);
+ }
+
+ /**
+ * Sets the max # of values that may be returned for an individual Accumulo cell. By default, applied before all other Accumulo iterators (highest priority)
+ * leveraged in the scan by the record reader. To adjust priority use setIterator() & setIteratorOptions() w/ the VersioningIterator type explicitly.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param maxVersions
+ * the max number of versions per accumulo cell
+ * @throws IOException
+ * if maxVersions is < 1
+ */
+ public static void setMaxVersions(Configuration conf, int maxVersions) throws IOException {
+ if (maxVersions < 1)
+ throw new IOException("Invalid maxVersions: " + maxVersions + ". Must be >= 1");
+ conf.setInt(MAX_VERSIONS, maxVersions);
+ }
+
+ /**
+ * <p>
+ * Enable reading offline tables. This will make the map reduce job directly read the tables files. If the table is not offline, then the job will fail. If
+ * the table comes online during the map reduce job, its likely that the job will fail.
+ *
+ * <p>
+ * To use this option, the map reduce user will need access to read the accumulo directory in HDFS.
+ *
+ * <p>
+ * Reading the offline table will create the scan time iterator stack in the map process. So any iterators that are configured for the table will need to be
+ * on the mappers classpath. The accumulo-site.xml may need to be on the mappers classpath if HDFS or the accumlo directory in HDFS are non-standard.
+ *
+ * <p>
+ * One way to use this feature is to clone a table, take the clone offline, and use the clone as the input table for a map reduce job. If you plan to map
+ * reduce over the data many times, it may be better to the compact the table, clone it, take it offline, and use the clone for all map reduce jobs. The
+ * reason to do this is that compaction will reduce each tablet in the table to one file, and its faster to read from one file.
+ *
+ * <p>
+ * There are two possible advantages to reading a tables file directly out of HDFS. First, you may see better read performance. Second, it will support
+ * speculative execution better. When reading an online table speculative execution can put more load on an already slow tablet server.
+ *
+ * @param conf
+ * the job
+ * @param scanOff
+ * pass true to read offline tables
+ */
+
+ public static void setScanOffline(Configuration conf, boolean scanOff) {
+ conf.setBoolean(READ_OFFLINE, scanOff);
+ }
+
+ /**
+ * Restricts the columns that will be mapped over for this configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param columnFamilyColumnQualifierPairs
+ * A pair of {@link Text} objects corresponding to column family and column qualifier. If the column qualifier is null, the entire column family is
+ * selected. An empty set is the default and is equivalent to scanning the all columns.
+ */
+ public static void fetchColumns(Configuration conf, Collection<Pair<Text,Text>> columnFamilyColumnQualifierPairs) {
+ ArgumentChecker.notNull(columnFamilyColumnQualifierPairs);
+ ArrayList<String> columnStrings = new ArrayList<String>(columnFamilyColumnQualifierPairs.size());
+ for (Pair<Text,Text> column : columnFamilyColumnQualifierPairs) {
+ if (column.getFirst() == null)
+ throw new IllegalArgumentException("Column family can not be null");
+
+ String col = new String(Base64.encodeBase64(TextUtil.getBytes(column.getFirst())));
+ if (column.getSecond() != null)
+ col += ":" + new String(Base64.encodeBase64(TextUtil.getBytes(column.getSecond())));
+ columnStrings.add(col);
+ }
+ conf.setStrings(COLUMNS, columnStrings.toArray(new String[0]));
+ }
+
+ /**
+ * Sets the log level for this configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param level
+ * the logging level
+ */
+ public static void setLogLevel(Configuration conf, Level level) {
+ ArgumentChecker.notNull(level);
+ log.setLevel(level);
+ conf.setInt(LOGLEVEL, level.toInt());
+ }
+
+ /**
+ * Encode an iterator on the input for this configuration object.
+ *
+ * @param conf
+ * The Hadoop configuration in which to save the iterator configuration
+ * @param cfg
+ * The configuration of the iterator
+ */
+ public static void addIterator(Configuration conf, IteratorSetting cfg) {
+ // First check to see if anything has been set already
+ String iterators = conf.get(ITERATORS);
+
+ // No iterators specified yet, create a new string
+ if (iterators == null || iterators.isEmpty()) {
+ iterators = new AccumuloIterator(cfg.getPriority(), cfg.getIteratorClass(), cfg.getName()).toString();
+ } else {
+ // append the next iterator & reset
+ iterators = iterators.concat(ITERATORS_DELIM + new AccumuloIterator(cfg.getPriority(), cfg.getIteratorClass(), cfg.getName()).toString());
+ }
+ // Store the iterators w/ the job
+ conf.set(ITERATORS, iterators);
+ for (Entry<String,String> entry : cfg.getOptions().entrySet()) {
+ if (entry.getValue() == null)
+ continue;
+
+ String iteratorOptions = conf.get(ITERATORS_OPTIONS);
+
+ // No options specified yet, create a new string
+ if (iteratorOptions == null || iteratorOptions.isEmpty()) {
+ iteratorOptions = new AccumuloIteratorOption(cfg.getName(), entry.getKey(), entry.getValue()).toString();
+ } else {
+ // append the next option & reset
+ iteratorOptions = iteratorOptions.concat(ITERATORS_DELIM + new AccumuloIteratorOption(cfg.getName(), entry.getKey(), entry.getValue()));
+ }
+
+ // Store the options w/ the job
+ conf.set(ITERATORS_OPTIONS, iteratorOptions);
+ }
+ }
+
+ /**
+ * Determines whether a configuration has isolation enabled.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return true if isolation is enabled, false otherwise
+ * @see #setIsolated(Configuration, boolean)
+ */
+ protected static boolean isIsolated(Configuration conf) {
+ return conf.getBoolean(ISOLATED, false);
+ }
+
+ /**
+ * Determines whether a configuration uses local iterators.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return true if uses local iterators, false otherwise
+ * @see #setLocalIterators(Configuration, boolean)
+ */
+ protected static boolean usesLocalIterators(Configuration conf) {
+ return conf.getBoolean(LOCAL_ITERATORS, false);
+ }
+
+ /**
+ * Gets the user name from the configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the user name
+ * @see #setInputInfo(Configuration, String, byte[], String, Authorizations)
+ */
+ protected static String getUsername(Configuration conf) {
+ return conf.get(USERNAME);
+ }
+
+ /**
+ * Gets the password from the configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the BASE64-encoded password
+ * @throws IOException
+ * @see #setInputInfo(Configuration, String, byte[], String, Authorizations)
+ */
+ protected static byte[] getPassword(Configuration conf) throws IOException {
+ FileSystem fs = FileSystem.get(conf);
+ Path file = new Path(conf.get(PASSWORD_PATH));
+
+ FSDataInputStream fdis = fs.open(file);
+ int length = fdis.readInt();
+ byte[] encodedPassword = new byte[length];
+ fdis.read(encodedPassword);
+ fdis.close();
+
+ return Base64.decodeBase64(encodedPassword);
+ }
+
+ /**
+ * Gets the table name from the configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the table name
+ * @see #setInputInfo(Configuration, String, byte[], String, Authorizations)
+ */
+ protected static String getTablename(Configuration conf) {
+ return conf.get(TABLE_NAME);
+ }
+
+ /**
+ * Gets the authorizations to set for the scans from the configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the accumulo scan authorizations
+ * @see #setInputInfo(Configuration, String, byte[], String, Authorizations)
+ */
+ protected static Authorizations getAuthorizations(Configuration conf) {
+ String authString = conf.get(AUTHORIZATIONS);
+ return authString == null ? Constants.NO_AUTHS : new Authorizations(authString.split(","));
+ }
+
+ /**
+ * Initializes an Accumulo {@link Instance} based on the configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return an accumulo instance
+ * @see #setZooKeeperInstance(Configuration, String, String)
+ * @see #setMockInstance(Configuration, String)
+ */
+ protected static Instance getInstance(Configuration conf) {
+ if (conf.getBoolean(MOCK, false))
+ return new MockInstance(conf.get(INSTANCE_NAME));
+ return new ZooKeeperInstance(conf.get(INSTANCE_NAME), conf.get(ZOOKEEPERS));
+ }
+
+ /**
+ * Initializes an Accumulo {@link TabletLocator} based on the configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return an accumulo tablet locator
+ * @throws TableNotFoundException
+ * if the table name set on the configuration doesn't exist
+ * @throws IOException
+ * if the input format is unable to read the password file from the FileSystem
+ */
+ protected static TabletLocator getTabletLocator(Configuration conf) throws TableNotFoundException, IOException {
+ if (conf.getBoolean(MOCK, false))
+ return new MockTabletLocator();
+ Instance instance = getInstance(conf);
+ String username = getUsername(conf);
+ byte[] password = getPassword(conf);
+ String tableName = getTablename(conf);
+ return TabletLocator.getInstance(instance, new AuthInfo(username, ByteBuffer.wrap(password), instance.getInstanceID()),
+ new Text(Tables.getTableId(instance, tableName)));
+ }
+
+ /**
+ * Gets the ranges to scan over from a configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the ranges
+ * @throws IOException
+ * if the ranges have been encoded improperly
+ * @see #setRanges(Configuration, Collection)
+ */
+ protected static List<Range> getRanges(Configuration conf) throws IOException {
+ ArrayList<Range> ranges = new ArrayList<Range>();
+ for (String rangeString : conf.getStringCollection(RANGES)) {
+ ByteArrayInputStream bais = new ByteArrayInputStream(Base64.decodeBase64(rangeString.getBytes()));
+ Range range = new Range();
+ range.readFields(new DataInputStream(bais));
+ ranges.add(range);
+ }
+ return ranges;
+ }
+
+ /**
+ * Gets the columns to be mapped over from this configuration object.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return a set of columns
+ * @see #fetchColumns(Configuration, Collection)
+ */
+ protected static Set<Pair<Text,Text>> getFetchedColumns(Configuration conf) {
+ Set<Pair<Text,Text>> columns = new HashSet<Pair<Text,Text>>();
+ for (String col : conf.getStringCollection(COLUMNS)) {
+ int idx = col.indexOf(":");
+ Text cf = new Text(idx < 0 ? Base64.decodeBase64(col.getBytes()) : Base64.decodeBase64(col.substring(0, idx).getBytes()));
+ Text cq = idx < 0 ? null : new Text(Base64.decodeBase64(col.substring(idx + 1).getBytes()));
+ columns.add(new Pair<Text,Text>(cf, cq));
+ }
+ return columns;
+ }
+
+ /**
+ * Determines whether a configuration has auto-adjust ranges enabled.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return true if auto-adjust is enabled, false otherwise
+ * @see #disableAutoAdjustRanges(Configuration)
+ */
+ protected static boolean getAutoAdjustRanges(Configuration conf) {
+ return conf.getBoolean(AUTO_ADJUST_RANGES, true);
+ }
+
+ /**
+ * Gets the log level from this configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the log level
+ * @see #setLogLevel(Configuration, Level)
+ */
+ protected static Level getLogLevel(Configuration conf) {
+ return Level.toLevel(conf.getInt(LOGLEVEL, Level.INFO.toInt()));
+ }
+
+ // InputFormat doesn't have the equivalent of OutputFormat's checkOutputSpecs(JobContext job)
+ /**
+ * Check whether a configuration is fully configured to be used with an Accumulo {@link org.apache.hadoop.mapreduce.InputFormat}.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @throws IOException
+ * if the configuration is improperly configured
+ */
+ protected static void validateOptions(Configuration conf) throws IOException {
+ if (!conf.getBoolean(INPUT_INFO_HAS_BEEN_SET, false))
+ throw new IOException("Input info has not been set.");
+ if (!conf.getBoolean(INSTANCE_HAS_BEEN_SET, false))
+ throw new IOException("Instance info has not been set.");
+ // validate that we can connect as configured
+ try {
+ Connector c = getInstance(conf).getConnector(getUsername(conf), getPassword(conf));
+ if (!c.securityOperations().authenticateUser(getUsername(conf), getPassword(conf)))
+ throw new IOException("Unable to authenticate user");
+ if (!c.securityOperations().hasTablePermission(getUsername(conf), getTablename(conf), TablePermission.READ))
+ throw new IOException("Unable to access table");
+
+ if (!usesLocalIterators(conf)) {
+ // validate that any scan-time iterators can be loaded by the the tablet servers
+ for (AccumuloIterator iter : getIterators(conf)) {
+ if (!c.instanceOperations().testClassLoad(iter.getIteratorClass(), SortedKeyValueIterator.class.getName()))
+ throw new AccumuloException("Servers are unable to load " + iter.getIteratorClass() + " as a " + SortedKeyValueIterator.class.getName());
+ }
+ }
+
+ } catch (AccumuloException e) {
+ throw new IOException(e);
+ } catch (AccumuloSecurityException e) {
+ throw new IOException(e);
+ }
+ }
+
+ /**
+ * Gets the maxVersions to use for the {@link VersioningIterator} from this configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return the max versions, -1 if not configured
+ * @see #setMaxVersions(Configuration, int)
+ */
+ protected static int getMaxVersions(Configuration conf) {
+ return conf.getInt(MAX_VERSIONS, -1);
+ }
+
+ protected static boolean isOfflineScan(Configuration conf) {
+ return conf.getBoolean(READ_OFFLINE, false);
+ }
+
+ // Return a list of the iterator settings (for iterators to apply to a scanner)
+
+ /**
+ * Gets a list of the iterator settings (for iterators to apply to a scanner) from this configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return a list of iterators
+ * @see #addIterator(Configuration, IteratorSetting)
+ */
+ protected static List<AccumuloIterator> getIterators(Configuration conf) {
+
+ String iterators = conf.get(ITERATORS);
+
+ // If no iterators are present, return an empty list
+ if (iterators == null || iterators.isEmpty())
+ return new ArrayList<AccumuloIterator>();
+
+ // Compose the set of iterators encoded in the job configuration
+ StringTokenizer tokens = new StringTokenizer(conf.get(ITERATORS), ITERATORS_DELIM);
+ List<AccumuloIterator> list = new ArrayList<AccumuloIterator>();
+ while (tokens.hasMoreTokens()) {
+ String itstring = tokens.nextToken();
+ list.add(new AccumuloIterator(itstring));
+ }
+ return list;
+ }
+
+ /**
+ * Gets a list of the iterator options specified on this configuration.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @return a list of iterator options
+ * @see #addIterator(Configuration, IteratorSetting)
+ */
+ protected static List<AccumuloIteratorOption> getIteratorOptions(Configuration conf) {
+ String iteratorOptions = conf.get(ITERATORS_OPTIONS);
+
+ // If no options are present, return an empty list
+ if (iteratorOptions == null || iteratorOptions.isEmpty())
+ return new ArrayList<AccumuloIteratorOption>();
+
+ // Compose the set of options encoded in the job configuration
+ StringTokenizer tokens = new StringTokenizer(conf.get(ITERATORS_OPTIONS), ITERATORS_DELIM);
+ List<AccumuloIteratorOption> list = new ArrayList<AccumuloIteratorOption>();
+ while (tokens.hasMoreTokens()) {
+ String optionString = tokens.nextToken();
+ list.add(new AccumuloIteratorOption(optionString));
+ }
+ return list;
+ }
+
+ protected abstract static class RecordReaderBase<K,V> extends RecordReader<K,V> {
+ protected long numKeysRead;
+ protected Iterator<Entry<Key,Value>> scannerIterator;
+ protected RangeInputSplit split;
+
+ /**
+ * Apply the configured iterators from the configuration to the scanner.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param scanner
+ * the scanner to configure
+ * @throws AccumuloException
+ */
+ protected void setupIterators(Configuration conf, Scanner scanner) throws AccumuloException {
+ List<AccumuloIterator> iterators = getIterators(conf);
+ List<AccumuloIteratorOption> options = getIteratorOptions(conf);
+
+ Map<String,IteratorSetting> scanIterators = new HashMap<String,IteratorSetting>();
+ for (AccumuloIterator iterator : iterators) {
+ scanIterators.put(iterator.getIteratorName(), new IteratorSetting(iterator.getPriority(), iterator.getIteratorName(), iterator.getIteratorClass()));
+ }
+ for (AccumuloIteratorOption option : options) {
+ scanIterators.get(option.iteratorName).addOption(option.getKey(), option.getValue());
+ }
+ for (AccumuloIterator iterator : iterators) {
+ scanner.addScanIterator(scanIterators.get(iterator.getIteratorName()));
+ }
+ }
+
+ /**
+ * If maxVersions has been set, configure a {@link VersioningIterator} at priority 0 for this scanner.
+ *
+ * @param conf
+ * the Hadoop configuration object
+ * @param scanner
+ * the scanner to configure
+ */
+ protected void setupMaxVersions(Configuration conf, Scanner scanner) {
+ int maxVersions = getMaxVersions(conf);
+ // Check to make sure its a legit value
+ if (maxVersions >= 1) {
+ IteratorSetting vers = new IteratorSetting(0, "vers", VersioningIterator.class);
+ VersioningIterator.setMaxVersions(vers, maxVersions);
+ scanner.addScanIterator(vers);
+ }
+ }
+
+ /**
+ * Initialize a scanner over the given input split using this task attempt configuration.
+ */
+ public void initialize(InputSplit inSplit, TaskAttemptContext attempt) throws IOException {
+ initialize(inSplit, attempt.getConfiguration());
+ }
+
+ public void initialize(InputSplit inSplit, Configuration conf) throws IOException {
+ Scanner scanner;
+ split = (RangeInputSplit) inSplit;
+ log.debug("Initializing input split: " + split.range);
+ Instance instance = getInstance(conf);
+ String user = getUsername(conf);
+ byte[] password = getPassword(conf);
+ Authorizations authorizations = getAuthorizations(conf);
+
+ try {
+ log.debug("Creating connector with user: " + user);
+ Connector conn = instance.getConnector(user, password);
+ log.debug("Creating scanner for table: " + getTablename(conf));
+ log.debug("Authorizations are: " + authorizations);
+ if (isOfflineScan(conf)) {
+ scanner = new OfflineScanner(instance, new AuthInfo(user, ByteBuffer.wrap(password), instance.getInstanceID()), Tables.getTableId(instance,
+ getTablename(conf)), authorizations);
+ } else {
+ scanner = conn.createScanner(getTablename(conf), authorizations);
+ }
+ if (isIsolated(conf)) {
+ log.info("Creating isolated scanner");
+ scanner = new IsolatedScanner(scanner);
+ }
+ if (usesLocalIterators(conf)) {
+ log.info("Using local iterators");
+ scanner = new ClientSideIteratorScanner(scanner);
+ }
+ setupMaxVersions(conf, scanner);
+ setupIterators(conf, scanner);
+ } catch (Exception e) {
+ throw new IOException(e);
+ }
+
+ // setup a scanner within the bounds of this split
+ for (Pair<Text,Text> c : getFetchedColumns(conf)) {
+ if (c.getSecond() != null) {
+ log.debug("Fetching column " + c.getFirst() + ":" + c.getSecond());
+ scanner.fetchColumn(c.getFirst(), c.getSecond());
+ } else {
+ log.debug("Fetching column family " + c.getFirst());
+ scanner.fetchColumnFamily(c.getFirst());
+ }
+ }
+
+ scanner.setRange(split.range);
+
+ numKeysRead = 0;
+
+ // do this last after setting all scanner options
+ scannerIterator = scanner.iterator();
+ }
+
+ public void close() {}
+
+ public float getProgress() throws IOException {
+ if (numKeysRead > 0 && currentKey == null)
+ return 1.0f;
+ return split.getProgress(currentKey);
+ }
+
+ protected K currentK = null;
+ protected V currentV = null;
+ protected Key currentKey = null;
+ protected Value currentValue = null;
+
+ @Override
+ public K getCurrentKey() throws IOException, InterruptedException {
+ return currentK;
+ }
+
+ @Override
+ public V getCurrentValue() throws IOException, InterruptedException {
+ return currentV;
+ }
+ }
+
+ Map<String,Map<KeyExtent,List<Range>>> binOfflineTable(Configuration conf, String tableName, List<Range> ranges) throws TableNotFoundException,
+ AccumuloException, AccumuloSecurityException, IOException {
+
+ Map<String,Map<KeyExtent,List<Range>>> binnedRanges = new HashMap<String,Map<KeyExtent,List<Range>>>();
+
+ Instance instance = getInstance(conf);
+ Connector conn = instance.getConnector(getUsername(conf), getPassword(conf));
+ String tableId = Tables.getTableId(instance, tableName);
+
+ if (Tables.getTableState(instance, tableId) != TableState.OFFLINE) {
+ Tables.clearCache(instance);
+ if (Tables.getTableState(instance, tableId) != TableState.OFFLINE) {
+ throw new AccumuloException("Table is online " + tableName + "(" + tableId + ") cannot scan table in offline mode ");
+ }
+ }
+
+ for (Range range : ranges) {
+ Text startRow;
+
+ if (range.getStartKey() != null)
+ startRow = range.getStartKey().getRow();
+ else
+ startRow = new Text();
+
+ Range metadataRange = new Range(new KeyExtent(new Text(tableId), startRow, null).getMetadataEntry(), true, null, false);
+ Scanner scanner = conn.createScanner(Constants.METADATA_TABLE_NAME, Constants.NO_AUTHS);
+ Constants.METADATA_PREV_ROW_COLUMN.fetch(scanner);
+ scanner.fetchColumnFamily(Constants.METADATA_LAST_LOCATION_COLUMN_FAMILY);
+ scanner.fetchColumnFamily(Constants.METADATA_CURRENT_LOCATION_COLUMN_FAMILY);
+ scanner.fetchColumnFamily(Constants.METADATA_FUTURE_LOCATION_COLUMN_FAMILY);
+ scanner.setRange(metadataRange);
+
+ RowIterator rowIter = new RowIterator(scanner);
+
+ // TODO check that extents match prev extent
+
+ KeyExtent lastExtent = null;
+
+ while (rowIter.hasNext()) {
+ Iterator<Entry<Key,Value>> row = rowIter.next();
+ String last = "";
+ KeyExtent extent = null;
+ String location = null;
+
+ while (row.hasNext()) {
+ Entry<Key,Value> entry = row.next();
+ Key key = entry.getKey();
+
+ if (key.getColumnFamily().equals(Constants.METADATA_LAST_LOCATION_COLUMN_FAMILY)) {
+ last = entry.getValue().toString();
+ }
+
+ if (key.getColumnFamily().equals(Constants.METADATA_CURRENT_LOCATION_COLUMN_FAMILY)
+ || key.getColumnFamily().equals(Constants.METADATA_FUTURE_LOCATION_COLUMN_FAMILY)) {
+ location = entry.getValue().toString();
+ }
+
+ if (Constants.METADATA_PREV_ROW_COLUMN.hasColumns(key)) {
+ extent = new KeyExtent(key.getRow(), entry.getValue());
+ }
+
+ }
+
+ if (location != null)
+ return null;
+
+ if (!extent.getTableId().toString().equals(tableId)) {
+ throw new AccumuloException("Saw unexpected table Id " + tableId + " " + extent);
+ }
+
+ if (lastExtent != null && !extent.isPreviousExtent(lastExtent)) {
+ throw new AccumuloException(" " + lastExtent + " is not previous extent " + extent);
+ }
+
+ Map<KeyExtent,List<Range>> tabletRanges = binnedRanges.get(last);
+ if (tabletRanges == null) {
+ tabletRanges = new HashMap<KeyExtent,List<Range>>();
+ binnedRanges.put(last, tabletRanges);
+ }
+
+ List<Range> rangeList = tabletRanges.get(extent);
+ if (rangeList == null) {
+ rangeList = new ArrayList<Range>();
+ tabletRanges.put(extent, rangeList);
+ }
+
+ rangeList.add(range);
+
+ if (extent.getEndRow() == null || range.afterEndKey(new Key(extent.getEndRow()).followingKey(PartialKey.ROW))) {
+ break;
+ }
+
+ lastExtent = extent;
+ }
+
+ }
+
+ return binnedRanges;
+ }
+
+ /**
+ * Read the metadata table to get tablets and match up ranges to them.
+ */
+ public List<InputSplit> getSplits(JobContext job) throws IOException {
+ return getSplits(job.getConfiguration());
+ }
+
+ public List<InputSplit> getSplits(Configuration conf) throws IOException {
+ log.setLevel(getLogLevel(conf));
+ validateOptions(conf);
+
+ String tableName = getTablename(conf);
+ boolean autoAdjust = getAutoAdjustRanges(conf);
+ List<Range> ranges = autoAdjust ? Range.mergeOverlapping(getRanges(conf)) : getRanges(conf);
+
+ if (ranges.isEmpty()) {
+ ranges = new ArrayList<Range>(1);
+ ranges.add(new Range());
+ }
+
+ // get the metadata information for these ranges
+ Map<String,Map<KeyExtent,List<Range>>> binnedRanges = new HashMap<String,Map<KeyExtent,List<Range>>>();
+ TabletLocator tl;
+ try {
+ if (isOfflineScan(conf)) {
+ binnedRanges = binOfflineTable(conf, tableName, ranges);
+ while (binnedRanges == null) {
+ // Some tablets were still online, try again
+ UtilWaitThread.sleep(100 + (int) (Math.random() * 100)); // sleep randomly between 100 and 200 ms
+ binnedRanges = binOfflineTable(conf, tableName, ranges);
+ }
+ } else {
+ Instance instance = getInstance(conf);
+ String tableId = null;
+ tl = getTabletLocator(conf);
+ // its possible that the cache could contain complete, but old information about a tables tablets... so clear it
+ tl.invalidateCache();
+ while (!tl.binRanges(ranges, binnedRanges).isEmpty()) {
+ if (!(instance instanceof MockInstance)) {
+ if (tableId == null)
+ tableId = Tables.getTableId(instance, tableName);
+ if (!Tables.exists(instance, tableId))
+ throw new TableDeletedException(tableId);
+ if (Tables.getTableState(instance, tableId) == TableState.OFFLINE)
+ throw new TableOfflineException(instance, tableId);
+ }
+ binnedRanges.clear();
+ log.warn("Unable to locate bins for specified ranges. Retrying.");
+ UtilWaitThread.sleep(100 + (int) (Math.random() * 100)); // sleep randomly between 100 and 200 ms
+ tl.invalidateCache();
+ }
+ }
+ } catch (Exception e) {
+ throw new IOException(e);
+ }
+
+ ArrayList<InputSplit> splits = new ArrayList<InputSplit>(ranges.size());
+ HashMap<Range,ArrayList<String>> splitsToAdd = null;
+
+ if (!autoAdjust)
+ splitsToAdd = new HashMap<Range,ArrayList<String>>();
+
+ HashMap<String,String> hostNameCache = new HashMap<String,String>();
+
+ for (Entry<String,Map<KeyExtent,List<Range>>> tserverBin : binnedRanges.entrySet()) {
+ String ip = tserverBin.getKey().split(":", 2)[0];
+ String location = hostNameCache.get(ip);
+ if (location == null) {
+ InetAddress inetAddress = InetAddress.getByName(ip);
+ location = inetAddress.getHostName();
+ hostNameCache.put(ip, location);
+ }
+
+ for (Entry<KeyExtent,List<Range>> extentRanges : tserverBin.getValue().entrySet()) {
+ Range ke = extentRanges.getKey().toDataRange();
+ for (Range r : extentRanges.getValue()) {
+ if (autoAdjust) {
+ // divide ranges into smaller ranges, based on the tablets
+ splits.add(new RangeInputSplit(tableName, ke.clip(r), new String[] {location}));
+ } else {
+ // don't divide ranges
+ ArrayList<String> locations = splitsToAdd.get(r);
+ if (locations == null)
+ locations = new ArrayList<String>(1);
+ locations.add(location);
+ splitsToAdd.put(r, locations);
+ }
+ }
+ }
+ }
+
+ if (!autoAdjust)
+ for (Entry<Range,ArrayList<String>> entry : splitsToAdd.entrySet())
+ splits.add(new RangeInputSplit(tableName, entry.getKey(), entry.getValue().toArray(new String[0])));
+ return splits;
+ }
+
+ /**
+ * The Class RangeInputSplit. Encapsulates an Accumulo range for use in Map Reduce jobs.
+ */
+ public static class RangeInputSplit extends InputSplit implements Writable {
+ private Range range;
+ private String[] locations;
+
+ public RangeInputSplit() {
+ range = new Range();
+ locations = new String[0];
+ }
+
+ public RangeInputSplit(RangeInputSplit split) throws IOException {
+ this.setRange(split.getRange());
+ this.setLocations(split.getLocations());
+ }
+
+ public Range getRange() {
+ return range;
+ }
+
+ public void setRange(Range range) {
+ this.range = range;
+ }
+
+ private static byte[] extractBytes(ByteSequence seq, int numBytes) {
+ byte[] bytes = new byte[numBytes + 1];
+ bytes[0] = 0;
+ for (int i = 0; i < numBytes; i++) {
+ if (i >= seq.length())
+ bytes[i + 1] = 0;
+ else
+ bytes[i + 1] = seq.byteAt(i);
+ }
+ return bytes;
+ }
+
+ public static float getProgress(ByteSequence start, ByteSequence end, ByteSequence position) {
+ int maxDepth = Math.min(Math.max(end.length(), start.length()), position.length());
+ BigInteger startBI = new BigInteger(extractBytes(start, maxDepth));
+ BigInteger endBI = new BigInteger(extractBytes(end, maxDepth));
+ BigInteger positionBI = new BigInteger(extractBytes(position, maxDepth));
+ return (float) (positionBI.subtract(startBI).doubleValue() / endBI.subtract(startBI).doubleValue());
+ }
+
+ public float getProgress(Key currentKey) {
+ if (currentKey == null)
+ return 0f;
+ if (range.getStartKey() != null && range.getEndKey() != null) {
+ if (!range.getStartKey().equals(range.getEndKey(), PartialKey.ROW)) {
+ // just look at the row progress
+ return getProgress(range.getStartKey().getRowData(), range.getEndKey().getRowData(), currentKey.getRowData());
+ } else if (!range.getStartKey().equals(range.getEndKey(), PartialKey.ROW_COLFAM)) {
+ // just look at the column family progress
+ return getProgress(range.getStartKey().getColumnFamilyData(), range.getEndKey().getColumnFamilyData(), currentKey.getColumnFamilyData());
+ } else if (!range.getStartKey().equals(range.getEndKey(), PartialKey.ROW_COLFAM_COLQUAL)) {
+ // just look at the column qualifier progress
+ return getProgress(range.getStartKey().getColumnQualifierData(), range.getEndKey().getColumnQualifierData(), currentKey.getColumnQualifierData());
+ }
+ }
+ // if we can't figure it out, then claim no progress
+ return 0f;
+ }
+
+ RangeInputSplit(String table, Range range, String[] locations) {
+ this.range = range;
+ this.locations = locations;
+ }
+
+ /**
+ * This implementation of length is only an estimate, it does not provide exact values. Do not have your code rely on this return value.
+ */
+ public long getLength() throws IOException {
+ Text startRow = range.isInfiniteStartKey() ? new Text(new byte[] {Byte.MIN_VALUE}) : range.getStartKey().getRow();
+ Text stopRow = range.isInfiniteStopKey() ? new Text(new byte[] {Byte.MAX_VALUE}) : range.getEndKey().getRow();
+ int maxCommon = Math.min(7, Math.min(startRow.getLength(), stopRow.getLength()));
+ long diff = 0;
+
+ byte[] start = startRow.getBytes();
+ byte[] stop = stopRow.getBytes();
+ for (int i = 0; i < maxCommon; ++i) {
+ diff |= 0xff & (start[i] ^ stop[i]);
+ diff <<= Byte.SIZE;
+ }
+
+ if (startRow.getLength() != stopRow.getLength())
+ diff |= 0xff;
+
+ return diff + 1;
+ }
+
+ public String[] getLocations() throws IOException {
+ return locations;
+ }
+
+ public void setLocations(String[] locations) {
+ this.locations = locations;
+ }
+
+ public void readFields(DataInput in) throws IOException {
+ range.readFields(in);
+ int numLocs = in.readInt();
+ locations = new String[numLocs];
+ for (int i = 0; i < numLocs; ++i)
+ locations[i] = in.readUTF();
+ }
+
+ public void write(DataOutput out) throws IOException {
+ range.write(out);
+ out.writeInt(locations.length);
+ for (int i = 0; i < locations.length; ++i)
+ out.writeUTF(locations[i]);
+ }
+ }
+
+ /**
+ * The Class IteratorSetting. Encapsulates specifics for an Accumulo iterator's name & priority.
+ */
+ static class AccumuloIterator {
+
+ private static final String FIELD_SEP = ":";
+
+ private int priority;
+ private String iteratorClass;
+ private String iteratorName;
+
+ public AccumuloIterator(int priority, String iteratorClass, String iteratorName) {
+ this.priority = priority;
+ this.iteratorClass = iteratorClass;
+ this.iteratorName = iteratorName;
+ }
+
+ // Parses out a setting given an string supplied from an earlier toString() call
+ public AccumuloIterator(String iteratorSetting) {
+ // Parse the string to expand the iterator
+ StringTokenizer tokenizer = new StringTokenizer(iteratorSetting, FIELD_SEP);
+ priority = Integer.parseInt(tokenizer.nextToken());
+ iteratorClass = tokenizer.nextToken();
+ iteratorName = tokenizer.nextToken();
+ }
+
+ public int getPriority() {
+ return priority;
+ }
+
+ public String getIteratorClass() {
+ return iteratorClass;
+ }
+
+ public String getIteratorName() {
+ return iteratorName;
+ }
+
+ @Override
+ public String toString() {
+ return new String(priority + FIELD_SEP + iteratorClass + FIELD_SEP + iteratorName);
+ }
+
+ }
+
+ /**
+ * The Class AccumuloIteratorOption. Encapsulates specifics for an Accumulo iterator's optional configuration details - associated via the iteratorName.
+ */
+ static class AccumuloIteratorOption {
+ private static final String FIELD_SEP = ":";
+
+ private String iteratorName;
+ private String key;
+ private String value;
+
+ public AccumuloIteratorOption(String iteratorName, String key, String value) {
+ this.iteratorName = iteratorName;
+ this.key = key;
+ this.value = value;
+ }
+
+ // Parses out an option given a string supplied from an earlier toString() call
+ public AccumuloIteratorOption(String iteratorOption) {
+ StringTokenizer tokenizer = new StringTokenizer(iteratorOption, FIELD_SEP);
+ this.iteratorName = tokenizer.nextToken();
+ try {
+ this.key = URLDecoder.decode(tokenizer.nextToken(), "UTF-8");
+ this.value = URLDecoder.decode(tokenizer.nextToken(), "UTF-8");
+ } catch (UnsupportedEncodingException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ public String getIteratorName() {
+ return iteratorName;
+ }
+
+ public String getKey() {
+ return key;
+ }
+
+ public String getValue() {
+ return value;
+ }
+
+ @Override
+ public String toString() {
+ try {
+ return new String(iteratorName + FIELD_SEP + URLEncoder.encode(key, "UTF-8") + FIELD_SEP + URLEncoder.encode(value, "UTF-8"));
+ } catch (UnsupportedEncodingException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ }
+
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/KeyRangePartitioner.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/KeyRangePartitioner.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/KeyRangePartitioner.java
new file mode 100644
index 0000000..62c6911
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/KeyRangePartitioner.java
@@ -0,0 +1,60 @@
+/*
+ * 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.core.client.mapreduce.lib.partition;
+
+import org.apache.accumulo.core.data.Key;
+import org.apache.hadoop.conf.Configurable;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.io.Writable;
+import org.apache.hadoop.mapreduce.JobContext;
+import org.apache.hadoop.mapreduce.Partitioner;
+
+/**
+ * Hadoop partitioner that uses ranges based on row keys, and optionally sub-bins based on hashing.
+ */
+public class KeyRangePartitioner extends Partitioner<Key,Writable> implements Configurable {
+ private RangePartitioner rp = new RangePartitioner();
+
+ @Override
+ public int getPartition(Key key, Writable value, int numPartitions) {
+ return rp.getPartition(key.getRow(), value, numPartitions);
+ }
+
+ @Override
+ public Configuration getConf() {
+ return rp.getConf();
+ }
+
+ @Override
+ public void setConf(Configuration conf) {
+ rp.setConf(conf);
+ }
+
+ /**
+ * Sets the hdfs file name to use, containing a newline separated list of Base64 encoded split points that represent ranges for partitioning
+ */
+ public static void setSplitFile(JobContext job, String file) {
+ RangePartitioner.setSplitFile(job, file);
+ }
+
+ /**
+ * Sets the number of random sub-bins per range
+ */
+ public static void setNumSubBins(JobContext job, int num) {
+ RangePartitioner.setNumSubBins(job, num);
+ }
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/RangePartitioner.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/RangePartitioner.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/RangePartitioner.java
new file mode 100644
index 0000000..ae537f9
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mapreduce/lib/partition/RangePartitioner.java
@@ -0,0 +1,132 @@
+/*
+ * 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.core.client.mapreduce.lib.partition;
+
+import java.io.BufferedReader;
+import java.io.FileNotFoundException;
+import java.io.FileReader;
+import java.io.IOException;
+import java.net.URI;
+import java.util.Arrays;
+import java.util.Scanner;
+import java.util.TreeSet;
+
+import org.apache.commons.codec.binary.Base64;
+import org.apache.hadoop.conf.Configurable;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.filecache.DistributedCache;
+import org.apache.hadoop.fs.Path;
+import org.apache.hadoop.io.Text;
+import org.apache.hadoop.io.Writable;
+import org.apache.hadoop.mapreduce.JobContext;
+import org.apache.hadoop.mapreduce.Partitioner;
+
+/**
+ * Hadoop partitioner that uses ranges, and optionally sub-bins based on hashing.
+ */
+public class RangePartitioner extends Partitioner<Text,Writable> implements Configurable {
+ private static final String PREFIX = RangePartitioner.class.getName();
+ private static final String CUTFILE_KEY = PREFIX + ".cutFile";
+ private static final String NUM_SUBBINS = PREFIX + ".subBins";
+
+ private Configuration conf;
+
+ public int getPartition(Text key, Writable value, int numPartitions) {
+ try {
+ return findPartition(key, getCutPoints(), getNumSubBins());
+ } catch (IOException e) {
+ throw new RuntimeException(e);
+ }
+ }
+
+ int findPartition(Text key, Text[] array, int numSubBins) {
+ // find the bin for the range, and guarantee it is positive
+ int index = Arrays.binarySearch(array, key);
+ index = index < 0 ? (index + 1) * -1 : index;
+
+ // both conditions work with numSubBins == 1, but this check is to avoid
+ // hashing, when we don't need to, for speed
+ if (numSubBins < 2)
+ return index;
+ return (key.toString().hashCode() & Integer.MAX_VALUE) % numSubBins + index * numSubBins;
+ }
+
+ private int _numSubBins = 0;
+
+ private synchronized int getNumSubBins() {
+ if (_numSubBins < 1) {
+ // get number of sub-bins and guarantee it is positive
+ _numSubBins = Math.max(1, getConf().getInt(NUM_SUBBINS, 1));
+ }
+ return _numSubBins;
+ }
+
+ private Text cutPointArray[] = null;
+
+ private synchronized Text[] getCutPoints() throws IOException {
+ if (cutPointArray == null) {
+ String cutFileName = conf.get(CUTFILE_KEY);
+ Path[] cf = DistributedCache.getLocalCacheFiles(conf);
+
+ if (cf != null) {
+ for (Path path : cf) {
+ if (path.toUri().getPath().endsWith(cutFileName.substring(cutFileName.lastIndexOf('/')))) {
+ TreeSet<Text> cutPoints = new TreeSet<Text>();
+ Scanner in = new Scanner(new BufferedReader(new FileReader(path.toString())));
+ try {
+ while (in.hasNextLine())
+ cutPoints.add(new Text(Base64.decodeBase64(in.nextLine().getBytes())));
+ } finally {
+ in.close();
+ }
+ cutPointArray = cutPoints.toArray(new Text[cutPoints.size()]);
+ break;
+ }
+ }
+ }
+ if (cutPointArray == null)
+ throw new FileNotFoundException(cutFileName + " not found in distributed cache");
+ }
+ return cutPointArray;
+ }
+
+ @Override
+ public Configuration getConf() {
+ return conf;
+ }
+
+ @Override
+ public void setConf(Configuration conf) {
+ this.conf = conf;
+ }
+
+ /**
+ * Sets the hdfs file name to use, containing a newline separated list of Base64 encoded split points that represent ranges for partitioning
+ */
+ public static void setSplitFile(JobContext job, String file) {
+ URI uri = new Path(file).toUri();
+ DistributedCache.addCacheFile(uri, job.getConfiguration());
+ job.getConfiguration().set(CUTFILE_KEY, uri.getPath());
+ }
+
+ /**
+ * Sets the number of random sub-bins per range
+ */
+ public static void setNumSubBins(JobContext job, int num) {
+ job.getConfiguration().setInt(NUM_SUBBINS, num);
+ }
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/IteratorAdapter.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/IteratorAdapter.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/IteratorAdapter.java
new file mode 100644
index 0000000..840db41
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/IteratorAdapter.java
@@ -0,0 +1,57 @@
+/*
+ * 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.core.client.mock;
+
+import java.io.IOException;
+import java.util.Iterator;
+import java.util.Map.Entry;
+import java.util.NoSuchElementException;
+
+import org.apache.accumulo.core.data.Key;
+import org.apache.accumulo.core.data.KeyValue;
+import org.apache.accumulo.core.data.Value;
+import org.apache.accumulo.core.iterators.SortedKeyValueIterator;
+
+public class IteratorAdapter implements Iterator<Entry<Key,Value>> {
+
+ SortedKeyValueIterator<Key,Value> inner;
+
+ public IteratorAdapter(SortedKeyValueIterator<Key,Value> inner) {
+ this.inner = inner;
+ }
+
+ @Override
+ public boolean hasNext() {
+ return inner.hasTop();
+ }
+
+ @Override
+ public Entry<Key,Value> next() {
+ try {
+ Entry<Key,Value> result = new KeyValue(new Key(inner.getTopKey()), new Value(inner.getTopValue()).get());
+ inner.next();
+ return result;
+ } catch (IOException ex) {
+ throw new NoSuchElementException();
+ }
+ }
+
+ @Override
+ public void remove() {
+ throw new UnsupportedOperationException();
+ }
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockAccumulo.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockAccumulo.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockAccumulo.java
new file mode 100644
index 0000000..eca8c1e
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockAccumulo.java
@@ -0,0 +1,80 @@
+/*
+ * 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.core.client.mock;
+
+import java.util.Collection;
+import java.util.EnumSet;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.SortedSet;
+
+import org.apache.accumulo.core.Constants;
+import org.apache.accumulo.core.client.BatchScanner;
+import org.apache.accumulo.core.client.admin.TimeType;
+import org.apache.accumulo.core.data.Mutation;
+import org.apache.accumulo.core.security.Authorizations;
+import org.apache.accumulo.core.security.SystemPermission;
+import org.apache.accumulo.core.security.TablePermission;
+import org.apache.hadoop.io.Text;
+
+public class MockAccumulo {
+ final Map<String,MockTable> tables = new HashMap<String,MockTable>();
+ final Map<String,String> systemProperties = new HashMap<String,String>();
+ Map<String,MockUser> users = new HashMap<String,MockUser>();
+
+ {
+ MockUser root = new MockUser("root", new byte[] {}, Constants.NO_AUTHS);
+ root.permissions.add(SystemPermission.SYSTEM);
+ users.put(root.name, root);
+ createTable("root", Constants.METADATA_TABLE_NAME, true, TimeType.LOGICAL);
+ }
+
+ void setProperty(String key, String value) {
+ systemProperties.put(key, value);
+ }
+
+ String removeProperty(String key) {
+ return systemProperties.remove(key);
+ }
+
+ void createTable(String user, String table) {
+ createTable(user, table, true, TimeType.MILLIS);
+ }
+
+ public void addMutation(String table, Mutation m) {
+ MockTable t = tables.get(table);
+ t.addMutation(m);
+ }
+
+ public BatchScanner createBatchScanner(String tableName, Authorizations authorizations) {
+ return new MockBatchScanner(tables.get(tableName), authorizations);
+ }
+
+ public void createTable(String username, String tableName, boolean useVersions, TimeType timeType) {
+ MockTable t = new MockTable(useVersions, timeType);
+ t.userPermissions.put(username, EnumSet.allOf(TablePermission.class));
+ tables.put(tableName, t);
+ }
+
+ public void addSplits(String tableName, SortedSet<Text> partitionKeys) {
+ tables.get(tableName).addSplits(partitionKeys);
+ }
+
+ public Collection<Text> getSplits(String tableName) {
+ return tables.get(tableName).getSplits();
+ }
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchDeleter.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchDeleter.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchDeleter.java
new file mode 100644
index 0000000..67362a2
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchDeleter.java
@@ -0,0 +1,77 @@
+/*
+ * 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.core.client.mock;
+
+import java.util.Iterator;
+import java.util.Map.Entry;
+
+import org.apache.accumulo.core.client.BatchDeleter;
+import org.apache.accumulo.core.client.BatchWriter;
+import org.apache.accumulo.core.client.MutationsRejectedException;
+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.Value;
+import org.apache.accumulo.core.security.Authorizations;
+import org.apache.accumulo.core.security.ColumnVisibility;
+
+/**
+ * {@link BatchDeleter} for a {@link MockAccumulo} instance. Behaves similarly to a regular {@link BatchDeleter}, with a few exceptions:
+ * <ol>
+ * <li>There is no waiting for memory to fill before flushing</li>
+ * <li>Only one thread is used for writing</li>
+ * </ol>
+ *
+ * Otherwise, it behaves as expected.
+ */
+public class MockBatchDeleter extends MockBatchScanner implements BatchDeleter {
+
+ private final MockAccumulo acc;
+ private final String tableName;
+
+ /**
+ * Create a {@link BatchDeleter} for the specified instance on the specified table where the writer uses the specified {@link Authorizations}.
+ *
+ * @param acc
+ * @param tableName
+ * @param auths
+ */
+ public MockBatchDeleter(MockAccumulo acc, String tableName, Authorizations auths) {
+ super(acc.tables.get(tableName), auths);
+ this.acc = acc;
+ this.tableName = tableName;
+ }
+
+ @Override
+ public void delete() throws MutationsRejectedException, TableNotFoundException {
+
+ BatchWriter writer = new MockBatchWriter(acc, tableName);
+ try {
+ Iterator<Entry<Key,Value>> iter = super.iterator();
+ while (iter.hasNext()) {
+ Entry<Key,Value> next = iter.next();
+ Key k = next.getKey();
+ Mutation m = new Mutation(k.getRow());
+ m.putDelete(k.getColumnFamily(), k.getColumnQualifier(), new ColumnVisibility(k.getColumnVisibility()), k.getTimestamp());
+ writer.addMutation(m);
+ }
+ } finally {
+ writer.close();
+ }
+ }
+
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchScanner.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchScanner.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchScanner.java
new file mode 100644
index 0000000..807a711
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchScanner.java
@@ -0,0 +1,102 @@
+/*
+ * 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.core.client.mock;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map.Entry;
+
+import org.apache.accumulo.core.client.BatchScanner;
+import org.apache.accumulo.core.data.Key;
+import org.apache.accumulo.core.data.Range;
+import org.apache.accumulo.core.data.Value;
+import org.apache.accumulo.core.iterators.Filter;
+import org.apache.accumulo.core.iterators.SortedKeyValueIterator;
+import org.apache.accumulo.core.iterators.SortedMapIterator;
+import org.apache.accumulo.core.security.Authorizations;
+import org.apache.commons.collections.iterators.IteratorChain;
+
+public class MockBatchScanner extends MockScannerBase implements BatchScanner {
+
+ List<Range> ranges = null;
+
+ public MockBatchScanner(MockTable mockTable, Authorizations authorizations) {
+ super(mockTable, authorizations);
+ }
+
+ @Override
+ public void setRanges(Collection<Range> ranges) {
+ if (ranges == null || ranges.size() == 0) {
+ throw new IllegalArgumentException("ranges must be non null and contain at least 1 range");
+ }
+
+ this.ranges = new ArrayList<Range>(ranges);
+ }
+
+ static class RangesFilter extends Filter {
+ List<Range> ranges;
+
+ RangesFilter(SortedKeyValueIterator<Key,Value> iterator, List<Range> ranges) {
+ setSource(iterator);
+ this.ranges = ranges;
+ }
+
+ @Override
+ public boolean accept(Key k, Value v) {
+ for (Range r : ranges) {
+ if (r.contains(k))
+ return true;
+ }
+ return false;
+ }
+ }
+
+ @SuppressWarnings("unchecked")
+ @Override
+ public Iterator<Entry<Key,Value>> iterator() {
+ if (ranges == null) {
+ throw new IllegalStateException("ranges not set");
+ }
+
+ IteratorChain chain = new IteratorChain();
+ for (Range range : ranges) {
+ SortedKeyValueIterator<Key,Value> i = new SortedMapIterator(table.table);
+ try {
+ i = new RangesFilter(createFilter(i), ranges);
+ i.seek(range, createColumnBSS(fetchedColumns), !fetchedColumns.isEmpty());
+ chain.addIterator(new IteratorAdapter(i));
+ } catch (IOException e) {
+ throw new RuntimeException(e);
+ }
+ }
+ return chain;
+ }
+
+ @Override
+ public void close() {}
+
+ @Override
+ public void setTimeOut(int timeout) {}
+
+ @Override
+ public int getTimeOut() {
+ return Integer.MAX_VALUE;
+ }
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchWriter.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchWriter.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchWriter.java
new file mode 100644
index 0000000..b33ebcb
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockBatchWriter.java
@@ -0,0 +1,51 @@
+/*
+ * 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.core.client.mock;
+
+import org.apache.accumulo.core.client.BatchWriter;
+import org.apache.accumulo.core.client.MutationsRejectedException;
+import org.apache.accumulo.core.data.Mutation;
+
+public class MockBatchWriter implements BatchWriter {
+
+ final String tablename;
+ final MockAccumulo acu;
+
+ MockBatchWriter(MockAccumulo acu, String tablename) {
+ this.acu = acu;
+ this.tablename = tablename;
+ }
+
+ @Override
+ public void addMutation(Mutation m) throws MutationsRejectedException {
+ acu.addMutation(tablename, m);
+ }
+
+ @Override
+ public void addMutations(Iterable<Mutation> iterable) throws MutationsRejectedException {
+ for (Mutation m : iterable) {
+ acu.addMutation(tablename, m);
+ }
+ }
+
+ @Override
+ public void flush() throws MutationsRejectedException {}
+
+ @Override
+ public void close() throws MutationsRejectedException {}
+
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConfiguration.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConfiguration.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConfiguration.java
new file mode 100644
index 0000000..b3a0007
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConfiguration.java
@@ -0,0 +1,46 @@
+/*
+ * 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.core.client.mock;
+
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.apache.accumulo.core.conf.AccumuloConfiguration;
+import org.apache.accumulo.core.conf.Property;
+
+class MockConfiguration extends AccumuloConfiguration {
+ Map<String,String> map;
+
+ MockConfiguration(Map<String,String> settings) {
+ map = settings;
+ }
+
+ public void put(String k, String v) {
+ map.put(k, v);
+ }
+
+ @Override
+ public String get(Property property) {
+ return map.get(property.getKey());
+ }
+
+ @Override
+ public Iterator<Entry<String,String>> iterator() {
+ return map.entrySet().iterator();
+ }
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConnector.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConnector.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConnector.java
new file mode 100644
index 0000000..c6f3dcc
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockConnector.java
@@ -0,0 +1,108 @@
+/*
+ * 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.core.client.mock;
+
+import org.apache.accumulo.core.client.BatchDeleter;
+import org.apache.accumulo.core.client.BatchScanner;
+import org.apache.accumulo.core.client.BatchWriter;
+import org.apache.accumulo.core.client.Connector;
+import org.apache.accumulo.core.client.Instance;
+import org.apache.accumulo.core.client.MultiTableBatchWriter;
+import org.apache.accumulo.core.client.Scanner;
+import org.apache.accumulo.core.client.TableNotFoundException;
+import org.apache.accumulo.core.client.admin.InstanceOperations;
+import org.apache.accumulo.core.client.admin.SecurityOperations;
+import org.apache.accumulo.core.client.admin.TableOperations;
+import org.apache.accumulo.core.security.Authorizations;
+
+public class MockConnector extends Connector {
+
+ String username;
+ private final MockAccumulo acu;
+
+ MockConnector(String username) {
+ this(username, new MockAccumulo());
+ }
+
+ @SuppressWarnings("deprecation")
+ // Not really deprecated, just discouraging client use.
+ MockConnector(String username, MockAccumulo acu) {
+ this.username = username;
+ this.acu = acu;
+ }
+
+ @Override
+ public BatchScanner createBatchScanner(String tableName, Authorizations authorizations, int numQueryThreads) throws TableNotFoundException {
+ if (acu.tables.get(tableName) == null)
+ throw new TableNotFoundException(tableName, tableName, "no such table");
+ return acu.createBatchScanner(tableName, authorizations);
+ }
+
+ @Override
+ public BatchDeleter createBatchDeleter(String tableName, Authorizations authorizations, int numQueryThreads, long maxMemory, long maxLatency,
+ int maxWriteThreads) throws TableNotFoundException {
+ if (acu.tables.get(tableName) == null)
+ throw new TableNotFoundException(tableName, tableName, "no such table");
+ return new MockBatchDeleter(acu, tableName, authorizations);
+ }
+
+ @Override
+ public BatchWriter createBatchWriter(String tableName, long maxMemory, long maxLatency, int maxWriteThreads) throws TableNotFoundException {
+ if (acu.tables.get(tableName) == null)
+ throw new TableNotFoundException(tableName, tableName, "no such table");
+ return new MockBatchWriter(acu, tableName);
+ }
+
+ @Override
+ public MultiTableBatchWriter createMultiTableBatchWriter(long maxMemory, long maxLatency, int maxWriteThreads) {
+ return new MockMultiTableBatchWriter(acu);
+ }
+
+ @Override
+ public Scanner createScanner(String tableName, Authorizations authorizations) throws TableNotFoundException {
+ MockTable table = acu.tables.get(tableName);
+ if (table == null)
+ throw new TableNotFoundException(tableName, tableName, "no such table");
+ return new MockScanner(table, authorizations);
+ }
+
+ @Override
+ public Instance getInstance() {
+ return new MockInstance();
+ }
+
+ @Override
+ public String whoami() {
+ return username;
+ }
+
+ @Override
+ public TableOperations tableOperations() {
+ return new MockTableOperations(acu, username);
+ }
+
+ @Override
+ public SecurityOperations securityOperations() {
+ return new MockSecurityOperations(acu);
+ }
+
+ @Override
+ public InstanceOperations instanceOperations() {
+ return new MockInstanceOperations(acu);
+ }
+
+}
http://git-wip-us.apache.org/repos/asf/accumulo/blob/7bdbfccb/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockInstance.java
----------------------------------------------------------------------
diff --git a/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockInstance.java b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockInstance.java
new file mode 100644
index 0000000..af93066
--- /dev/null
+++ b/1.5/core/src/main/java/org/apache/accumulo/core/client/mock/MockInstance.java
@@ -0,0 +1,123 @@
+/*
+ * 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.core.client.mock;
+
+import java.nio.ByteBuffer;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.apache.accumulo.core.client.AccumuloException;
+import org.apache.accumulo.core.client.AccumuloSecurityException;
+import org.apache.accumulo.core.client.Connector;
+import org.apache.accumulo.core.client.Instance;
+import org.apache.accumulo.core.conf.AccumuloConfiguration;
+import org.apache.accumulo.core.security.Authorizations;
+import org.apache.accumulo.core.security.thrift.AuthInfo;
+import org.apache.accumulo.core.util.ByteBufferUtil;
+import org.apache.accumulo.core.util.TextUtil;
+import org.apache.hadoop.io.Text;
+
+public class MockInstance implements Instance {
+
+ static final String genericAddress = "localhost:1234";
+ static final Map<String,MockAccumulo> instances = new HashMap<String,MockAccumulo>();
+ MockAccumulo acu;
+ String instanceName;
+
+ public MockInstance() {
+ acu = new MockAccumulo();
+ instanceName = "mock-instance";
+ }
+
+ public MockInstance(String instanceName) {
+ synchronized (instances) {
+ if (instances.containsKey(instanceName))
+ acu = instances.get(instanceName);
+ else
+ instances.put(instanceName, acu = new MockAccumulo());
+ }
+ this.instanceName = instanceName;
+ }
+
+ @Override
+ public String getRootTabletLocation() {
+ return genericAddress;
+ }
+
+ @Override
+ public List<String> getMasterLocations() {
+ return Collections.singletonList(genericAddress);
+ }
+
+ @Override
+ public String getInstanceID() {
+ return "mock-instance-id";
+ }
+
+ @Override
+ public String getInstanceName() {
+ return instanceName;
+ }
+
+ @Override
+ public String getZooKeepers() {
+ return "localhost";
+ }
+
+ @Override
+ public int getZooKeepersSessionTimeOut() {
+ return 30 * 1000;
+ }
+
+ @Override
+ public Connector getConnector(String user, byte[] pass) throws AccumuloException, AccumuloSecurityException {
+ Connector conn = new MockConnector(user, acu);
+ conn.securityOperations().createUser(user, pass, new Authorizations());
+ return conn;
+ }
+
+ @Override
+ public Connector getConnector(String user, ByteBuffer pass) throws AccumuloException, AccumuloSecurityException {
+ return getConnector(user, ByteBufferUtil.toBytes(pass));
+ }
+
+ @Override
+ public Connector getConnector(String user, CharSequence pass) throws AccumuloException, AccumuloSecurityException {
+ return getConnector(user, TextUtil.getBytes(new Text(pass.toString())));
+ }
+
+ AccumuloConfiguration conf = null;
+
+ @Override
+ public AccumuloConfiguration getConfiguration() {
+ if (conf == null)
+ conf = AccumuloConfiguration.getDefaultConfiguration();
+ return conf;
+ }
+
+ @Override
+ public void setConfiguration(AccumuloConfiguration conf) {
+ this.conf = conf;
+ }
+
+ @Override
+ public Connector getConnector(AuthInfo auth) throws AccumuloException, AccumuloSecurityException {
+ return getConnector(auth.user, auth.password);
+ }
+}