From aa335e615064de005fb98040354147c0d3fa6352 Mon Sep 17 00:00:00 2001
From: Ryan Boren <ryan@git.wordpress.org>
Date: Mon, 19 Jun 2006 17:17:52 +0000
Subject: [PATCH] wp_update_comment_count() from MarkJaquith. fixes #2836

git-svn-id: https://develop.svn.wordpress.org/trunk@3887 602fd350-edb4-49c9-b593-d223f7449a82
---
 wp-includes/comment.php | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/wp-includes/comment.php b/wp-includes/comment.php
index 45547db110..9751c362c5 100644
--- a/wp-includes/comment.php
+++ b/wp-includes/comment.php
@@ -252,7 +252,7 @@ function wp_delete_comment($comment_id) {
 
 	$post_id = $comment->comment_post_ID;
 	if ( $post_id && $comment->comment_approved == 1 )
-		$wpdb->query( "UPDATE $wpdb->posts SET comment_count = comment_count - 1 WHERE ID = '$post_id'" );
+		wp_update_comment_count($post_id);
 
 	do_action('wp_set_comment_status', $comment_id, 'delete');
 	return true;
@@ -300,10 +300,9 @@ function wp_insert_comment($commentdata) {
 
 	$id = $wpdb->insert_id;
 
-	if ( $comment_approved == 1) {
-		$count = $wpdb->get_var("SELECT COUNT(*) FROM $wpdb->comments WHERE comment_post_ID = '$comment_post_ID' AND comment_approved = '1'");
-		$wpdb->query( "UPDATE $wpdb->posts SET comment_count = $count WHERE ID = '$comment_post_ID'" );
-	}
+	if ( $comment_approved == 1)
+		wp_update_comment_count($comment_post_ID);
+
 	return $id;
 }
 
@@ -419,15 +418,24 @@ function wp_update_comment($commentarr) {
 
 	$rval = $wpdb->rows_affected;
 
-	$c = $wpdb->get_row( "SELECT count(*) as c FROM {$wpdb->comments} WHERE comment_post_ID = '$comment_post_ID' AND comment_approved = '1'" );
-	if( is_object( $c ) )
-		$wpdb->query( "UPDATE $wpdb->posts SET comment_count = '$c->c' WHERE ID = '$comment_post_ID'" );
+	wp_update_comment_count($comment_post_ID);
 
 	do_action('edit_comment', $comment_ID);
 
 	return $rval;
 }
 
+function wp_update_comment_count($post_id) {
+	global $wpdb, $comment_count_cache;
+	$post_id = (int) $post_id;
+	if ( !$post_id )
+		return false;
+	$count = $wpdb->get_var("SELECT COUNT(*) FROM $wpdb->comments WHERE comment_post_ID = '$post_id' AND comment_approved = '1'");
+	$wpdb->query("UPDATE $wpdb->posts SET comment_count = $count WHERE ID = '$post_id'");
+	$comment_count_cache[$post_id] = $count;
+	return true;
+}
+
 function pingback($content, $post_ID) {
 	global $wp_version, $wpdb;
 	include_once (ABSPATH . WPINC . '/class-IXR.php');