コード例 #1
0
  /**
   * Generates {@link #countToGenerate} ids, and insures that they are all unique.
   *
   * @throws Exception
   */
  @Test
  public void testGenerateUniqueIdsSingleThread() throws Exception {
    long start = Calendar.getInstance().getTimeInMillis();
    for (int i = 0; i < countToGenerate; i++) {
      // Select an ID type to generate, based on a randomized seed.
      double seed = Math.random();
      log.trace("Seed is {}", seed);
      Types t = selectType(seed, Types.values());
      log.trace("Selected type {} with seed value {}", t, seed);

      // If debugging is enabled, keep track of the number of ids created for each type.
      if (log.isDebugEnabled()) {
        if (idTypeDistribution.containsKey(t)) {
          idTypeDistribution.get(t).getAndAdd(1);
        } else {
          idTypeDistribution.put(t, new AtomicInteger(1));
        }
      }

      // Create an ID, and keep it in a Set.
      generatedIds.add(reqFactory.createIdApiRequest(t).execute(hc));
    }

    if (log.isDebugEnabled()) {
      StringBuilder sb = new StringBuilder("ID distribution:\n");
      int totalGenerated = 0;
      for (Types t : Types.values()) {
        final Integer typeTotal = idTypeDistribution.get(t).get();
        totalGenerated += typeTotal;
        sb.append("Type: ").append(t).append(" Count: ").append(typeTotal).append("\n");
      }
      sb.append("Total generated: ").append(totalGenerated).append("\n");
      sb.append("Unique generated: ").append(generatedIds.size()).append("\n");
      sb.append("Execution time: ")
          .append(Calendar.getInstance().getTimeInMillis() - start)
          .append(" ms\n");
      log.debug(sb.toString());
    }

    // The number of generated IDs (stored in the Set) should equal 'countToGenerate'
    assertEquals(
        "Expected "
            + countToGenerate
            + " to be generated, but the Set contained "
            + generatedIds.size()
            + ".  Some ids may not have been unique.",
        countToGenerate,
        generatedIds.size());
  }
コード例 #2
0
  /**
   * Generates {@link #countToGenerate} ids, and insures that they are all unique. Uses multiple
   * threads to generate the ids.
   *
   * @throws Exception
   */
  @Test
  public void testGenerateUniqueIdsMultipleThreads() throws Exception {
    long start = Calendar.getInstance().getTimeInMillis();
    // The threads used to generate ids
    Thread threads[] = new Thread[5];

    // HttpClient requires a ThreadSafeClientConnectionManager
    final ThreadSafeClientConnManager conman = new ThreadSafeClientConnManager();
    conman.setMaxTotal(50);
    conman.setDefaultMaxPerRoute(5);
    hc = new DefaultHttpClient(conman);

    assertEquals(
        "The number of threads ("
            + threads.length
            + ") must evenly divide into thenumber of ids to be "
            + "generated ("
            + countToGenerate
            + ")",
        0,
        countToGenerate % threads.length);
    final int generatePerThread = countToGenerate / threads.length;

    // Launch a thread, with each thread being responsible for generating a portion of the total ids
    for (int j = 0; j < threads.length; j++) {
      threads[j] =
          new Thread(
              new Runnable() {
                @Override
                public void run() {
                  for (int i = 0; i < generatePerThread; i++) {
                    double seed = Math.random();
                    log.trace("Seed is {}", seed);
                    Types t = selectType(seed, Types.values());
                    log.trace("Selected type {} with seed value {}", t, seed);

                    if (log.isDebugEnabled()) {
                      idTypeDistribution.putIfAbsent(t, new AtomicInteger(0));
                      idTypeDistribution.get(t).getAndAdd(1);
                    }

                    try {
                      generatedIds.add(reqFactory.createIdApiRequest(t).execute(hc));
                    } catch (IOException e) {
                      fail(e.getMessage());
                    }
                  }
                }
              },
              "ID Generation Thread " + j);
      threads[j].start();
    }

    // Wait for threads to stop
    for (int j = 0; j < threads.length; j++) {
      threads[j].join();
    }

    if (log.isDebugEnabled()) {
      StringBuilder sb = new StringBuilder("ID distribution:\n");
      int totalGenerated = 0;
      for (Types t : Types.values()) {
        final Integer typeTotal = idTypeDistribution.get(t).intValue();
        totalGenerated += typeTotal;
        sb.append("Type: ").append(t).append(" Count: ").append(typeTotal).append("\n");
      }
      sb.append("Total generated: ").append(totalGenerated).append("\n");
      sb.append("Unique generated: ").append(generatedIds.size()).append("\n");
      sb.append("Number of threads: ").append(threads.length).append("\n");
      sb.append("Execution time: ")
          .append(Calendar.getInstance().getTimeInMillis() - start)
          .append(" ms\n");
      log.debug(sb.toString());
    }

    // The number of generated IDs (stored in the Set) should equal 'countToGenerate'
    assertEquals(
        "Expected "
            + countToGenerate
            + " to be generated, but the Set contained "
            + generatedIds.size()
            + ".  Some ids may not have been unique.",
        countToGenerate,
        generatedIds.size());
  }