diff --git a/application/controllers/jobs.php b/application/controllers/jobs.php index f488e03..ad2f774 100644 --- a/application/controllers/jobs.php +++ b/application/controllers/jobs.php @@ -4,7 +4,7 @@ class Jobs extends CI_Controller { public function getOwn() { $query = $this->db->order_by('progress', 'desc') - ->get_where('jobs', array('started_by' => $this->session->user_data('user_id'))); + ->get_where('jobs', array('started_by' => $this->session->userdata('user_id'))); $count = $query->num_rows(); $jobs = $query->result_array(); @@ -39,7 +39,7 @@ class Jobs extends CI_Controller { public function listResultsNotSeen() { $query = $this->db->order_by('started_at', 'asc') - ->get_where('jobs', array('started_by' => $this->session->user_data('user_id'), 'seen' => '0')); + ->get_where('jobs', array('started_by' => $this->session->userdata('user_id'), 'seen' => '0')); $count = $query->num_rows(); $jobs = $query->result_array(); diff --git a/application/controllers/statusrpc.php b/application/controllers/statusrpc.php new file mode 100644 index 0000000..3696f0b --- /dev/null +++ b/application/controllers/statusrpc.php @@ -0,0 +1,49 @@ +load->model('job'); + + $query = $this->db->get_where('servers', array('secret' => $secret)); + + // if we are in production mode, we do not want to tell the evil hacker + // that he used a wrong secret. That just encourages him. + if($query->num_rows() < 1 && ENVIRONMENT != 'production') { + die("Unauthorized access."); + } + + $this->job->update($job_id, $progress); + } + + /** + * Update the workload of the server. + */ + public function update_workload($secret, $workload) { + $this->load->model('server'); + $query = $this->db->get_where('servers', array('secret' => $secret)); + + // if we are in production mode, we do not want to tell the evil hacker + // that he used a wrong secret. That just encourages him. + if($query->num_rows() < 1 && ENVIRONMENT != 'production') { + die("Unauthorized access."); + } + + $this->server->update_workload($secret, $workload); + } +} \ No newline at end of file diff --git a/application/models/job.php b/application/models/job.php new file mode 100644 index 0000000..de6d45e --- /dev/null +++ b/application/models/job.php @@ -0,0 +1,23 @@ +db->query("UPDATE `jobs` SET `progress`=". + $this->db->escape($progress).$finished_at. " WHERE `id`=". + $this->db->escape($job_id)); + } +} diff --git a/application/models/project.php b/application/models/project.php index fb267e5..d51e3d0 100644 --- a/application/models/project.php +++ b/application/models/project.php @@ -14,7 +14,7 @@ class Project extends CI_Model { * @return array The user's projects. */ public function getOwn() { - $query = $this->db->where(array('owner' => $this->session->user_data('user_id'))) + $query = $this->db->where(array('owner' => $this->session->userdata('user_id'))) ->order_by('lastaccess', 'desc') ->get('projects'); $projects = $query->result_array(); @@ -25,6 +25,7 @@ class Project extends CI_Model { $ownProjects[$i]['id'] = '/projects/own/'.$project['id']; $ownProjects[$i]['cls'] = 'folder'; $ownProjects[$i]['text'] = $project['name']; + $ownProjects[$i]['prjId'] = $project['id']; $ownProjects[$i]['leaf'] = true; $i++; } @@ -38,7 +39,8 @@ class Project extends CI_Model { * @return array The shared projects. */ public function getShared() { - $this->db->select('*')->from('shares')->order_by('lastaccess', 'desc')->where(array('user_id' => $this->session->user_data('user_id'))); + $this->load->library('session'); + $this->db->select('*')->from('shares')->order_by('lastaccess', 'desc')->where(array('user_id' => $this->session->userdata('user_id'))); $this->db->join('projects', 'projects.id = shares.project_id'); $query = $this->db->get(); @@ -50,6 +52,7 @@ class Project extends CI_Model { $sharedProjects[$i]['id'] = '/projects/shared/'.$project['id']; $sharedProjects[$i]['cls'] = 'folder'; $sharedProjects[$i]['text'] = $project['name']; + $sharedProjects[$i]['prjId'] = $project['id']; $sharedProjects[$i]['leaf'] = true; $i++; } @@ -74,6 +77,7 @@ class Project extends CI_Model { $publicProjects[$i]['id'] = '/projects/public/'.$project['id']; $publicProjects[$i]['cls'] = 'folder'; $publicProjects[$i]['text'] = $project['name']; + $publicProjects[$i]['prjId'] = $project['id']; $publicProjects[$i]['leaf'] = true; $i++; } diff --git a/application/models/server.php b/application/models/server.php new file mode 100644 index 0000000..b4d8570 --- /dev/null +++ b/application/models/server.php @@ -0,0 +1,40 @@ +db->get('servers')->result_array(); + } + + /** + * Get a list of servers that could handle another job. + * + * @return array List of servers that could handle another job. + */ + public function get_idle() { + return $this->db->get_where('servers', 'workload <= 2')->result_array(); + } + + /** + * Set a server's workload. + * + * In order to check if a server can handle another job we need to know + * the workload of every server. + * + * @param type $secret The server's secret for basic authentication. + * @param type $workload The server's workload. + */ + public function update_workload($secret, $workload) { + $this->db->query("UPDATE `servers` SET `workload`=".$this->db->escape($workload) + . ", `last_update`=NOW()" + . " WHERE `secret`=".$this->db->escape($secret)); + } +}