예제 #1
0
 /*
  * Convenience method for displaying output when debugging.
  */
 private void displayRow(Table tbl) {
   final TableIterator<Row> itr = tableAPI.tableIterator(tbl.createPrimaryKey(), null, null);
   while (itr.hasNext()) {
     System.out.println(itr.next());
   }
   itr.close();
 }
예제 #2
0
  private void deleteExistingData() {

    /* Get an iterator over all the primary keys in the table. */
    final TableIterator<PrimaryKey> itr =
        tableAPI.tableKeysIterator(table.createPrimaryKey(), null, null);

    /* Delete each row from the table. */
    long cnt = 0;
    while (itr.hasNext()) {
      tableAPI.delete(itr.next(), null, null);
      cnt++;
    }
    itr.close();
    System.out.println(cnt + " records deleted");
  }
예제 #3
0
  private void addRow() {

    final Row row = table.createRow();

    /* Generate zipcode */
    final StringBuilder zipBuf = new StringBuilder();
    for (int i = 0; i < 5; i++) {
      final int zipDigit = secureRandom.nextInt(10);
      zipBuf.append(zipDigit);
    }
    final String zipStr = zipBuf.toString();
    row.put("zipcode", zipStr);

    /* Generate ssn */
    final StringBuilder ssnBuf = new StringBuilder();
    for (int i = 0; i < 9; i++) {
      final int ssnDigit = secureRandom.nextInt(10);
      ssnBuf.append(ssnDigit);
    }
    row.put("ssn", Long.parseLong(ssnBuf.toString()));

    /* Generate lastname */
    final int lastnameIndx = secureRandom.nextInt(LAST_NAMES.length);
    row.put("lastname", LAST_NAMES[lastnameIndx]);

    /* Generate firstname and gender */
    int firstnameIndx = secureRandom.nextInt(FIRST_NAMES_FEMALE.length);
    String firstname = FIRST_NAMES_FEMALE[firstnameIndx];
    String gender = "female";
    final int maleFemale = secureRandom.nextInt(2);
    if (maleFemale == 0) {
      firstnameIndx = secureRandom.nextInt(FIRST_NAMES_MALE.length);
      firstname = FIRST_NAMES_MALE[firstnameIndx];
      gender = "male";
    }
    row.put("firstname", firstname);
    row.putEnum("gender", gender);

    /* Generate license id as String, but store it as BINARY */
    final StringBuilder licenseBuf = new StringBuilder();
    for (int i = 0; i < 9; i++) {
      if (i > 0) {
        final int licenseDigit = secureRandom.nextInt(10);
        licenseBuf.append(String.valueOf(licenseDigit));
      } else {
        licenseBuf.append("S");
      }
    }
    row.putFixed("license", (licenseBuf.toString()).getBytes());

    /* Generate phoneinfo */
    final MapValue phoneMap = row.putMap("phoneinfo");
    for (int i = 0; i < PHONE_TYPE.length; i++) {
      final String phonetype = PHONE_TYPE[i];

      final int areaCodeIndx = secureRandom.nextInt(AREA_CODES.length);
      final int exchangeIndx = secureRandom.nextInt(EXCHANGES.length);
      final String areacode = AREA_CODES[areaCodeIndx];
      final String exchange = EXCHANGES[exchangeIndx];

      final StringBuilder phoneBuf = new StringBuilder();
      phoneBuf.append(areacode);
      phoneBuf.append("-");
      phoneBuf.append(exchange);
      phoneBuf.append("-");

      for (int j = 0; j < 4; j++) {
        final int phoneDigit = secureRandom.nextInt(10);
        phoneBuf.append(phoneDigit);
      }
      final String phone = phoneBuf.toString();
      phoneMap.put(phonetype, phone);
    }

    final RecordValue addressRec = row.putRecord("address");

    final int streetNumber = secureRandom.nextInt(99999);
    final int streetNameIndx = secureRandom.nextInt(STREET_NAMES.length);
    final int unitNumber = secureRandom.nextInt(11) - 1;
    final int cityNameIndx = secureRandom.nextInt(CITY_NAMES.length);
    final int stateIndx = secureRandom.nextInt(STATE_ABBREVIATIONS.length);

    addressRec.put("number", streetNumber);
    addressRec.put("street", STREET_NAMES[streetNameIndx]);
    addressRec.put("unit", unitNumber);
    addressRec.put("city", CITY_NAMES[cityNameIndx]);
    addressRec.put("state", STATE_ABBREVIATIONS[stateIndx]);
    addressRec.put("zip", Integer.parseInt(zipStr));

    /* Generate vehicleinfo */
    final ArrayValue vehicleArray = row.putArray("vehicleinfo");

    final int nVehicles = 1 + secureRandom.nextInt(3);

    for (int i = 0; i < nVehicles; i++) {

      final RecordValue vehicleRec = vehicleArray.addRecord();

      final int typeIndx = secureRandom.nextInt(TYPES.length);
      final int makeIndx = secureRandom.nextInt(MAKES.length);
      final int classIndx = secureRandom.nextInt(CLASSES.length);
      final int colorIndx = secureRandom.nextInt(COLORS.length);

      final String type = TYPES[typeIndx];
      final String make = MAKES[makeIndx];
      final String vClass = CLASSES[classIndx];
      final String color = COLORS[colorIndx];

      String[] models = MODELS_FORD_AUTO;
      float valueMult = 2.0f;

      if ("suv".equals(type)) {
        valueMult = 4.0f;
        if ("Chrysler".equals(make)) {
          models = MODELS_CHRYSLER_SUV;
        } else if ("GM".equals(make)) {
          models = MODELS_GM_SUV;
        } else {
          /* Default to make "Ford" */
          models = MODELS_FORD_SUV;
        }
      } else if ("truck".equals(type)) {
        valueMult = 3.0f;
        if ("Chrysler".equals(make)) {
          models = MODELS_CHRYSLER_TRUCK;
        } else if ("GM".equals(make)) {
          models = MODELS_GM_TRUCK;
        } else {
          /* Default to make "Ford" */
          models = MODELS_FORD_TRUCK;
        }
      } else {
        /* Default to type "auto" */
        if ("Chrysler".equals(make)) {
          models = MODELS_CHRYSLER_AUTO;
        } else if ("GM".equals(make)) {
          models = MODELS_GM_AUTO;
        }
      }
      final int modelIndx = secureRandom.nextInt(models.length);
      final String model = models[modelIndx];

      final float baseValue = 10371.59f;
      final float deltaValue = secureRandom.nextFloat();
      final float value = (valueMult * baseValue) + deltaValue;

      final double tax = TAX_RATE * value;

      boolean paid = true;
      final int paidOrNot = secureRandom.nextInt(2);
      if (paidOrNot == 0) {
        paid = false;
      }

      vehicleRec.put("type", type);
      vehicleRec.put("make", make);
      vehicleRec.put("model", model);
      vehicleRec.put("class", vClass);
      vehicleRec.put("color", color);
      vehicleRec.put("value", value);
      vehicleRec.put("tax", tax);
      vehicleRec.put("paid", paid);
    }

    /* Handle duplicates. */
    final PrimaryKey dupKey = row.createPrimaryKey();
    final Row dupRow = tableAPI.get(dupKey, null);
    if (dupRow != null) {
      /* Change the ssn */
      int newSsnDigit = secureRandom.nextInt(10);
      final StringBuilder newSsnBuf = new StringBuilder(newSsnDigit);
      for (int i = 0; i < 8; i++) {
        newSsnDigit = secureRandom.nextInt(10);
        newSsnBuf.append(newSsnDigit);
      }
      row.put("ssn", Long.parseLong(newSsnBuf.toString()));
    }

    /* Finally, write the row created above to the table */
    tableAPI.putIfAbsent(row, null, null);
  }