Examples with MetricArray org.apache.gobblin.rest.MetricArray used on opensource projects

Example 6 with MetricArray

use of org.apache.gobblin.rest.MetricArray in project incubator-gobblin by apache.

the class DatabaseJobHistoryStoreV101 method addMetricsToTasks.

private void addMetricsToTasks(Connection connection, JobExecutionQuery query, Filter tableFilter, Map<String, Map<String, TaskExecutionInfo>> taskExecutionInfos) throws SQLException {
    if (query.isIncludeTaskMetrics() && taskExecutionInfos.size() > 0) {
        int index = 1;
        String template = String.format(TASK_METRIC_QUERY_STATEMENT_TEMPLATE, getInPredicate(taskExecutionInfos.size()));
        if (tableFilter.isPresent()) {
            template += " AND t." + tableFilter;
        }
        try (PreparedStatement taskMetricQueryStatement = connection.prepareStatement(template)) {
            for (String jobId : taskExecutionInfos.keySet()) {
                taskMetricQueryStatement.setString(index++, jobId);
            }
            if (tableFilter.isPresent()) {
                tableFilter.addParameters(taskMetricQueryStatement, index);
            }
            try (ResultSet taskMetricRs = taskMetricQueryStatement.executeQuery()) {
                while (taskMetricRs.next()) {
                    String jobId = taskMetricRs.getString("job_id");
                    String taskId = taskMetricRs.getString("task_id");
                    TaskExecutionInfo taskExecutionInfo = taskExecutionInfos.get(jobId).get(taskId);
                    MetricArray metricsArray = taskExecutionInfo.getMetrics(GetMode.NULL);
                    if (metricsArray == null) {
                        metricsArray = new MetricArray();
                        taskExecutionInfo.setMetrics(metricsArray);
                    }
                    metricsArray.add(resultSetToMetric(taskMetricRs));
                }
            }
        }
    }
}