117 lines
3.4 KiB
JavaScript
117 lines
3.4 KiB
JavaScript
'use strict'
|
|
|
|
const { spawn } = require('child_process')
|
|
|
|
const cliSelect = require('cli-select')
|
|
const simpleGit = require('simple-git')
|
|
|
|
const git = simpleGit(process.cwd())
|
|
|
|
const COMMAND = 'npm run bench'
|
|
const DEFAULT_BRANCH = 'master'
|
|
const PERCENT_THRESHOLD = 5
|
|
const greyColor = '\x1b[30m'
|
|
const redColor = '\x1b[31m'
|
|
const greenColor = '\x1b[32m'
|
|
const resetColor = '\x1b[0m'
|
|
|
|
async function selectBranchName (message, branches) {
|
|
console.log(message)
|
|
const result = await cliSelect({
|
|
type: 'list',
|
|
name: 'branch',
|
|
values: branches
|
|
})
|
|
console.log(result.value)
|
|
return result.value
|
|
}
|
|
|
|
async function executeCommandOnBranch (command, branch) {
|
|
console.log(`${greyColor}Checking out "${branch}"${resetColor}`)
|
|
await git.checkout(branch)
|
|
|
|
console.log(`${greyColor}Execute "${command}"${resetColor}`)
|
|
const childProcess = spawn(command, { stdio: 'pipe', shell: true })
|
|
|
|
let result = ''
|
|
childProcess.stdout.on('data', (data) => {
|
|
process.stdout.write(data.toString())
|
|
result += data.toString()
|
|
})
|
|
|
|
await new Promise(resolve => childProcess.on('close', resolve))
|
|
|
|
console.log()
|
|
|
|
return parseBenchmarksStdout(result)
|
|
}
|
|
|
|
function parseBenchmarksStdout (text) {
|
|
const results = []
|
|
|
|
const lines = text.split('\n')
|
|
for (const line of lines) {
|
|
const match = /^(.+?)(\.*) x (.+) ops\/sec .*$/.exec(line)
|
|
if (match !== null) {
|
|
results.push({
|
|
name: match[1],
|
|
alignedName: match[1] + match[2],
|
|
result: parseInt(match[3].split(',').join(''))
|
|
})
|
|
}
|
|
}
|
|
|
|
return results
|
|
}
|
|
|
|
function compareResults (featureBranch, mainBranch) {
|
|
for (const { name, alignedName, result: mainBranchResult } of mainBranch) {
|
|
const featureBranchBenchmark = featureBranch.find(result => result.name === name)
|
|
if (featureBranchBenchmark) {
|
|
const featureBranchResult = featureBranchBenchmark.result
|
|
const percent = (featureBranchResult - mainBranchResult) * 100 / mainBranchResult
|
|
const roundedPercent = Math.round(percent * 100) / 100
|
|
|
|
const percentString = roundedPercent > 0 ? `+${roundedPercent}%` : `${roundedPercent}%`
|
|
const message = alignedName + percentString.padStart(7, '.')
|
|
|
|
if (roundedPercent > PERCENT_THRESHOLD) {
|
|
console.log(`${greenColor}${message}${resetColor}`)
|
|
} else if (roundedPercent < -PERCENT_THRESHOLD) {
|
|
console.log(`${redColor}${message}${resetColor}`)
|
|
} else {
|
|
console.log(message)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
(async function () {
|
|
const branches = await git.branch()
|
|
const currentBranch = branches.branches[branches.current]
|
|
|
|
let featureBranch = null
|
|
let mainBranch = null
|
|
|
|
if (process.argv[2] === '--ci') {
|
|
featureBranch = currentBranch.name
|
|
mainBranch = DEFAULT_BRANCH
|
|
} else {
|
|
featureBranch = await selectBranchName('Select the branch you want to compare (feature branch):', branches.all)
|
|
mainBranch = await selectBranchName('Select the branch you want to compare with (main branch):', branches.all)
|
|
}
|
|
|
|
try {
|
|
const featureBranchResult = await executeCommandOnBranch(COMMAND, featureBranch)
|
|
const mainBranchResult = await executeCommandOnBranch(COMMAND, mainBranch)
|
|
compareResults(featureBranchResult, mainBranchResult)
|
|
} catch (error) {
|
|
console.error('Switch to origin branch due to an error', error.message)
|
|
}
|
|
|
|
await git.checkout(currentBranch.commit)
|
|
await git.checkout(currentBranch.name)
|
|
|
|
console.log(`${greyColor}Back to ${currentBranch.name} ${currentBranch.commit}${resetColor}`)
|
|
})()
|