Examples with Query com.yahoo.bullet.parsing.Query used on opensource projects

Example 21 with Query

use of com.yahoo.bullet.parsing.Query in project bullet-core by yahoo.

the class WindowingOperationsTest method testTumblingWindow.

@Test
public void testTumblingWindow() {
    BulletConfig config = new BulletConfig();
    Query query = new Query();
    Window window = WindowUtils.makeTumblingWindow(1000);
    window.configure(config);
    query.setWindow(window);
    Assert.assertEquals(WindowingOperations.findScheme(query, null, config).getClass(), Tumbling.class);
}

Example 22 with Query

use of com.yahoo.bullet.parsing.Query in project bullet-core by yahoo.

the class WindowingOperationsTest method testNoWindow.

@Test
public void testNoWindow() {
    BulletConfig config = new BulletConfig();
    Query query = new Query();
    Assert.assertEquals(WindowingOperations.findScheme(query, null, config).getClass(), Basic.class);
}

Example 23 with Query

use of com.yahoo.bullet.parsing.Query in project bullet-core by yahoo.

the class WindowingOperationsTest method testNotForcingRawToReactive.

@Test
public void testNotForcingRawToReactive() {
    BulletConfig config = new BulletConfig();
    Query query = new Query();
    Window window = WindowUtils.makeWindow(Window.Unit.TIME, 1000, Window.Unit.ALL, null);
    window.configure(config);
    query.setWindow(window);
    Aggregation aggregation = new Aggregation();
    aggregation.setType(Aggregation.Type.RAW);
    query.setAggregation(aggregation);
    Assert.assertEquals(WindowingOperations.findScheme(query, null, config).getClass(), AdditiveTumbling.class);
}

Example 24 with Query

use of com.yahoo.bullet.parsing.Query in project bullet-core by yahoo.

the class Querier method initialize.

// ********************************* Monoidal Interface Overrides *********************************
/**
 * Starts the query.
 */
@Override
@SuppressWarnings("unchecked")
public Optional<List<BulletError>> initialize() {
    // Is an empty map if metadata was disabled
    metaKeys = (Map<String, String>) config.getAs(BulletConfig.RESULT_METADATA_METRICS, Map.class);
    Boolean shouldInjectTimestamp = config.getAs(BulletConfig.RECORD_INJECT_TIMESTAMP, Boolean.class);
    if (shouldInjectTimestamp) {
        timestampKey = config.getAs(BulletConfig.RECORD_INJECT_TIMESTAMP_KEY, String.class);
    }
    boolean isRateLimitEnabled = config.getAs(BulletConfig.RATE_LIMIT_ENABLE, Boolean.class);
    if (isRateLimitEnabled) {
        int maxEmit = config.getAs(BulletConfig.RATE_LIMIT_MAX_EMIT_COUNT, Integer.class);
        int timeInterval = config.getAs(BulletConfig.RATE_LIMIT_TIME_INTERVAL, Integer.class);
        rateLimit = new RateLimiter(maxEmit, timeInterval);
    }
    Optional<List<BulletError>> errors;
    errors = runningQuery.initialize();
    if (errors.isPresent()) {
        return errors;
    }
    Query query = this.runningQuery.getQuery();
    // Aggregation and Strategy are guaranteed to not be null.
    Strategy strategy = AggregationOperations.findStrategy(query.getAggregation(), config);
    errors = strategy.initialize();
    if (errors.isPresent()) {
        return errors;
    }
    // Scheme is guaranteed to not be null.
    window = WindowingOperations.findScheme(query, strategy, config);
    return window.initialize();
}