YAO/YAO-portal
1
0
Fork 0
mirror of https://github.com/newnius/YAO-portal.git synced 2025-12-13 00:56:44 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius
2020-07-01 10:52:31 +08:00
parent 72a5d1d299
commit 1db9445f84
4 changed files with 124 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
job.logic.php
View File

@@ -197,6 +197,54 @@ function job_predict_req(CRObject $job, $role)
return $res;
}
function job_predict_time(CRObject $job, $role)
{
if (!AccessController::hasAccess(Session::get('role', 'visitor'), 'job.list')) {
$res['errno'] = Code::NO_PRIVILEGE;
return $res;
}
$spider = new Spider();
$tasks = json_decode($job->get('tasks'), true);
foreach ($tasks as $i => $task) {
$task['cpu_number'] = intval($task['cpu_number']);
$task['memory'] = intval($task['memory']);
$task['gpu_number'] = intval($task['gpu_number']);
$task['gpu_memory'] = intval($task['gpu_memory']);
$task['is_ps'] = $task['is_ps'] == 1;
$tasks[$i] = $task;
}
$job->set('tasks', $tasks);
$job->set('workspace', $job->get('workspace'));
$job->set('group', $job->get('virtual_cluster'));
$job->set('priority', $job->getInt('priority'));
$job->set('locality', $job->getInt('locality'));
$job->set('run_before', $job->getInt('run_before'));
$job->set('created_by', $job->getInt('created_by'));
$data['job'] = json_encode($job);
$spider->doPost(YAO_SCHEDULER_ADDR . '?action=job_predict_time', $data);
$msg = json_decode($spider->getBody(), true);
if ($msg === NULL) {
$res['errno'] = Code::UNKNOWN_ERROR;
$res['msg'] = 'response is null';
return $res;
}
if ($msg['code'] !== 0) {
$res['errno'] = $msg['code'];
$res['msg'] = $msg['error'];
return $res;
}
$res['total'] = $msg['total'];
$res['pre'] = $msg['pre'];
$res['post'] = $msg['post'];
$res['errno'] = Code::SUCCESS;
return $res;
}
function summary_get()
{
if (!AccessController::hasAccess(Session::get('role', 'visitor'), 'system.summary')) {