import _ from 'lodash'; import d3 from 'd3'; // Inspired by Lee Byron's test data generator. /*eslint-disable */ function bumpLayer(n, maxValue) { function bump(a) { const x = 1 / (0.1 + Math.random()); const y = 2 * Math.random() - 0.5; const z = 10 / (0.1 + Math.random()); for (let i = 0; i < n; i++) { const w = (i / n - y) * z; a[i] += x * Math.exp(-w * w); } } const a = []; let i; for (i = 0; i < n; ++i) a[i] = 0; for (i = 0; i < 5; ++i) bump(a); const values = a.map(function(d) { return Math.max(0, d * maxValue); }); const s = d3.scale.linear().domain(d3.extent(values)).range([0, maxValue]); return values.map(s); } /*eslint-enable */ const nodeData = {}; function getNextValue(keyValues, maxValue) { const key = keyValues.join('-'); let series = nodeData[key]; if (!series || !series.length) { series = nodeData[key] = bumpLayer(100, maxValue); } const v = series.shift(); return v; } export const METRIC_LABELS = { docker_cpu_total_usage: 'CPU', docker_memory_usage: 'Memory', host_cpu_usage_percent: 'CPU', host_mem_usage_bytes: 'Memory', load1: 'Load 1', load15: 'Load 15', load5: 'Load 5', open_files_count: 'Open files', process_cpu_usage_percent: 'CPU', process_memory_usage_bytes: 'Memory' }; export function label(m) { return METRIC_LABELS[m.id]; } const memoryMetric = (node, name, max = 1024 * 1024 * 1024) => ({ samples: [{value: getNextValue([node.id, name], max)}], max }); const cpuMetric = (node, name, max = 100) => ({ samples: [{value: getNextValue([node.id, name], max)}], max }); const fileMetric = (node, name, max = 1000) => ({ samples: [{value: getNextValue([node.id, name], max)}], max }); const loadMetric = (node, name, max = 10) => ({ samples: [{value: getNextValue([node.id, name], max)}], max }); const metrics = { // process square: { process_cpu_usage_percent: cpuMetric, process_memory_usage_bytes: memoryMetric, open_files_count: fileMetric }, // container hexagon: { docker_cpu_total_usage: cpuMetric, docker_memory_usage: memoryMetric }, // host circle: { load5: loadMetric, host_cpu_usage_percent: cpuMetric, host_mem_usage_bytes: memoryMetric } }; function mergeMetrics(node) { if (node.pseudo || node.stack) { return node; } return Object.assign({}, node, { metrics: _(metrics[node.shape]) .map((fn, name) => [name, fn(node)]) .fromPairs() .value() }); } function handleAdd(nodes) { if (!nodes) { return nodes; } return nodes.map(mergeMetrics); } function handleUpdated(updatedNodes, prevNodes) { const modifiedNodesIndex = _.zipObject((updatedNodes || []).map(n => [n.id, n])); return prevNodes.toIndexedSeq().toJS().map(n => ( Object.assign({}, mergeMetrics(n), modifiedNodesIndex[n.id]) )); } export function addMetrics(delta, prevNodes) { return Object.assign({}, delta, { add: handleAdd(delta.add), update: handleUpdated(delta.update, prevNodes) }); }