Package eu.stratosphere.types

Examples of eu.stratosphere.types.Record


    do {
      lower = upper;
      upper = lower;
      // find the upper bound
      while (iterator.hasNext()) {
        Record target = iterator.next();
        int val = target.getField(0, IntValue.class).getValue();
        Assert.assertEquals(upper++, val);
      }
      // now reset the buffer a few times
      for (int i = 0; i < 5; ++i) {
        iterator.reset();
        int count = 0;
        while (iterator.hasNext()) {
          Record target = iterator.next();
          int val = target.getField(0, IntValue.class).getValue();
          Assert.assertEquals(lower + (count++), val);
        }
        Assert.assertEquals(upper - lower, count);
      }
    } while (iterator.nextBlock());
View Full Code Here


    do {
      lower = upper;
      upper = lower;
      // find the upper bound
      while (iterator.hasNext()) {
        Record target = iterator.next();
        int val = target.getField(0, IntValue.class).getValue();
        Assert.assertEquals(upper++, val);
      }
      // now reset the buffer a few times
      for (int i = 0; i < 5; ++i) {
        iterator.reset();
        int count = 0;
        while (iterator.hasNext()) {
          Record target = iterator.next();
          int val = target.getField(0, IntValue.class).getValue();
          Assert.assertEquals(lower + (count++), val);
        }
        Assert.assertEquals(upper - lower, count);
      }
    } while (iterator.nextBlock());
View Full Code Here

    this.queue = new ArrayBlockingQueue<Record>(size, false);
  }

  @Override
  public Record next(Record reuse) {
    Record r = null;
    while (r == null) {
      try {
        r = queue.take();
      } catch (InterruptedException iex) {
        throw new RuntimeException("Reader was interrupted.");
      }
    }

    if (r == SENTINEL) {
      // put the sentinel back, to ensure that repeated calls do not block
      try {
        queue.put(r);
      } catch (InterruptedException e) {
        throw new RuntimeException("Reader was interrupted.");
      }
      return null;
    } else {
      r.copyTo(reuse);
      return reuse;
    }
  }
View Full Code Here

       
        TestData.Value v1 = null;
        TestData.Value v2 = null;
       
        if (iter1.hasNext()) {
          Record rec = iter1.next();
          rec.getFieldInto(0, key);
          v1 = rec.getField(1, TestData.Value.class);
        }
        else if (iter2.hasNext()) {
          Record rec = iter2.next();
          rec.getFieldInto(0, key);
          v2 = rec.getField(1, TestData.Value.class);
        }
        else {
          Assert.fail("No input on both sides.");
        }
 
        // assert that matches for this key exist
        Assert.assertTrue("No matches for key " + key, expectedCoGroupsMap.containsKey(key));
       
        Collection<TestData.Value> expValues1 = expectedCoGroupsMap.get(key).get(0);
        Collection<TestData.Value> expValues2 = expectedCoGroupsMap.get(key).get(1);
       
        if (v1 != null) {
          expValues1.remove(v1);
        }
        else {
          expValues2.remove(v2);
        }
       
        while(iter1.hasNext()) {
          Record rec = iter1.next();
          Assert.assertTrue("Value not in expected set of first input", expValues1.remove(rec.getField(1, TestData.Value.class)));
        }
        Assert.assertTrue("Expected set of first input not empty", expValues1.isEmpty());
       
        while(iter2.hasNext()) {
          Record rec = iter2.next();
          Assert.assertTrue("Value not in expected set of second input", expValues2.remove(rec.getField(1, TestData.Value.class)));
        }
        Assert.assertTrue("Expected set of second input not empty", expValues2.isEmpty());
 
        expectedCoGroupsMap.remove(key);
      }
View Full Code Here

  private Map<TestData.Key, Collection<TestData.Value>> collectData(Generator iter, int num)
  throws Exception
  {
    Map<TestData.Key, Collection<TestData.Value>> map = new HashMap<TestData.Key, Collection<TestData.Value>>();
    Record pair = new Record();
   
    for (int i = 0; i < num; i++) {
      iter.next(pair);
      TestData.Key key = pair.getField(0, TestData.Key.class);
     
      if (!map.containsKey(key)) {
        map.put(new TestData.Key(key.getKey()), new ArrayList<TestData.Value>());
      }

      Collection<TestData.Value> values = map.get(key);
      values.add(new TestData.Value(pair.getField(1, TestData.Value.class).getValue()));
    }
    return map;
  }
