Server: add update case to real world script

pull/30/head
Chocobozzz 2017-01-11 18:41:28 +01:00
parent 45abb8b97b
commit f2cdb86675
1 changed files with 26 additions and 2 deletions

View File

@ -17,6 +17,7 @@ const videosUtils = require('../utils/videos')
program program
.option('-c, --create [weight]', 'Weight for creating videos') .option('-c, --create [weight]', 'Weight for creating videos')
.option('-r, --remove [weight]', 'Weight for removing videos') .option('-r, --remove [weight]', 'Weight for removing videos')
.option('-u, --update [weight]', 'Weight for updating videos')
.option('-p, --pods [n]', 'Number of pods to run (3 or 6)', /^3|6$/, 3) .option('-p, --pods [n]', 'Number of pods to run (3 or 6)', /^3|6$/, 3)
.option('-a, --action [interval]', 'Interval in ms for an action') .option('-a, --action [interval]', 'Interval in ms for an action')
.option('-i, --integrity [interval]', 'Interval in ms for an integrity check') .option('-i, --integrity [interval]', 'Interval in ms for an integrity check')
@ -26,6 +27,7 @@ program
const createWeight = program.create !== undefined ? parseInt(program.create) : 5 const createWeight = program.create !== undefined ? parseInt(program.create) : 5
const removeWeight = program.remove !== undefined ? parseInt(program.remove) : 4 const removeWeight = program.remove !== undefined ? parseInt(program.remove) : 4
const updateWeight = program.update !== undefined ? parseInt(program.update) : 4
const flushAtExit = program.flush || false const flushAtExit = program.flush || false
const actionInterval = program.action !== undefined ? parseInt(program.action) : 500 const actionInterval = program.action !== undefined ? parseInt(program.action) : 500
let integrityInterval = program.integrity !== undefined ? parseInt(program.integrity) : 60000 let integrityInterval = program.integrity !== undefined ? parseInt(program.integrity) : 60000
@ -39,7 +41,7 @@ const requestsMaxPerInterval = baseRequestInterval / actionInterval
const intervalsToMakeAllRequests = Math.ceil(requestsMaxPerInterval / (constants.REQUESTS_LIMIT_PER_POD * numberOfPods)) const intervalsToMakeAllRequests = Math.ceil(requestsMaxPerInterval / (constants.REQUESTS_LIMIT_PER_POD * numberOfPods))
const waitForBeforeIntegrityCheck = (intervalsToMakeAllRequests * constants.REQUESTS_INTERVAL) + 1000 const waitForBeforeIntegrityCheck = (intervalsToMakeAllRequests * constants.REQUESTS_INTERVAL) + 1000
console.log('Create weight: %d, remove weight: %d.', createWeight, removeWeight) console.log('Create weight: %d, update weight: %d, remove weight: %d.', createWeight, updateWeight, removeWeight)
if (flushAtExit) { if (flushAtExit) {
console.log('Program will flush data on exit.') console.log('Program will flush data on exit.')
} else { } else {
@ -71,10 +73,12 @@ runServers(numberOfPods, function (err, servers) {
setInterval(function () { setInterval(function () {
if (checking === true) return if (checking === true) return
const rand = getRandomInt(0, createWeight + removeWeight) const rand = getRandomInt(0, createWeight + updateWeight + removeWeight)
if (rand < createWeight) { if (rand < createWeight) {
upload(servers, getRandomNumServer(servers)) upload(servers, getRandomNumServer(servers))
} else if (rand < createWeight + updateWeight) {
update(servers, getRandomNumServer(servers))
} else { } else {
remove(servers, getRandomNumServer(servers)) remove(servers, getRandomNumServer(servers))
} }
@ -180,6 +184,26 @@ function upload (servers, numServer, callback) {
videosUtils.uploadVideo(servers[numServer].url, servers[numServer].accessToken, name, description, tags, file, callback) videosUtils.uploadVideo(servers[numServer].url, servers[numServer].accessToken, name, description, tags, file, callback)
} }
function update (servers, numServer, callback) {
if (!callback) callback = function () {}
videosUtils.getVideosList(servers[numServer].url, function (err, res) {
if (err) throw err
const videos = res.body.data.filter(function (video) { return video.isLocal })
if (videos.length === 0) return callback()
const toUpdate = videos[getRandomInt(0, videos.length)].id
const name = Date.now() + ' name'
const description = Date.now() + ' description'
const tags = [ Date.now().toString().substring(0, 5) + 't1', Date.now().toString().substring(0, 5) + 't2' ]
console.log('Updating video of server ' + numServer)
videosUtils.updateVideo(servers[numServer].url, servers[numServer].accessToken, toUpdate, name, description, tags, callback)
})
}
function remove (servers, numServer, callback) { function remove (servers, numServer, callback) {
if (!callback) callback = function () {} if (!callback) callback = function () {}