| /* |
| * 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.hadoop.fi; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.Random; |
| |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| |
| /** Test Utilities */ |
| public class FiTestUtil { |
| /** Logging */ |
| public static final Log LOG = LogFactory.getLog(FiTestUtil.class); |
| |
| /** Random source */ |
| public static final ThreadLocal<Random> RANDOM = new ThreadLocal<Random>() { |
| protected Random initialValue() { |
| final Random r = new Random(); |
| final long seed = r.nextLong(); |
| LOG.info(Thread.currentThread() + ": seed=" + seed); |
| r.setSeed(seed); |
| return r; |
| } |
| }; |
| |
| /** |
| * Return a random integer uniformly distributed over the interval [min,max). |
| */ |
| public static int nextRandomInt(final int min, final int max) { |
| final int d = max - min; |
| if (d <= 0) { |
| throw new IllegalArgumentException("d <= 0, min=" + min + ", max=" + max); |
| } |
| return d == 1? min: min + RANDOM.get().nextInt(d); |
| } |
| |
| /** |
| * Return a random integer, with type long, |
| * uniformly distributed over the interval [min,max). |
| * Assume max - min <= Integer.MAX_VALUE. |
| */ |
| public static long nextRandomLong(final long min, final long max) { |
| final long d = max - min; |
| if (d <= 0 || d > Integer.MAX_VALUE) { |
| throw new IllegalArgumentException( |
| "d <= 0 || d > Integer.MAX_VALUE, min=" + min + ", max=" + max); |
| } |
| return d == 1? min: min + RANDOM.get().nextInt((int)d); |
| } |
| |
| /** Return the method name of the callee. */ |
| public static String getMethodName() { |
| final StackTraceElement[] s = Thread.currentThread().getStackTrace(); |
| return s[s.length > 2? 2: s.length - 1].getMethodName(); |
| } |
| |
| /** |
| * Sleep. |
| * @return true if sleep exits normally; false if InterruptedException. |
| */ |
| public static boolean sleep(long ms) { |
| LOG.info("Sleep " + ms + " ms"); |
| try { |
| Thread.sleep(ms); |
| } catch (InterruptedException e) { |
| LOG.info("Sleep is interrupted", e); |
| return false; |
| } |
| return true; |
| } |
| |
| /** |
| * Sleep a random number of milliseconds over the interval [min, max). |
| * If there is an InterruptedException, re-throw it as a RuntimeException. |
| */ |
| public static void sleep(final long min, final long max) { |
| final long n = nextRandomLong(min, max); |
| LOG.info(Thread.currentThread().getName() + " sleeps for " + n +"ms"); |
| if (n > 0) { |
| sleep(n); |
| } |
| } |
| |
| /** Action interface */ |
| public static interface Action<T, E extends Exception> { |
| /** Run the action with the parameter. */ |
| public void run(T parameter) throws E; |
| } |
| |
| /** An ActionContainer contains at most one action. */ |
| public static class ActionContainer<T, E extends Exception> { |
| private List<Action<T, E>> actionList = new ArrayList<Action<T, E>>(); |
| /** Create an empty container. */ |
| public ActionContainer() {} |
| |
| /** Set action. */ |
| public void set(Action<T, E> a) {actionList.add(a);} |
| |
| /** Run the action if it exists. */ |
| public void run(T obj) throws E { |
| for (Action<T, E> action : actionList) { |
| action.run(obj); |
| } |
| } |
| } |
| |
| /** Constraint interface */ |
| public static interface Constraint { |
| /** Is this constraint satisfied? */ |
| public boolean isSatisfied(); |
| } |
| |
| /** Counting down, the constraint is satisfied if the count is one. */ |
| public static class CountdownConstraint implements Constraint { |
| private int count; |
| |
| /** Initialize the count. */ |
| public CountdownConstraint(int count) { |
| if (count < 1) { |
| throw new IllegalArgumentException(count + " = count < 1"); |
| } |
| this.count = count; |
| } |
| |
| /** Counting down, the constraint is satisfied if the count is zero. */ |
| public boolean isSatisfied() { |
| if (count > 1) { |
| count--; |
| return false; |
| } |
| return true; |
| } |
| } |
| |
| /** An action is fired if all the constraints are satisfied. */ |
| public static class ConstraintSatisfactionAction<T, E extends Exception> |
| implements Action<T, E> { |
| private final Action<T, E> action; |
| private final Constraint[] constraints; |
| |
| /** Constructor */ |
| public ConstraintSatisfactionAction( |
| Action<T, E> action, Constraint... constraints) { |
| this.action = action; |
| this.constraints = constraints; |
| } |
| |
| /** |
| * Fire the action if all the constraints are satisfied. |
| * Short-circuit-and is used. |
| */ |
| @Override |
| public final void run(T parameter) throws E { |
| for(Constraint c : constraints) { |
| if (!c.isSatisfied()) { |
| return; |
| } |
| } |
| |
| //all constraints are satisfied, fire the action |
| action.run(parameter); |
| } |
| } |
| |
| /** A MarkerConstraint is satisfied if it is marked. */ |
| public static class MarkerConstraint implements Constraint { |
| private final String name; |
| private boolean marked = false; |
| |
| /** Construct an object. */ |
| public MarkerConstraint(String name) { |
| this.name = name; |
| } |
| |
| /** Set marker to be marked. */ |
| public void mark() { |
| marked = true; |
| LOG.info("Marking this " + this); |
| } |
| |
| /** Is the marker marked? */ |
| @Override |
| public boolean isSatisfied() { |
| return marked; |
| } |
| |
| /** {@inheritDoc} */ |
| public String toString() { |
| return getClass().getSimpleName() + "[" + name + ": " + marked + "]"; |
| } |
| } |
| } |