wehub-resource-sync
149 lines
4.1 KiB
TypeScript
149 lines
4.1 KiB
TypeScript
import * as chalk from 'chalk'
|
|
import * as commander from 'commander'
|
|
|
|
import { asBold, asTimestamp, withUnderline } from 'src/utils/format'
|
|
import { wait } from 'src/utils/wait'
|
|
import { formatEnum, Format, isRunning } from './utils'
|
|
import { Sandbox } from 'e2b'
|
|
import { ensureAPIKey } from '../../api'
|
|
|
|
export const metricsCommand = new commander.Command('metrics')
|
|
.description('show metrics for sandbox')
|
|
.argument(
|
|
'<sandboxID>',
|
|
`show metrics for sandbox specified by ${asBold('<sandboxID>')}`
|
|
)
|
|
.alias('mt')
|
|
.option('-f, --follow', 'keep streaming metrics until the sandbox is closed')
|
|
.option(
|
|
'--format <format>',
|
|
`specify format for printing metrics (${formatEnum(Format)})`,
|
|
Format.PRETTY
|
|
)
|
|
.action(
|
|
async (
|
|
sandboxID: string,
|
|
opts?: {
|
|
follow: boolean
|
|
format: Format
|
|
}
|
|
) => {
|
|
try {
|
|
const format = opts?.format.toLowerCase() as Format | undefined
|
|
if (format && !Object.values(Format).includes(format)) {
|
|
throw new Error(`Invalid log format: ${format}`)
|
|
}
|
|
|
|
let start: Date | undefined
|
|
let isFirstRun = true
|
|
let firstMetricsPrinted = false
|
|
|
|
if (format === Format.PRETTY) {
|
|
console.log(`\nMetrics for sandbox ${asBold(sandboxID)}:`)
|
|
}
|
|
|
|
const apiKey = ensureAPIKey()
|
|
const isRunningPromise = isRunning(sandboxID)
|
|
|
|
do {
|
|
const metrics = await Sandbox.getMetrics(sandboxID, { start, apiKey })
|
|
|
|
if (metrics.length !== 0 && !firstMetricsPrinted) {
|
|
firstMetricsPrinted = true
|
|
process.stdout.write('\n')
|
|
}
|
|
|
|
for (const metric of metrics) {
|
|
if (start && metric.timestamp <= start) {
|
|
// Skip the metric if it has the same timestamp as the last one
|
|
continue
|
|
}
|
|
start = metric.timestamp
|
|
|
|
printMetric(metric.timestamp, JSON.stringify(metric), format)
|
|
}
|
|
|
|
const isRunning = await isRunningPromise
|
|
|
|
if (!isRunning && metrics.length === 0 && isFirstRun) {
|
|
if (format === Format.PRETTY) {
|
|
console.log(
|
|
`\nStopped printing metrics — sandbox ${withUnderline(
|
|
'not found'
|
|
)}`
|
|
)
|
|
}
|
|
break
|
|
}
|
|
|
|
if (!isRunning) {
|
|
if (format === Format.PRETTY) {
|
|
console.log(
|
|
`\nStopped printing metrics — sandbox is ${withUnderline(
|
|
'closed'
|
|
)}`
|
|
)
|
|
}
|
|
break
|
|
}
|
|
|
|
await wait(400)
|
|
isFirstRun = false
|
|
} while (opts?.follow)
|
|
} catch (err: any) {
|
|
console.error(err)
|
|
process.exit(1)
|
|
}
|
|
}
|
|
)
|
|
|
|
function printMetric(
|
|
timestamp: Date,
|
|
line: string,
|
|
format: Format | undefined
|
|
) {
|
|
const metric = JSON.parse(line)
|
|
const level = chalk.default.green()
|
|
|
|
if (format === Format.JSON) {
|
|
console.log(
|
|
JSON.stringify({
|
|
timestamp: timestamp.toISOString(),
|
|
...metric,
|
|
})
|
|
)
|
|
} else {
|
|
const time = `[${timestamp
|
|
.toISOString()
|
|
.replace(/\.\d{3}Z/, 'Z')
|
|
.replace(/T/, ' ')}]`
|
|
delete metric['timestamp']
|
|
const multipleCores = metric.cpuCount > 1
|
|
metric.cpuCount += 0
|
|
console.log(
|
|
`${asTimestamp(time)} ${level} ` +
|
|
asBold('CPU') +
|
|
`: ${metric.cpuUsedPct.toString().padStart(5)}% / ${metric.cpuCount
|
|
.toString()
|
|
.padStart(2)} Core${multipleCores && 's'} | ` +
|
|
asBold('Memory') +
|
|
`: ${toMB(metric.memUsed).toFixed(0).padStart(5)} / ${toMB(
|
|
metric.memTotal
|
|
)
|
|
.toFixed(0)
|
|
.padEnd(5)} MiB | ` +
|
|
asBold('Disk') +
|
|
`: ${toMB(metric.diskUsed).toFixed(0).padStart(5)} / ${toMB(
|
|
metric.diskTotal
|
|
)
|
|
.toFixed(0)
|
|
.padEnd(5)} MiB`
|
|
)
|
|
}
|
|
}
|
|
|
|
// we can't use bite shift here because shift is 32 bit operation and disk sizes can be greater than 2^32
|
|
function toMB(bytes: number): number {
|
|
return bytes / 1024 / 1024
|
|
}
|