GH-79: Implement wp_cache_(add|delete|set)_multiple (#87)

* Implement `wp_cache_(add|delete|set)_multiple`

* Add tests

Tests were taken from https://github.com/WordPress/wordpress-develop/blob/5d7297d/tests/phpunit/tests/cache.php

Author: sjinks

object-cache.php

@@ -22,6 +22,12 @@ function wp_cache_add( $key, $data, $group = '', $expire = 0 ) {
 	return $wp_object_cache->add( $key, $data, $group, $expire );
 }
 
+function wp_cache_add_multiple( array $data, $group = '', $expire = 0 ) {
+	global $wp_object_cache;
+
+	return $wp_object_cache->add_multiple( $data, $group, $expire );
+}
+
 function wp_cache_incr( $key, $n = 1, $group = '' ) {
 	global $wp_object_cache;
 
@@ -46,6 +52,12 @@ function wp_cache_delete( $key, $group = '' ) {
 	return $wp_object_cache->delete( $key, $group );
 }
 
+function wp_cache_delete_multiple( array $keys, $group = '' ) {
+	global $wp_object_cache;
+
+	return $wp_object_cache->delete_multiple( $keys, $group );
+}
+
 function wp_cache_flush() {
 	global $wp_object_cache;
 
@@ -98,6 +110,12 @@ function wp_cache_set( $key, $data, $group = '', $expire = 0 ) {
 	}
 }
 
+function wp_cache_set_multiple( array $data, $group = '', $expire = 0 ) {
+	global $wp_object_cache;
+
+	return $wp_object_cache->set_multiple( $data, $group, $expire );
+}
+
 function wp_cache_switch_to_blog( $blog_id ) {
 	global $wp_object_cache;
 
@@ -210,6 +228,16 @@ function add( $id, $data, $group = 'default', $expire = 0 ) {
 		return $result;
 	}
 
+	public function add_multiple( array $data, $group = '', $expire = 0 ) {
+		$values = array();
+
+		foreach ( $data as $key => $value ) {
+			$values[ $key ] = $this->add( $key, $value, $group, $expire );
+		}
+
+		return $values;
+	}
+
 	function add_global_groups( $groups ) {
 		if ( ! is_array( $groups ) ) {
 			$groups = (array) $groups;
@@ -285,6 +313,16 @@ function delete( $id, $group = 'default' ) {
 		return $result;
 	}
 
+	public function delete_multiple( array $keys, $group = '' ) {
+		$values = array();
+
+		foreach ( $keys as $key ) {
+			$values[ $key ] = $this->delete( $key, $group );
+		}
+
+		return $values;
+	}
+
 	// Gets number from all default servers, replicating if needed
 	function get_max_flush_number( $group ) {
 		$key = $this->key( $this->flush_key, $group );
@@ -633,6 +671,16 @@ function set( $id, $data, $group = 'default', $expire = 0 ) {
 		return $result;
 	}
 
+	public function set_multiple( array $data, $group = '', $expire = 0 ) {
+		$values = array();
+
+		foreach ( $data as $key => $value ) {
+			$values[ $key ] = $this->set( $key, $value, $group, $expire );
+		}
+
+		return $values;
+	}
+
 	function switch_to_blog( $blog_id ) {
 		global $table_prefix;
 

tests/test-object-cache.php

@@ -1041,4 +1041,61 @@ public function test_colorize_debug_line( $command, $expected_color ) {
 
 		$this->assertStringContainsString( $expected_color, $colorized );
 	}
+
+	public function test_wp_cache_add_multiple() {
+		$found = wp_cache_add_multiple(
+			array(
+				'foo1' => 'bar',
+				'foo2' => 'bar',
+				'foo3' => 'bar',
+			),
+			'group1'
+		);
+
+		$expected = array(
+			'foo1' => true,
+			'foo2' => true,
+			'foo3' => true,
+		);
+
+		$this->assertSame( $expected, $found );
+	}
+
+	public function test_wp_cache_set_multiple() {
+		$found = wp_cache_set_multiple(
+			array(
+				'foo1' => 'bar',
+				'foo2' => 'bar',
+				'foo3' => 'bar',
+			),
+			'group1'
+		);
+
+		$expected = array(
+			'foo1' => true,
+			'foo2' => true,
+			'foo3' => true,
+		);
+
+		$this->assertSame( $expected, $found );
+	}
+
+	public function test_wp_cache_delete_multiple() {
+		wp_cache_set( 'foo1', 'bar', 'group1' );
+		wp_cache_set( 'foo2', 'bar', 'group1' );
+		wp_cache_set( 'foo3', 'bar', 'group2' );
+
+		$found = wp_cache_delete_multiple(
+			array( 'foo1', 'foo2', 'foo3' ),
+			'group1'
+		);
+
+		$expected = array(
+			'foo1' => true,
+			'foo2' => true,
+			'foo3' => false,
+		);
+
+		$this->assertSame( $expected, $found );
+	}
 }