Examples with TimelineReaderContext org.apache.hadoop.yarn.server.timelineservice.reader.TimelineReaderContext used on opensource projects

Example 61 with TimelineReaderContext

use of org.apache.hadoop.yarn.server.timelineservice.reader.TimelineReaderContext in project hadoop by apache.

the class ApplicationEntityReader method augmentParams.

@Override
protected void augmentParams(Configuration hbaseConf, Connection conn) throws IOException {
    TimelineReaderContext context = getContext();
    if (isSingleEntityRead()) {
        // Get flow context information from AppToFlow table.
        if (context.getFlowName() == null || context.getFlowRunId() == null || context.getUserId() == null) {
            AppToFlowRowKey appToFlowRowKey = new AppToFlowRowKey(context.getClusterId(), context.getAppId());
            FlowContext flowContext = lookupFlowContext(appToFlowRowKey, hbaseConf, conn);
            context.setFlowName(flowContext.getFlowName());
            context.setFlowRunId(flowContext.getFlowRunId());
            context.setUserId(flowContext.getUserId());
        }
    }
    // Add configs/metrics to fields to retrieve if confsToRetrieve and/or
    // metricsToRetrieve are specified.
    getDataToRetrieve().addFieldsBasedOnConfsAndMetricsToRetrieve();
    if (!isSingleEntityRead()) {
        createFiltersIfNull();
    }
}

Example 62 with TimelineReaderContext

use of org.apache.hadoop.yarn.server.timelineservice.reader.TimelineReaderContext in project hadoop by apache.

the class FlowRunEntityReader method getResults.

@Override
protected ResultScanner getResults(Configuration hbaseConf, Connection conn, FilterList filterList) throws IOException {
    Scan scan = new Scan();
    TimelineReaderContext context = getContext();
    RowKeyPrefix<FlowRunRowKey> flowRunRowKeyPrefix = new FlowRunRowKeyPrefix(context.getClusterId(), context.getUserId(), context.getFlowName());
    scan.setRowPrefixFilter(flowRunRowKeyPrefix.getRowKeyPrefix());
    FilterList newList = new FilterList();
    newList.addFilter(new PageFilter(getFilters().getLimit()));
    if (filterList != null && !filterList.getFilters().isEmpty()) {
        newList.addFilter(filterList);
    }
    scan.setFilter(newList);
    scan.setMaxVersions(Integer.MAX_VALUE);
    return getTable().getResultScanner(hbaseConf, conn, scan);
}

Example 63 with TimelineReaderContext

use of org.apache.hadoop.yarn.server.timelineservice.reader.TimelineReaderContext in project hadoop by apache.

the class GenericEntityReader method augmentParams.

@Override
protected void augmentParams(Configuration hbaseConf, Connection conn) throws IOException {
    TimelineReaderContext context = getContext();
    // In reality all three should be null or neither should be null
    if (context.getFlowName() == null || context.getFlowRunId() == null || context.getUserId() == null) {
        // Get flow context information from AppToFlow table.
        AppToFlowRowKey appToFlowRowKey = new AppToFlowRowKey(context.getClusterId(), context.getAppId());
        FlowContext flowContext = lookupFlowContext(appToFlowRowKey, hbaseConf, conn);
        context.setFlowName(flowContext.flowName);
        context.setFlowRunId(flowContext.flowRunId);
        context.setUserId(flowContext.userId);
    }
    // Add configs/metrics to fields to retrieve if confsToRetrieve and/or
    // metricsToRetrieve are specified.
    getDataToRetrieve().addFieldsBasedOnConfsAndMetricsToRetrieve();
    if (!isSingleEntityRead()) {
        createFiltersIfNull();
    }
}