View Full Code Here

    MutableObjectIterator<Record> iterator = merger.getIterator();
   
    LOG.debug("Checking results...");
    int pairsEmitted = 1;

    Record rec1 = new Record();
    Record rec2 = new Record();
   
    Assert.assertTrue((rec1 = iterator.next(rec1)) != null);
    while ((rec2 = iterator.next(rec2)) != null) {
      final Key k1 = rec1.getField(0, TestData.Key.class);
      final Key k2 = rec2.getField(0, TestData.Key.class);
      pairsEmitted++;
     
      Assert.assertTrue(keyComparator.compare(k1, k2) <= 0);
     
      Record tmp = rec1;
      rec1 = rec2;
      k1.setKey(k2.getKey());
     
      rec2 = tmp;
    }
View Full Code Here

    MutableObjectIterator<Record> iterator = merger.getIterator();
   
    LOG.debug("Checking results...");
    int pairsEmitted = 1;

    Record rec1 = new Record();
    Record rec2 = new Record();
   
    Assert.assertTrue((rec1 = iterator.next(rec1)) != null);
    while ((rec2 = iterator.next(rec2)) != null) {
      final Key k1 = rec1.getField(0, TestData.Key.class);
      final Key k2 = rec2.getField(0, TestData.Key.class);
      pairsEmitted++;
     
      Assert.assertTrue(keyComparator.compare(k1, k2) <= 0);
     
      Record tmp = rec1;
      rec1 = rec2;
      k1.setKey(k2.getKey());
     
      rec2 = tmp;
    }
View Full Code Here

    MutableObjectIterator<Record> iterator = merger.getIterator();
   
    LOG.debug("Checking results...");
    int pairsEmitted = 1;

    Record rec1 = new Record();
    Record rec2 = new Record();
   
    Assert.assertTrue((rec1 = iterator.next(rec1)) != null);
    while ((rec2 = iterator.next(rec2)) != null) {
      final Key k1 = rec1.getField(0, TestData.Key.class);
      final Key k2 = rec2.getField(0, TestData.Key.class);
      pairsEmitted++;
     
      Assert.assertTrue(keyComparator.compare(k1, k2) <= 0);
     
      Record tmp = rec1;
      rec1 = rec2;
      k1.setKey(k2.getKey());
     
      rec2 = tmp;
    }
View Full Code Here

   
    LOG.debug("Checking results...");
    int pairsRead = 1;
    int nextStep = PAIRS / 20;

    Record rec1 = new Record();
    Record rec2 = new Record();
   
    Assert.assertTrue((rec1 = iterator.next(rec1)) != null);
    while ((rec2 = iterator.next(rec2)) != null) {
      final Key k1 = rec1.getField(0, TestData.Key.class);
      final Key k2 = rec2.getField(0, TestData.Key.class);
      pairsRead++;
     
      Assert.assertTrue(keyComparator.compare(k1, k2) <= 0);
     
      Record tmp = rec1;
      rec1 = rec2;
      k1.setKey(k2.getKey());
      rec2 = tmp;
     
      // log
View Full Code Here

 
  private Map<TestData.Key, Collection<TestData.Value>> collectData(MutableObjectIterator<Record> iter)
  throws Exception
  {
    Map<TestData.Key, Collection<TestData.Value>> map = new HashMap<TestData.Key, Collection<TestData.Value>>();
    Record pair = new Record();
   
    while ((pair = iter.next(pair)) != null) {
      TestData.Key key = pair.getField(0, TestData.Key.class);
     
      if (!map.containsKey(key)) {
        map.put(new TestData.Key(key.getKey()), new ArrayList<TestData.Value>());
      }

      Collection<TestData.Value> values = map.get(key);
      values.add(new TestData.Value(pair.getField(1, TestData.Value.class).getValue()));
    }

    return map;
  }
View Full Code Here

TOP

Related Classes of eu.stratosphere.types.Record

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.