Examples with FieldExpression com.yahoo.bullet.query.expressions.FieldExpression used on opensource projects

Example 11 with FieldExpression

use of com.yahoo.bullet.query.expressions.FieldExpression in project bullet-core by yahoo.

the class OrderByStrategyTest method testOrderByAscendingWithDifferentType.

@Test
public void testOrderByAscendingWithDifferentType() {
    OrderByStrategy orderByStrategy = makeOrderBy(Arrays.asList(new OrderBy.SortItem(new FieldExpression("a"), OrderBy.Direction.ASC), new OrderBy.SortItem(new FieldExpression("b"), OrderBy.Direction.ASC)));
    List<BulletRecord> records = new ArrayList<>();
    records.add(RecordBox.get().add("a", 5).add("b", 2.0).getRecord());
    records.add(RecordBox.get().add("a", 2).add("b", 4).getRecord());
    records.add(RecordBox.get().add("a", 2).add("b", 2.0).getRecord());
    records.add(RecordBox.get().add("a", 1).add("b", 7).getRecord());
    Clip clip = new Clip();
    clip.add(records);
    Clip result = orderByStrategy.execute(clip);
    Assert.assertEquals(result.getRecords().get(0).typedGet("a").getValue(), 1);
    Assert.assertEquals(result.getRecords().get(0).typedGet("b").getValue(), 7);
    Assert.assertEquals(result.getRecords().get(1).typedGet("a").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(1).typedGet("b").getValue(), 2.0);
    Assert.assertEquals(result.getRecords().get(2).typedGet("a").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(2).typedGet("b").getValue(), 4);
    Assert.assertEquals(result.getRecords().get(3).typedGet("a").getValue(), 5);
    Assert.assertEquals(result.getRecords().get(3).typedGet("b").getValue(), 2.0);
}

Example 12 with FieldExpression

use of com.yahoo.bullet.query.expressions.FieldExpression in project bullet-core by yahoo.

the class OrderByStrategyTest method testOrderByComputationWithMultipleFields.

@Test
public void testOrderByComputationWithMultipleFields() {
    OrderByStrategy orderByStrategy = makeOrderBy(Collections.singletonList(new OrderBy.SortItem(new BinaryExpression(new FieldExpression("a"), new FieldExpression("b"), Operation.ADD), OrderBy.Direction.DESC)));
    List<BulletRecord> records = new ArrayList<>();
    records.add(RecordBox.get().add("a", 5).add("b", 2).getRecord());
    records.add(RecordBox.get().add("a", 2).add("b", 4).getRecord());
    records.add(RecordBox.get().add("a", 2).add("b", 2).getRecord());
    records.add(RecordBox.get().add("a", 1).add("b", 7).getRecord());
    Clip clip = new Clip();
    clip.add(records);
    Clip result = orderByStrategy.execute(clip);
    Assert.assertEquals(result.getRecords().get(0).typedGet("a").getValue(), 1);
    Assert.assertEquals(result.getRecords().get(0).typedGet("b").getValue(), 7);
    Assert.assertEquals(result.getRecords().get(1).typedGet("a").getValue(), 5);
    Assert.assertEquals(result.getRecords().get(1).typedGet("b").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(2).typedGet("a").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(2).typedGet("b").getValue(), 4);
    Assert.assertEquals(result.getRecords().get(3).typedGet("a").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(3).typedGet("b").getValue(), 2);
}

Example 13 with FieldExpression

use of com.yahoo.bullet.query.expressions.FieldExpression in project bullet-core by yahoo.

the class OrderByStrategyTest method testOrderByAscendingWithSomeMissingField.

@Test
public void testOrderByAscendingWithSomeMissingField() {
    OrderByStrategy orderByStrategy = makeOrderBy(Arrays.asList(new OrderBy.SortItem(new FieldExpression("a"), OrderBy.Direction.ASC), new OrderBy.SortItem(new FieldExpression("c"), OrderBy.Direction.ASC)));
    List<BulletRecord> records = new ArrayList<>();
    records.add(RecordBox.get().add("a", 5).add("b", 2).getRecord());
    records.add(RecordBox.get().add("a", 2).add("b", 4).getRecord());
    records.add(RecordBox.get().add("b", 2).getRecord());
    records.add(RecordBox.get().add("a", 1).add("b", 7).getRecord());
    Clip clip = new Clip();
    clip.add(records);
    Clip result = orderByStrategy.execute(clip);
    Assert.assertEquals(result.getRecords().get(0).typedGet("a").getValue(), null);
    Assert.assertEquals(result.getRecords().get(0).typedGet("b").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(1).typedGet("a").getValue(), 1);
    Assert.assertEquals(result.getRecords().get(1).typedGet("b").getValue(), 7);
    Assert.assertEquals(result.getRecords().get(2).typedGet("a").getValue(), 2);
    Assert.assertEquals(result.getRecords().get(2).typedGet("b").getValue(), 4);
    Assert.assertEquals(result.getRecords().get(3).typedGet("a").getValue(), 5);
    Assert.assertEquals(result.getRecords().get(3).typedGet("b").getValue(), 2);
}

