Update Profile, fix status count

pixelfed · Dec 27, 2020 · 6dcd472 · 6dcd472
1 parent b74e14c
commit 6dcd472
Show file tree

Hide file tree

Showing 5 changed files with 80 additions and 15 deletions.
diff --git a/app/Console/Commands/FixStatusCount.php b/app/Console/Commands/FixStatusCount.php
@@ -0,0 +1,56 @@
+<?php
+
+namespace App\Console\Commands;
+
+use Illuminate\Console\Command;
+use App\Profile;
+
+class FixStatusCount extends Command
+{
+    /**
+     * The name and signature of the console command.
+     *
+     * @var string
+     */
+    protected $signature = 'fix:statuscount';
+
+    /**
+     * The console command description.
+     *
+     * @var string
+     */
+    protected $description = 'fix profile status count';
+
+    /**
+     * Create a new command instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        parent::__construct();
+    }
+
+    /**
+     * Execute the console command.
+     *
+     * @return int
+     */
+    public function handle()
+    {
+        Profile::whereNull('domain')
+        ->chunk(50, function($profiles) {
+            foreach($profiles as $profile) {
+                $profile->status_count = $profile->statuses()
+                ->getQuery()
+                ->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
+                ->whereNull('in_reply_to_id')
+                ->whereNull('reblog_of_id')
+                ->count();
+                $profile->save();
+            }
+        });
+
+        return 0;
+    }
+}
diff --git a/app/Http/Controllers/StatusController.php b/app/Http/Controllers/StatusController.php
@@ -210,6 +210,7 @@ public function delete(Request $request)
         }
 
         Cache::forget('_api:statuses:recent_9:' . $status->profile_id);
+        Cache::forget('profile:status_count:' . $status->profile_id);
         if ($status->profile_id == $user->profile->id || $user->is_admin == true) {
             Cache::forget('profile:status_count:'.$status->profile_id);
             StatusDelete::dispatch($status);

diff --git a/app/Jobs/StatusPipeline/StatusDelete.php b/app/Jobs/StatusPipeline/StatusDelete.php
@@ -56,6 +56,17 @@ public function __construct(Status $status)
     public function handle()
     {
         $status = $this->status;
+        $profile = $this->status->profile;
+
+        $count = $profile->statuses()
+        ->getQuery()
+        ->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
+        ->whereNull('in_reply_to_id')
+        ->whereNull('reblog_of_id')
+        ->count();
+
+        $profile->status_count = ($count - 1);
+        $profile->save();
 
         if(config('federation.activitypub.enabled') == true) {
             $this->fanoutDelete($status);

diff --git a/app/Jobs/StatusPipeline/StatusEntityLexer.php b/app/Jobs/StatusPipeline/StatusEntityLexer.php
@@ -52,6 +52,17 @@ public function __construct(Status $status)
     public function handle()
     {
         $profile = $this->status->profile;
+
+        $count = $profile->statuses()
+        ->getQuery()
+        ->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
+        ->whereNull('in_reply_to_id')
+        ->whereNull('reblog_of_id')
+        ->count();
+
+        $profile->status_count = $count;
+        $profile->save();
+
         if($profile->no_autolink == false) {
             $this->parseEntities();
         }

diff --git a/app/Profile.php b/app/Profile.php
@@ -91,21 +91,7 @@ public function followerCount($short = false)
 
     public function statusCount()
     {
-        return Cache::remember('profile:status_count:'.$this->id, now()->addMonths(1), function() {
-            $count = $this->status_count;
-            if($count) {
-                return $count;
-            }
-            $count = $this->statuses()
-                ->getQuery()
-                ->whereHas('media')
-                ->whereNull('in_reply_to_id')
-                ->whereNull('reblog_of_id')
-                ->count();
-            $this->status_count = $count;
-            $this->save();
-            return $count;
-        });
+        return $this->status_count;
     }
 
     public function following()