DefaultRandomGenerator.java

package com.linkedin.databus.eventgenerator;
/*
 *
 * Copyright 2013 LinkedIn Corp. All rights reserved
 *
 * Licensed 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.
 *
*/


import java.util.Random;

public class DefaultRandomGenerator implements RandomDataGenerator
{

  Random rand;
  public String validChars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"; //.,/';\\][=-`<>?\":|}{+_)(*&^%$#@!~";

  public DefaultRandomGenerator() {
    rand = new Random();
  }

  public int getNextInt()
  {
    //return getNextInt(IntegerFieldGenerate.getMinIntLength(), IntegerFieldGenerate.getMaxIntLength());
    return getNextInt(0, IntegerFieldGenerate.getMaxIntLength());
  }
  

  public int getNextInt(int min, int max)
  {
    if( max == min ) return min;

    int range = max - min;
    range = range > 0 ? range: -range; //Prevent integer overflow.
    int generated = min + rand.nextInt(range);
    generated = generated > max ? max: generated;
    return generated;
  }

  public String getNextString()
  {
    return getNextString(StringFieldGenerate.getMinStringLength(), StringFieldGenerate.getMaxStringLength());
  }

  public String getNextString(int min, int max)
  {     
    int length = getNextInt(min, max);

    StringBuffer strbld = new StringBuffer();
    for( int i = 0; i < length; i++)
    {
      char ch = validChars.charAt(rand.nextInt(validChars.length()));
      strbld.append(ch);
    }

    return strbld.toString();
  }

  public double getNextDouble() {
    return rand.nextDouble();
  }

  public float getNextFloat() {
    return rand.nextFloat();
  }

  public long getNextLong() {
    long randomLong = rand.nextLong();

    return randomLong == Long.MIN_VALUE ? 0: Math.abs(randomLong);
  }

  public boolean getNextBoolean()
  {
    return rand.nextBoolean();
  }

  public  byte[] getNextBytes(int maxBytesLength)
  {
    byte[] bytes = new byte[this.getNextInt(0, maxBytesLength)];
    rand.nextBytes(bytes);
    return bytes;
  }
}