144 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			144 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| namespace W3TC;
 | |
| 
 | |
| /**
 | |
|  * Purge using AmazonSNS object
 | |
|  */
 | |
| class Enterprise_SnsServer extends Enterprise_SnsBase {
 | |
| 
 | |
| 	/**
 | |
| 	 * Processes message from SNS
 | |
| 	 *
 | |
| 	 * @throws Exception
 | |
| 	 */
 | |
| 	function process_message( $message ) {
 | |
| 		$this->_log( 'Received message' );
 | |
| 
 | |
| 		try {
 | |
| 			$message = new \Aws\Sns\Message( $message );
 | |
| 			$validator = new \Aws\Sns\MessageValidator();
 | |
| 			$error = '';
 | |
| 			if ( $validator->isValid( $message ) ) {
 | |
| 				$topic_arn = $this->_config->get_string( 'cluster.messagebus.sns.topic_arn' );
 | |
| 
 | |
| 				if ( empty( $topic_arn ) || $topic_arn != $message['TopicArn'] )
 | |
| 					throw new \Exception( 'Not my Topic. Request came from ' .
 | |
| 						$message['TopicArn'] );
 | |
| 
 | |
| 				if ( $message['Type'] == 'SubscriptionConfirmation' )
 | |
| 					$this->_subscription_confirmation( $message );
 | |
| 				elseif ( $message['Type'] == 'Notification' )
 | |
| 					$this->_notification( $message['Message'] );
 | |
| 			} else {
 | |
| 				$this->_log( 'Error processing message it was not valid.' );
 | |
| 			}
 | |
| 		} catch ( \Exception $e ) {
 | |
| 			$this->_log( 'Error processing message: ' . $e->getMessage() );
 | |
| 		}
 | |
| 		$this->_log( 'Message processed' );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Confirms subscription
 | |
| 	 *
 | |
| 	 * @param Message $message
 | |
| 	 * @throws Exception
 | |
| 	 */
 | |
| 	private function _subscription_confirmation( $message ) {
 | |
| 		$this->_log( 'Issuing confirm_subscription' );
 | |
| 		$topic_arn = $this->_config->get_string( 'cluster.messagebus.sns.topic_arn' );
 | |
| 
 | |
| 		$response = $this->_get_api()->confirmSubscription( array(
 | |
| 			'Token' => $message['Token'],
 | |
| 			'TopicArn' => $topic_arn
 | |
| 		) );
 | |
| 		$this->_log( 'Subscription confirmed: ' .
 | |
| 			( $response['@metadata']['statusCode'] == 200 ? 'OK' : 'Error' ) );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Processes notification
 | |
| 	 *
 | |
| 	 * @param array   $v
 | |
| 	 */
 | |
| 	private function _notification( $v ) {
 | |
| 		$m = json_decode( $v, true );
 | |
| 		if ( isset( $m['hostname'] ) )
 | |
| 			$this->_log( 'Message originated from hostname: ' . $m['hostname'] );
 | |
| 
 | |
| 		define( 'DOING_SNS', true );
 | |
| 		$this->_log( 'Actions executing' );
 | |
| 		do_action( 'w3tc_messagebus_message_received' );
 | |
| 
 | |
| 		if ( isset( $m['actions'] ) ) {
 | |
| 			$actions = $m['actions'];
 | |
| 			foreach ( $actions as $action )
 | |
| 				$this->_execute( $action );
 | |
| 		} else {
 | |
| 			$this->_execute( $m['action'] );
 | |
| 		}
 | |
| 
 | |
| 		do_action( 'w3tc_messagebus_message_processed' );
 | |
| 		$this->_log( 'Actions executed' );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Execute action
 | |
| 	 *
 | |
| 	 * @param unknown $m
 | |
| 	 * @throws Exception
 | |
| 	 */
 | |
| 	private function _execute( $m ) {
 | |
| 		$action = $m['action'];
 | |
| 		$this->_log( 'Executing action ' . $action );
 | |
| 		//Needed for cache flushing
 | |
| 		$executor = new CacheFlush_Locally();
 | |
| 		//Needed for cache cleanup
 | |
| 		$pgcache_admin = Dispatcher::component( 'PgCache_Plugin_Admin' );
 | |
| 
 | |
| 		//See which message we got
 | |
| 		if ( $action == 'dbcache_flush' )
 | |
| 			$executor->dbcache_flush();
 | |
| 		elseif ( $action == 'objectcache_flush' )
 | |
| 			$executor->objectcache_flush();
 | |
| 		elseif ( $action == 'fragmentcache_flush' )
 | |
| 			$executor->fragmentcache_flush();
 | |
| 		elseif ( $action == 'fragmentcache_flush_group' )
 | |
| 			$executor->fragmentcache_flush_group( $m['group'] );
 | |
| 		elseif ( $action == 'minifycache_flush' )
 | |
| 			$executor->minifycache_flush();
 | |
| 		elseif ( $action == 'browsercache_flush' )
 | |
| 			$executor->browsercache_flush();
 | |
| 		elseif ( $action == 'cdn_purge_all' )
 | |
| 			$executor->cdn_purge_all(
 | |
| 				isset( $m['extras'] ) ? $m['extras'] : null );
 | |
| 		elseif ( $action == 'cdn_purge_files' )
 | |
| 			$executor->cdn_purge_files( $m['purgefiles'] );
 | |
| 		elseif ( $action == 'pgcache_cleanup' )
 | |
| 			$pgcache_admin->cleanup_local();
 | |
| 		elseif ( $action == 'opcache_flush' )
 | |
| 			$executor->opcache_flush();
 | |
| 		elseif ( $action == 'flush_all' )
 | |
| 			$executor->flush_all(
 | |
| 				isset( $m['extras'] ) ? $m['extras'] : null );
 | |
| 		elseif ( $action == 'flush_group' )
 | |
| 			$executor->flush_group(
 | |
| 				isset( $m['group'] ) ? $m['group'] : null,
 | |
| 				isset( $m['extras'] ) ? $m['extras'] : null );
 | |
| 		elseif ( $action == 'flush_post' )
 | |
| 			$executor->flush_post( $m['post_id'] );
 | |
| 		elseif ( $action == 'flush_posts' )
 | |
| 			$executor->flush_posts();
 | |
| 		elseif ( $action == 'flush_url' )
 | |
| 			$executor->flush_url( $m['url'] );
 | |
| 		elseif ( $action == 'prime_post' )
 | |
| 			$executor->prime_post( $m['post_id'] );
 | |
| 		else
 | |
| 			throw new \Exception( 'Unknown action ' . $action );
 | |
| 
 | |
| 		$executor->execute_delayed_operations();
 | |
| 
 | |
| 		$this->_log( 'succeeded' );
 | |
| 	}
 | |
| }
 |