@Test
  public void test_walk() {
    final RiakClient c = new RiakClient(RIAK_URL);

    final String ROOT = "root";
    final String LEAF1 = "leaf1";
    final String LEAF2 = "leaf2";
    final String EXCLUDED_LEAF = "excluded_leaf";
    final byte[] INCLUDED_VALUE = CharsetUtils.utf8StringToBytes("included");
    final byte[] EXCLUDED_VALUE = CharsetUtils.utf8StringToBytes("excluded");
    final String TAG_INCLUDE = "tag_include";
    final String TAG_EXCLUDE = "tag_exclude";

    // Clear out the objects we're testing with
    assertSuccess(c.delete(BUCKET, ROOT));
    assertSuccess(c.delete(BUCKET, LEAF1));
    assertSuccess(c.delete(BUCKET, LEAF2));
    assertSuccess(c.delete(BUCKET, EXCLUDED_LEAF));

    // Add a few objects
    RiakObject leaf1 = new RiakObject(BUCKET, LEAF1, INCLUDED_VALUE);
    RiakObject leaf2 = new RiakObject(BUCKET, LEAF2, INCLUDED_VALUE);
    RiakObject excludedLeaf = new RiakObject(BUCKET, EXCLUDED_LEAF, EXCLUDED_VALUE);
    RiakObject root =
        new RiakObject(c, BUCKET, ROOT)
            .addLink(new RiakLink(BUCKET, LEAF1, TAG_INCLUDE))
            .addLink(new RiakLink(BUCKET, LEAF2, TAG_INCLUDE))
            .addLink(new RiakLink(BUCKET, EXCLUDED_LEAF, TAG_EXCLUDE));
    assertSuccess(c.store(root, WRITE_3_REPLICAS()));
    assertSuccess(c.store(leaf1, WRITE_3_REPLICAS()));
    assertSuccess(c.store(leaf2, WRITE_3_REPLICAS()));
    assertSuccess(c.store(excludedLeaf, WRITE_3_REPLICAS()));

    // Perform walk
    WalkResponse walkresp = root.walk(BUCKET, TAG_INCLUDE).run();
    assertSuccess(walkresp);
    assertTrue(walkresp.hasSteps());
    assertEquals(1, walkresp.getSteps().size());
    assertEquals(2, walkresp.getSteps().get(0).size());

    // Verify expected only linked to objects are returned
    List<? extends List<RiakObject>> steps = walkresp.getSteps();
    List<String> keys = new ArrayList<String>();
    for (List<RiakObject> step : steps) {
      for (RiakObject object : step) {
        keys.add(object.getKey());
        assertArrayEquals(INCLUDED_VALUE, object.getValueAsBytes());
      }
    }
    assertTrue(keys.contains(LEAF1));
    assertTrue(keys.contains(LEAF2));
  }
Exemplo n.º 2
0
    private void dumpToRiak() {
      try {
        if (blogStats.blogs.size() == 0) return;
        int count = 0;
        Date t0 = new Date();
        System.out.println(
            "dumping " + blogStats.blogs.size() + " blogstats into riak. First key: " + firstKey);
        for (String blogId : blogStats.blogs.keySet()) {
          count++;
          Blog blog = blogStats.blogs.get(blogId);

          RiakObject riakObject = new RiakObject(bucketName, blogId);
          String json = serialize(blog);
          if (count % 1000 == 0) {
            long millis = System.currentTimeMillis() - start.getTime();
            System.out.println(
                "dumped: "
                    + count
                    + " blogs. firstKey: "
                    + firstKey
                    + " totalDumped: "
                    + totalDumped
                    + " totalSecs: "
                    + millis / 1000
                    + " prsec "
                    + (1000 * totalDumped / millis));
          }

          riakObject.setValue(json);
          riakObject.setContentType("application/json");
          riakClient.store(riakObject);
          totalDumped++;
        }
        long elapsed = System.currentTimeMillis() - t0.getTime();
        System.out.println(
            "Done dumping data into riak. TooK: "
                + elapsed
                + " milliseconds. records pr. sec: "
                + 1000 * (blogStats.blogs.size() / elapsed));
        blogStats.blogs = new HashMap<String, Blog>();
      } catch (Exception e) {
        e.printStackTrace(); // To change body of catch statement use File | Settings | File
        // Templates.
      }
    }