/** Test the Client.getShardMaps() without a match pattern (all maps). */
  public void testDumpShardMapsAll() throws Exception {
    Set<ShardMapping> shardMappings = this.client.getShardMappings().getData();
    assertEquals(1, shardMappings.size());
    ShardMapping shardMapping = shardMappings.iterator().next();

    assertEquals(1, shardMapping.getShardTables().size());
  }
Exemplo n.º 2
0
  public void testHashShardMappingKeyLookup() throws Exception {
    final String globalGroupName = "My global group";

    final String lowerBounds[] =
        new String[] {
          /* 0 = */ "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
          /* 1 = */ "66666666666666666666666666666666",
          /* 2 = */ "2809A05A22A4A9C1882A580BCC0AD8A6",
          /* 3 = */ "DDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDD"
        };

    // setup the mapping
    Set<ShardIndex> shardIndices = new HashSet<ShardIndex>();
    int shardId = 0; // shard id's added sequentially increasing from 0
    for (String lowerBound : lowerBounds) {
      ShardIndex i = new ShardIndex(lowerBound, shardId, "server_group_" + shardId);
      shardId++;
      shardIndices.add(i);
    }
    ShardMapping mapping =
        new HashShardMapping(5000, ShardingType.HASH, globalGroupName, null, shardIndices);

    // test lookups mapping of test value to the group it maps to test values are hashed with MD5
    // and compared to lowerBounds values
    String testPairs[][] =
        new String[][] {
          // exact match should be in that shard
          new String[] {"Jess", "server_group_2"}, // hash = 2809a05a22a4a9c1882a580bcc0ad8a6
          new String[] {"x", "server_group_1"}, // hash = 9dd4e461268c8034f5c8564e155c67a6
          new String[] {"X", "server_group_3"}, // hash = 02129bb861061d1a052c592e2dc6b383
          new String[] {"Y", "server_group_2"}, // hash = 57cec4137b614c87cb4e24a3d003a3e0
          new String[] {"g", "server_group_0"}, // hash = b2f5ff47436671b6e533d8dc3614845d
          // leading zeroes
          new String[] {"168", "server_group_3"}, // hash = 006f52e9102a8d3be2fe5614f42ba989
        };

    for (String[] testPair : testPairs) {
      String key = testPair[0];
      String serverGroup = testPair[1];
      assertEquals(serverGroup, mapping.getGroupNameForKey(key));
    }

    // test a random set of values. we should never return null
    for (int i = 0; i < 1000; ++i) {
      assertNotNull(mapping.getGroupNameForKey("" + i));
    }
  }
Exemplo n.º 3
0
  /** Test that a range based shard mapping looks up groups for keys correctly. */
  public void testRangeShardMappingKeyLookup() throws Exception {
    final String globalGroupName = "My global group";

    final int lowerBounds[] = new int[] {1, 10000, 1001, 400, 1000, 470};

    final int lowestLowerBound = 1;

    // setup the mapping
    Set<ShardIndex> shardIndices = new HashSet<ShardIndex>();
    int shardId = 0; // shard id's added sequentially increasing from 0
    for (Integer lowerBound : lowerBounds) {
      ShardIndex i = new ShardIndex(String.valueOf(lowerBound), shardId, "shard_group_" + shardId);
      shardId++;
      shardIndices.add(i);
    }
    ShardMapping mapping =
        new RangeShardMapping(5000, ShardingType.RANGE, globalGroupName, null, shardIndices);

    // try adding a second shard index with a lower bound that conflicts with an existing one this
    // should be prohibited
    // mapping.addShardIndex(new ShardIndex(mapping, "1", 0, ""));

    // test looking up a key out of range doesn't work
    try {
      mapping.getGroupNameForKey(String.valueOf(lowestLowerBound - 1));
      fail("Looking up a key with a value below the lowest bound is invalid");
    } catch (Exception ex) {
    }
    try {
      mapping.getGroupNameForKey(String.valueOf(lowestLowerBound - 1000));
      fail("Looking up a key with a value below the lowest bound is invalid");
    } catch (Exception ex) {
    }

    // test key lookups for lower bound values
    for (shardId = 0; shardId < lowerBounds.length; ++shardId) {
      int lowerBound = lowerBounds[shardId];
      String groupName = mapping.getGroupNameForKey(String.valueOf(lowerBound));
      assertEquals("Exact lookup for key " + lowerBound, "shard_group_" + shardId, groupName);
    }
  }