diff options
Diffstat (limited to 'plugins/jetpack/sync/class.jetpack-sync-module-full-sync.php')
-rw-r--r-- | plugins/jetpack/sync/class.jetpack-sync-module-full-sync.php | 477 |
1 files changed, 477 insertions, 0 deletions
diff --git a/plugins/jetpack/sync/class.jetpack-sync-module-full-sync.php b/plugins/jetpack/sync/class.jetpack-sync-module-full-sync.php new file mode 100644 index 00000000..4ad60896 --- /dev/null +++ b/plugins/jetpack/sync/class.jetpack-sync-module-full-sync.php @@ -0,0 +1,477 @@ +<?php + +/** + * This class does a full resync of the database by + * enqueuing an outbound action for every single object + * that we care about. + * + * This class, and its related class Jetpack_Sync_Module, contain a few non-obvious optimisations that should be explained: + * - we fire an action called jetpack_full_sync_start so that WPCOM can erase the contents of the cached database + * - for each object type, we page through the object IDs and enqueue them by firing some monitored actions + * - we load the full objects for those IDs in chunks of Jetpack_Sync_Module::ARRAY_CHUNK_SIZE (to reduce the number of MySQL calls) + * - we fire a trigger for the entire array which the Jetpack_Sync_Listener then serializes and queues. + */ + +class Jetpack_Sync_Module_Full_Sync extends Jetpack_Sync_Module { + const STATUS_OPTION_PREFIX = 'jetpack_sync_full_'; + const FULL_SYNC_TIMEOUT = 3600; + + public function name() { + return 'full-sync'; + } + + function init_full_sync_listeners( $callable ) { + // synthetic actions for full sync + add_action( 'jetpack_full_sync_start', $callable, 10, 2 ); + add_action( 'jetpack_full_sync_end', $callable, 10, 2 ); + add_action( 'jetpack_full_sync_cancelled', $callable ); + } + + function init_before_send() { + // this is triggered after actions have been processed on the server + add_action( 'jetpack_sync_processed_actions', array( $this, 'update_sent_progress_action' ) ); + } + + function start( $module_configs = null ) { + $was_already_running = $this->is_started() && ! $this->is_finished(); + + // remove all evidence of previous full sync items and status + $this->reset_data(); + + if ( $was_already_running ) { + /** + * Fires when a full sync is cancelled. + * + * @since 4.2.0 + */ + do_action( 'jetpack_full_sync_cancelled' ); + } + + $this->update_status_option( 'started', time() ); + $this->update_status_option( 'params', $module_configs ); + + $enqueue_status = array(); + $full_sync_config = array(); + + // default value is full sync + if ( ! is_array( $module_configs ) ) { + $module_configs = array(); + foreach ( Jetpack_Sync_Modules::get_modules() as $module ) { + $module_configs[ $module->name() ] = true; + } + } + + // set default configuration, calculate totals, and save configuration if totals > 0 + foreach ( Jetpack_Sync_Modules::get_modules() as $module ) { + $module_name = $module->name(); + $module_config = isset( $module_configs[ $module_name ] ) ? $module_configs[ $module_name ] : false; + + if ( ! $module_config ) { + continue; + } + + if ( 'users' === $module_name && 'initial' === $module_config ) { + $module_config = $module->get_initial_sync_user_config(); + } + + $enqueue_status[ $module_name ] = false; + + $total_items = $module->estimate_full_sync_actions( $module_config ); + + // if there's information to process, configure this module + if ( ! is_null( $total_items ) && $total_items > 0 ) { + $full_sync_config[ $module_name ] = $module_config; + $enqueue_status[ $module_name ] = array( + $total_items, // total + 0, // queued + false, // current state + ); + } + } + + $this->set_config( $full_sync_config ); + $this->set_enqueue_status( $enqueue_status ); + + $range = $this->get_content_range( $full_sync_config ); + /** + * Fires when a full sync begins. This action is serialized + * and sent to the server so that it knows a full sync is coming. + * + * @since 4.2.0 + * @since 7.3.0 Added $range arg. + * + * @param $full_sync_config - array + * @param $range array + */ + do_action( 'jetpack_full_sync_start', $full_sync_config, $range ); + + $this->continue_enqueuing( $full_sync_config, $enqueue_status ); + + return true; + } + + function continue_enqueuing( $configs = null, $enqueue_status = null ) { + if ( ! $this->is_started() || $this->get_status_option( 'queue_finished' ) ) { + return; + } + + // if full sync queue is full, don't enqueue more items + $max_queue_size_full_sync = Jetpack_Sync_Settings::get_setting( 'max_queue_size_full_sync' ); + $full_sync_queue = new Jetpack_Sync_Queue( 'full_sync' ); + + $available_queue_slots = $max_queue_size_full_sync - $full_sync_queue->size(); + + if ( $available_queue_slots <= 0 ) { + return; + } else { + $remaining_items_to_enqueue = min( Jetpack_Sync_Settings::get_setting( 'max_enqueue_full_sync' ), $available_queue_slots ); + } + + if ( ! $configs ) { + $configs = $this->get_config(); + } + + if ( ! $enqueue_status ) { + $enqueue_status = $this->get_enqueue_status(); + } + + foreach ( Jetpack_Sync_Modules::get_modules() as $module ) { + $module_name = $module->name(); + + // skip module if not configured for this sync or module is done + if ( ! isset( $configs[ $module_name ] ) + || // no module config + ! $configs[ $module_name ] + || // no enqueue status + ! $enqueue_status[ $module_name ] + || // finished enqueuing this module + true === $enqueue_status[ $module_name ][2] ) { + continue; + } + + list( $items_enqueued, $next_enqueue_state ) = $module->enqueue_full_sync_actions( $configs[ $module_name ], $remaining_items_to_enqueue, $enqueue_status[ $module_name ][2] ); + + $enqueue_status[ $module_name ][2] = $next_enqueue_state; + + // if items were processed, subtract them from the limit + if ( ! is_null( $items_enqueued ) && $items_enqueued > 0 ) { + $enqueue_status[ $module_name ][1] += $items_enqueued; + $remaining_items_to_enqueue -= $items_enqueued; + } + + // stop processing if we've reached our limit of items to enqueue + if ( 0 >= $remaining_items_to_enqueue ) { + $this->set_enqueue_status( $enqueue_status ); + return; + } + } + + $this->set_enqueue_status( $enqueue_status ); + + // setting autoload to true means that it's faster to check whether we should continue enqueuing + $this->update_status_option( 'queue_finished', time(), true ); + + $range = $this->get_content_range( $configs ); + + /** + * Fires when a full sync ends. This action is serialized + * and sent to the server. + * + * @since 4.2.0 + * @since 7.3.0 Added $range arg. + * + * @param args '' + * @param $range array + */ + do_action( 'jetpack_full_sync_end', '', $range ); + } + + function get_range( $type ) { + global $wpdb; + if ( ! in_array( $type, array( 'comments', 'posts' ) ) ) { + return array(); + } + + switch ( $type ) { + case 'posts': + $table = $wpdb->posts; + $id = 'ID'; + $where_sql = Jetpack_Sync_Settings::get_blacklisted_post_types_sql(); + + break; + case 'comments': + $table = $wpdb->comments; + $id = 'comment_ID'; + $where_sql = Jetpack_Sync_Settings::get_comments_filter_sql(); + break; + } + $results = $wpdb->get_results( "SELECT MAX({$id}) as max, MIN({$id}) as min, COUNT({$id}) as count FROM {$table} WHERE {$where_sql}" ); + if ( isset( $results[0] ) ) { + return $results[0]; + } + + return array(); + } + + private function get_content_range( $config ) { + $range = array(); + // Only when we are sending the whole range do we want to send also the range + if ( isset( $config['posts'] ) && $config['posts'] === true ) { + $range['posts'] = $this->get_range( 'posts' ); + } + + if ( isset( $config['comments'] ) && $config['comments'] === true ) { + $range['comments'] = $this->get_range( 'comments' ); + } + return $range; + } + + function update_sent_progress_action( $actions ) { + // quick way to map to first items with an array of arrays + $actions_with_counts = array_count_values( array_filter( array_map( array( $this, 'get_action_name' ), $actions ) ) ); + + // Total item counts for each action. + $actions_with_total_counts = $this->get_actions_totals( $actions ); + + if ( ! $this->is_started() || $this->is_finished() ) { + return; + } + + if ( isset( $actions_with_counts['jetpack_full_sync_start'] ) ) { + $this->update_status_option( 'send_started', time() ); + } + + foreach ( Jetpack_Sync_Modules::get_modules() as $module ) { + $module_actions = $module->get_full_sync_actions(); + $status_option_name = "{$module->name()}_sent"; + $total_option_name = "{$status_option_name}_total"; + $items_sent = $this->get_status_option( $status_option_name, 0 ); + $items_sent_total = $this->get_status_option( $total_option_name, 0 ); + + foreach ( $module_actions as $module_action ) { + if ( isset( $actions_with_counts[ $module_action ] ) ) { + $items_sent += $actions_with_counts[ $module_action ]; + } + + if ( ! empty( $actions_with_total_counts[ $module_action ] ) ) { + $items_sent_total += $actions_with_total_counts[ $module_action ]; + } + } + + if ( $items_sent > 0 ) { + $this->update_status_option( $status_option_name, $items_sent ); + } + + if ( 0 !== $items_sent_total ) { + $this->update_status_option( $total_option_name, $items_sent_total ); + } + } + + if ( isset( $actions_with_counts['jetpack_full_sync_end'] ) ) { + $this->update_status_option( 'finished', time() ); + } + } + + public function get_action_name( $queue_item ) { + if ( is_array( $queue_item ) && isset( $queue_item[0] ) ) { + return $queue_item[0]; + } + return false; + } + + /** + * Retrieve the total number of items we're syncing in a particular queue item (action). + * `$queue_item[1]` is expected to contain chunks of items, and `$queue_item[1][0]` + * represents the first (and only) chunk of items to sync in that action. + * + * @param array $queue_item Item of the sync queue that corresponds to a particular action. + * @return int Total number of items in the action. + */ + public function get_action_totals( $queue_item ) { + if ( is_array( $queue_item ) && isset( $queue_item[1][0] ) ) { + if ( is_array( $queue_item[1][0] ) ) { + // Let's count the items we sync in this action. + return count( $queue_item[1][0] ); + } + // -1 indicates that this action syncs all items by design. + return -1; + } + return 0; + } + + /** + * Retrieve the total number of items for a set of actions, grouped by action name. + * + * @param array $actions An array of actions. + * @return array An array, representing the total number of items, grouped per action. + */ + public function get_actions_totals( $actions ) { + $totals = array(); + + foreach ( $actions as $action ) { + $name = $this->get_action_name( $action ); + $action_totals = $this->get_action_totals( $action ); + if ( ! isset( $totals[ $name ] ) ) { + $totals[ $name ] = 0; + } + $totals[ $name ] += $action_totals; + } + + return $totals; + } + + public function is_started() { + return ! ! $this->get_status_option( 'started' ); + } + + public function is_finished() { + return ! ! $this->get_status_option( 'finished' ); + } + + public function get_status() { + $status = array( + 'started' => $this->get_status_option( 'started' ), + 'queue_finished' => $this->get_status_option( 'queue_finished' ), + 'send_started' => $this->get_status_option( 'send_started' ), + 'finished' => $this->get_status_option( 'finished' ), + 'sent' => array(), + 'sent_total' => array(), + 'queue' => array(), + 'config' => $this->get_status_option( 'params' ), + 'total' => array(), + ); + + $enqueue_status = $this->get_enqueue_status(); + + foreach ( Jetpack_Sync_Modules::get_modules() as $module ) { + $name = $module->name(); + + if ( ! isset( $enqueue_status[ $name ] ) ) { + continue; + } + + list( $total, $queued, $state ) = $enqueue_status[ $name ]; + + if ( $total ) { + $status['total'][ $name ] = $total; + } + + if ( $queued ) { + $status['queue'][ $name ] = $queued; + } + + if ( $sent = $this->get_status_option( "{$name}_sent" ) ) { + $status['sent'][ $name ] = $sent; + } + + $sent_total = $this->get_status_option( "{$name}_sent_total" ); + if ( $sent_total ) { + $status['sent_total'][ $name ] = $sent_total; + } + } + + return $status; + } + + public function clear_status() { + $prefix = self::STATUS_OPTION_PREFIX; + Jetpack_Options::delete_raw_option( "{$prefix}_started" ); + Jetpack_Options::delete_raw_option( "{$prefix}_params" ); + Jetpack_Options::delete_raw_option( "{$prefix}_queue_finished" ); + Jetpack_Options::delete_raw_option( "{$prefix}_send_started" ); + Jetpack_Options::delete_raw_option( "{$prefix}_finished" ); + + $this->delete_enqueue_status(); + + foreach ( Jetpack_Sync_Modules::get_modules() as $module ) { + Jetpack_Options::delete_raw_option( "{$prefix}_{$module->name()}_sent" ); + Jetpack_Options::delete_raw_option( "{$prefix}_{$module->name()}_sent_total" ); + } + } + + public function reset_data() { + $this->clear_status(); + $this->delete_config(); + require_once dirname( __FILE__ ) . '/class.jetpack-sync-listener.php'; + $listener = Jetpack_Sync_Listener::get_instance(); + $listener->get_full_sync_queue()->reset(); + } + + private function get_status_option( $name, $default = null ) { + $value = Jetpack_Options::get_raw_option( self::STATUS_OPTION_PREFIX . "_$name", $default ); + + return is_numeric( $value ) ? intval( $value ) : $value; + } + + private function update_status_option( $name, $value, $autoload = false ) { + Jetpack_Options::update_raw_option( self::STATUS_OPTION_PREFIX . "_$name", $value, $autoload ); + } + + private function set_enqueue_status( $new_status ) { + Jetpack_Options::update_raw_option( 'jetpack_sync_full_enqueue_status', $new_status ); + } + + private function delete_enqueue_status() { + return Jetpack_Options::delete_raw_option( 'jetpack_sync_full_enqueue_status' ); + } + + private function get_enqueue_status() { + return Jetpack_Options::get_raw_option( 'jetpack_sync_full_enqueue_status' ); + } + + private function set_config( $config ) { + Jetpack_Options::update_raw_option( 'jetpack_sync_full_config', $config ); + } + + private function delete_config() { + return Jetpack_Options::delete_raw_option( 'jetpack_sync_full_config' ); + } + + private function get_config() { + return Jetpack_Options::get_raw_option( 'jetpack_sync_full_config' ); + } + + private function write_option( $name, $value ) { + // we write our own option updating code to bypass filters/caching/etc on set_option/get_option + global $wpdb; + $serialized_value = maybe_serialize( $value ); + // try updating, if no update then insert + // TODO: try to deal with the fact that unchanged values can return updated_num = 0 + // below we used "insert ignore" to at least suppress the resulting error + $updated_num = $wpdb->query( + $wpdb->prepare( + "UPDATE $wpdb->options SET option_value = %s WHERE option_name = %s", + $serialized_value, + $name + ) + ); + + if ( ! $updated_num ) { + $updated_num = $wpdb->query( + $wpdb->prepare( + "INSERT IGNORE INTO $wpdb->options ( option_name, option_value, autoload ) VALUES ( %s, %s, 'no' )", + $name, + $serialized_value + ) + ); + } + return $updated_num; + } + + private function read_option( $name, $default = null ) { + global $wpdb; + $value = $wpdb->get_var( + $wpdb->prepare( + "SELECT option_value FROM $wpdb->options WHERE option_name = %s LIMIT 1", + $name + ) + ); + $value = maybe_unserialize( $value ); + + if ( $value === null && $default !== null ) { + return $default; + } + + return $value; + } +} |