diff --git a/includes/batch.inc b/includes/batch.inc index 611875154c1..ad563036824 100644 --- a/includes/batch.inc +++ b/includes/batch.inc @@ -11,14 +11,18 @@ function _batch_page() { $batch =& batch_get(); - // Retrieve the current state of batch from db. - if (isset($_REQUEST['id']) && $data = db_result(db_query("SELECT batch FROM {batch} WHERE bid = %d AND token = '%s'", $_REQUEST['id'], drupal_get_token($_REQUEST['id'])))) { - $batch = unserialize($data); - } - else { + if (!isset($_REQUEST['id'])) { return FALSE; } + // Retrieve the current state of batch from db. + $batch = db_query("SELECT batch FROM {batch} WHERE bid = :bid AND token = :token", array( + ':bid' => $_REQUEST['id'], + ':token' => drupal_get_token($_REQUEST['id'])) + )->fetchField(); + + $batch = unserialize($batch); + // Register database update for end of processing. register_shutdown_function('_batch_shutdown'); @@ -314,7 +318,9 @@ function _batch_finished() { // Cleanup the batch table and unset the global $batch variable. if ($batch['progressive']) { - db_query("DELETE FROM {batch} WHERE bid = %d", $batch['id']); + db_delete('batch') + ->condition('bid', $batch['id']) + ->execute(); } $_batch = $batch; $batch = NULL; @@ -358,6 +364,9 @@ function _batch_finished() { */ function _batch_shutdown() { if ($batch = batch_get()) { - db_query("UPDATE {batch} SET batch = '%s' WHERE bid = %d", serialize($batch), $batch['id']); + db_update('batch') + ->fields(array('batch' => serialize($batch))) + ->condition('bid', $batch['id']) + ->execute(); } }