YAO/YAO-portal
1
0
Fork 0
mirror of https://github.com/newnius/YAO-portal.git synced 2025-06-06 07:11:54 +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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ajax.php
View File

@ -112,6 +112,18 @@ switch ($action) {
$res = job_predict_req($job, $role);
break;
case 'job_predict_time':
$job = new CRObject();
$job->set('name', cr_get_POST('name', 'jobName'));
$job->set('virtual_cluster', cr_get_POST('cluster'));
$job->set('workspace', cr_get_POST('workspace'));
$job->set('priority', cr_get_POST('priority'));
$job->set('run_before', cr_get_POST('run_before'));
$job->set('locality', cr_get_POST('locality'));
$job->set('tasks', cr_get_POST('tasks'));
$res = job_predict_time($job, $role);
break;
case 'summary_get':
$res = summary_get();
break;

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')) {

3
modals.php
View File

@ -211,7 +211,8 @@
<div>
<button id="form-job-submit" type="submit" class="btn btn-primary btn-lg">Submit</button>
<button id="form-job-task-add" type="button" class="btn btn-default btn-lg">Add Task</button>
<button id="form-job-predict-req" type="button" class="btn btn-default btn-lg">Predict</button>
<button id="form-job-predict-req" type="button" class="btn btn-default">Predict</button>
<button id="form-job-predict-time" type="button" class="btn btn-default">PredictTime</button>
</div>
</form>
</div>

62
static/job.js
View File

@ -107,6 +107,68 @@ function register_events_job() {
});
});
$("#form-job-predict-time").click(function (e) {
var name = $('#form-job-name').val();
var workspace = $('#form-job-workspace').val();
var cluster = $('#form-job-cluster').val();
var priority = $('#form-job-priority').val();
var run_before = $('#form-job-run-before').val();
var locality = $('#form-job-locality').val();
if (run_before.length !== 0) {
run_before = moment(run_before).unix();
}
var tasks = [];
$('#form-job-tasks').find('.row').each(function () {
var task = {};
task['name'] = $(this).find('.task-name').eq(0).val();
task['image'] = $(this).find('.task-image').eq(0).val();
task['cmd'] = $(this).find('.task-cmd').eq(0).val();
task['cpu_number'] = $(this).find('.task-cpu').eq(0).val();
task['memory'] = $(this).find('.task-mem').eq(0).val();
task['gpu_number'] = $(this).find('.task-gpu-num').eq(0).val();
task['gpu_memory'] = $(this).find('.task-gpu-mem').eq(0).val();
task['is_ps'] = $(this).find('.task-is-ps').eq(0).val();
task['gpu_model'] = $(this).find('.task-gpu-model').eq(0).val();
tasks.push(task);
});
/* TODO validate form */
if (name.length === 0) {
return true;
}
$.each(tasks, function (i, task) {
if (task['name'].length === 0) {
return true;
}
});
var ajax = $.ajax({
url: "service?action=job_predict_time",
type: 'POST',
data: {
name: name,
workspace: workspace,
cluster: cluster,
priority: priority,
run_before: run_before,
locality: locality,
tasks: JSON.stringify(tasks)
}
});
ajax.done(function (res) {
if (res["errno"] !== 0) {
$("#modal-msg-content").html(res["msg"]);
$("#modal-msg").modal('show');
} else {
console.log(res);
}
});
ajax.fail(function (jqXHR, textStatus) {
$("#modal-msg-content").html("Request failed : " + jqXHR.statusText);
$("#modal-msg").modal('show');
});
});
$('#form-job-task-add').click(function (e) {
var tasks = $('#form-job-tasks');
var newTask = $('#form-job-tasks').find('.row').eq(0).clone();