bbovenzi commented on code in PR #22900:
URL: https://github.com/apache/airflow/pull/22900#discussion_r850456161
##########
airflow/www/static/js/dags.js:
##########
@@ -351,7 +359,117 @@ function hideSvgTooltip() {
$('#svg-tooltip').css('display', 'none');
}
+function RefreshDagRunsAndTasks(selector, dagId, states) {
+ d3.select(`svg#${selector}-${dagId.replace(/\./g, '__dot__')}`)
+ .selectAll('circle')
+ .data(states)
+ .attr('stroke-width', (d) => {
+ if (d.count > 0) return strokeWidth;
+ return 1;
+ })
+ .attr('stroke', (d) => {
+ if (d.count > 0) return STATE_COLOR[d.state];
+
+ return 'gainsboro';
+ })
+ .attr('fill', '#fff')
+ .attr('r', diameter / 2)
+ .attr('title', (d) => d.state)
+ .on('mouseover', (d) => {
+ if (d.count > 0) {
+ d3.select(this).transition().duration(400)
+ .attr('fill', '#e2e2e2')
+ .style('stroke-width', strokeWidthHover);
+ }
+ });
+ d3.select(`svg#${selector}-${dagId.replace(/\./g, '__dot__')}`)
+ .selectAll('text')
+ .data(states)
+ .text((d) => {
+ if (d.count > 0) {
+ return d.count;
+ }
+ return '';
+ });
+}
+
+function refreshTaskStateHandler(error, ts) {
+ Object.keys(ts).forEach((dagId) => {
+ const states = ts[dagId];
+ RefreshDagRunsAndTasks('task-run', dagId, states);
+ });
+}
+
+let refreshInterval;
+
+function checkActiveRuns(json) {
+ // filter latest dag runs and check if there are still running dags
+ const activeRuns = Object.keys(json).filter((dagId) => {
+ const dagRuns = json[dagId].filter((s) => s.state ===
'running').filter((r) => r.count > 0);
+ return (dagRuns.length > 0);
+ });
+ if (activeRuns.length === 0) {
+ // in case there are no active runs increase the interval for auto refresh
+ $('#auto_refresh').prop('checked', false);
+ clearInterval(refreshInterval);
+ }
+}
+
+function refreshDagRuns(error, json) {
+ checkActiveRuns(json);
+ Object.keys(json).forEach((dagId) => {
+ const states = json[dagId];
+ drawDagStatsForDag(dagId, states);
+ RefreshDagRunsAndTasks('dag-run', dagId, states);
+ });
+}
+
+function handleRefresh() {
+ $('#loading-dots').css('display', 'inline-block');
+ d3.json(lastDagRunsUrl)
+ .header('X-CSRFToken', csrfToken)
+ .post(encodedDagIds, lastDagRunsHandler);
+ d3.json(dagStatsUrl)
+ .header('X-CSRFToken', csrfToken)
+ .post(encodedDagIds, refreshDagRuns);
+ d3.json(taskStatsUrl)
+ .header('X-CSRFToken', csrfToken)
+ .post(encodedDagIds, refreshTaskStateHandler);
+ setTimeout(() => {
+ $('#loading-dots').css('display', 'none');
+ }, refreshIntervalMs);
+}
Review Comment:
```suggestion
}
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