Example 14 with FieldExpression

use of com.yahoo.bullet.query.expressions.FieldExpression in project bullet-core by yahoo.

the class LateralViewFunctorTest method testApplyWithExplode.

@Test
public void testApplyWithExplode() {
    LateralView lateralView = new LateralView(new Explode(new FieldExpression("mapA"), "foo", "bar", false));
    LateralViewFunctor functor = new LateralViewFunctor(lateralView);
    List<BulletRecord> records = functor.apply(record, provider);
    Assert.assertEquals(records.size(), 3);
    Assert.assertEquals(((LateralViewBulletRecord) records.get(0)).getBaseRecord().fieldCount(), 5);
    Assert.assertEquals(((LateralViewBulletRecord) records.get(0)).getTopRecord().fieldCount(), 2);
    Assert.assertEquals(records.get(0).typedGet("foo").getValue(), "a");
    Assert.assertEquals(records.get(0).typedGet("bar").getValue(), 0);
    Assert.assertFalse(records.get(0).typedGet("listA").isNull());
    Assert.assertFalse(records.get(0).typedGet("listB").isNull());
    Assert.assertFalse(records.get(0).typedGet("mapA").isNull());
    Assert.assertFalse(records.get(0).typedGet("mapB").isNull());
    Assert.assertEquals(((LateralViewBulletRecord) records.get(1)).getBaseRecord().fieldCount(), 5);
    Assert.assertEquals(((LateralViewBulletRecord) records.get(1)).getTopRecord().fieldCount(), 2);
    Assert.assertEquals(records.get(1).typedGet("foo").getValue(), "b");
    Assert.assertEquals(records.get(1).typedGet("bar").getValue(), 1);
    Assert.assertFalse(records.get(1).typedGet("listA").isNull());
    Assert.assertFalse(records.get(1).typedGet("listB").isNull());
    Assert.assertFalse(records.get(1).typedGet("mapA").isNull());
    Assert.assertFalse(records.get(1).typedGet("mapB").isNull());
    Assert.assertEquals(((LateralViewBulletRecord) records.get(2)).getBaseRecord().fieldCount(), 5);
    Assert.assertEquals(((LateralViewBulletRecord) records.get(2)).getTopRecord().fieldCount(), 2);
    Assert.assertEquals(records.get(2).typedGet("foo").getValue(), "c");
    Assert.assertEquals(records.get(2).typedGet("bar").getValue(), 2);
    Assert.assertFalse(records.get(2).typedGet("listA").isNull());
    Assert.assertFalse(records.get(2).typedGet("listB").isNull());
    Assert.assertFalse(records.get(2).typedGet("mapA").isNull());
    Assert.assertFalse(records.get(2).typedGet("mapB").isNull());
    lateralView = new LateralView(new Explode(new FieldExpression("mapB"), "foo", "bar", false));
    functor = new LateralViewFunctor(lateralView);
    records = functor.apply(record, provider);
    Assert.assertEquals(records.size(), 0);
}

Example 15 with FieldExpression

use of com.yahoo.bullet.query.expressions.FieldExpression in project bullet-core by yahoo.

the class LateralViewFunctorTest method testApplyWithOuterExplode.

@Test
public void testApplyWithOuterExplode() {
    LateralView lateralView = new LateralView(new Explode(new FieldExpression("mapB"), "foo", "bar", true));
    LateralViewFunctor functor = new LateralViewFunctor(lateralView);
    List<BulletRecord> records = functor.apply(record, provider);
    Assert.assertEquals(records.size(), 1);
    Assert.assertEquals(((LateralViewBulletRecord) records.get(0)).getBaseRecord().fieldCount(), 5);
    Assert.assertEquals(((LateralViewBulletRecord) records.get(0)).getTopRecord().fieldCount(), 0);
    Assert.assertFalse(records.get(0).typedGet("listA").isNull());
    Assert.assertFalse(records.get(0).typedGet("listB").isNull());
    Assert.assertFalse(records.get(0).typedGet("mapA").isNull());
    Assert.assertFalse(records.get(0).typedGet("mapB").isNull());
}