"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.gatherV1CgroupMetrics = gatherV1CgroupMetrics;
var _promises = _interopRequireDefault(require("fs/promises"));
var _path = require("path");
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */

const CPU_STATS_FILE = 'cpu.stat';
const CPUACCT_USAGE_FILE = 'cpuacct.usage';
const CPU_FS_QUOTA_US_FILE = 'cpu.cfs_quota_us';
const PROC_CGROUP_CPU_DIR = '/sys/fs/cgroup/cpu';
const CPU_FS_PERIOD_US_FILE = 'cpu.cfs_period_us';
const PROC_CGROUP_CPUACCT_DIR = '/sys/fs/cgroup/cpuacct';
async function gatherV1CgroupMetrics({
  cpuAcctPath,
  cpuPath
}) {
  const [cpuAcctUsage, cpuFsPeriod, cpuFsQuota, cpuStat] = await Promise.all([readCPUAcctUsage(cpuAcctPath), readCPUFsPeriod(cpuPath), readCPUFsQuota(cpuPath), readCPUStat(cpuPath)]);
  return {
    cpuacct: {
      control_group: cpuAcctPath,
      usage_nanos: cpuAcctUsage
    },
    cpu: {
      control_group: cpuPath,
      cfs_period_micros: cpuFsPeriod,
      cfs_quota_micros: cpuFsQuota,
      stat: cpuStat
    }
  };
}
async function fileContentsToInteger(path) {
  const data = await _promises.default.readFile(path);
  return parseInt(data.toString(), 10);
}
function readCPUAcctUsage(controlGroup) {
  return fileContentsToInteger((0, _path.join)(PROC_CGROUP_CPUACCT_DIR, controlGroup, CPUACCT_USAGE_FILE));
}
function readCPUFsPeriod(controlGroup) {
  return fileContentsToInteger((0, _path.join)(PROC_CGROUP_CPU_DIR, controlGroup, CPU_FS_PERIOD_US_FILE));
}
function readCPUFsQuota(controlGroup) {
  return fileContentsToInteger((0, _path.join)(PROC_CGROUP_CPU_DIR, controlGroup, CPU_FS_QUOTA_US_FILE));
}
async function readCPUStat(controlGroup) {
  const stat = {
    number_of_elapsed_periods: -1,
    number_of_times_throttled: -1,
    time_throttled_nanos: -1
  };
  try {
    const data = await _promises.default.readFile((0, _path.join)(PROC_CGROUP_CPU_DIR, controlGroup, CPU_STATS_FILE));
    return data.toString().split(/\n/).reduce((acc, line) => {
      const [key, value] = line.split(/\s+/);
      switch (key) {
        case 'nr_periods':
          acc.number_of_elapsed_periods = parseInt(value, 10);
          break;
        case 'nr_throttled':
          acc.number_of_times_throttled = parseInt(value, 10);
          break;
        case 'throttled_time':
          acc.time_throttled_nanos = parseInt(value, 10);
          break;
      }
      return acc;
    }, stat);
  } catch (err) {
    if (err.code === 'ENOENT') {
      return stat;
    }
    throw err;
  }
}