Add channel storage to LocalDriver to simplify PresenceChannel logic

src/HttpApi/Controllers/FetchChannelsController.php

@@ -32,7 +32,6 @@ class FetchChannelsController extends Controller
             });
         }
 
-        if (config('websockets.replication.enabled') === true) {
         // We want to get the channel user count all in one shot when
         // using a replication backend rather than doing individual queries.
         // To do so, we first collect the list of channel names.
@@ -54,11 +53,6 @@ class FetchChannelsController extends Controller
         });
     }
 
-        return $this->collectUserCounts($channels, $attributes, function (PresenceChannel $channel) {
-            return $channel->getUserCount();
-        });
-    }
-
     protected function collectUserCounts(Collection $channels, array $attributes, callable $transformer)
     {
         return [

src/PubSub/Drivers/LocalClient.php

@@ -10,6 +10,13 @@ use BeyondCode\LaravelWebSockets\PubSub\ReplicationInterface;
 
 class LocalClient implements ReplicationInterface
 {
+    /**
+     * Mapping of the presence JSON data for users in each channel
+     *
+     * @var string[][]
+     */
+    protected $channelData = [];
+
     /**
      * Boot the pub/sub provider (open connections, initial subscriptions, etc).
      *
@@ -31,6 +38,7 @@ class LocalClient implements ReplicationInterface
      */
     public function publish(string $appId, string $channel, stdClass $payload) : bool
     {
+        // Nothing to do, nobody to publish to
         return true;
     }
 
@@ -69,6 +77,7 @@ class LocalClient implements ReplicationInterface
      */
     public function joinChannel(string $appId, string $channel, string $socketId, string $data)
     {
+        $this->channelData["$appId:$channel"][$socketId] = $data;
     }
 
     /**
@@ -81,6 +90,10 @@ class LocalClient implements ReplicationInterface
      */
     public function leaveChannel(string $appId, string $channel, string $socketId)
     {
+        unset($this->channelData["$appId:$channel"][$socketId]);
+        if (empty($this->channelData["$appId:$channel"])) {
+            unset($this->channelData["$appId:$channel"]);
+        }
     }
 
     /**
@@ -92,7 +105,14 @@ class LocalClient implements ReplicationInterface
      */
     public function channelMembers(string $appId, string $channel) : PromiseInterface
     {
-        return new FulfilledPromise(null);
+        $members = $this->channelData["$appId:$channel"] ?? [];
+
+        // The data is expected as objects, so we need to JSON decode
+        $members = array_map(function ($user) {
+            return json_decode($user);
+        }, $members);
+
+        return new FulfilledPromise($members);
     }
 
     /**
@@ -104,6 +124,15 @@ class LocalClient implements ReplicationInterface
      */
     public function channelMemberCounts(string $appId, array $channelNames) : PromiseInterface
     {
-        return new FulfilledPromise(null);
+        $results = [];
+
+        // Count the number of users per channel
+        foreach ($channelNames as $channel) {
+            $results[$channel] = isset($this->channelData["$appId:$channel"])
+                ? count($this->channelData["$appId:$channel"])
+                : 0;
+        }
+
+        return new FulfilledPromise($results);
     }
 }

src/PubSub/Drivers/RedisClient.php

@@ -97,7 +97,7 @@ class RedisClient implements ReplicationInterface
         // expect the channel name to not include the app ID.
         $payload->channel = Str::after($redisChannel, "$appId:");
 
-        /* @var $channelManager ChannelManager */
+        /* @var ChannelManager $channelManager */
         $channelManager = app(ChannelManager::class);
 
         // Load the Channel instance, if any

src/WebSockets/Channels/PresenceChannel.php

@@ -10,6 +10,16 @@ use BeyondCode\LaravelWebSockets\WebSockets\Exceptions\InvalidSignature;
 
 class PresenceChannel extends Channel
 {
+    /**
+     * Data for the users connected to this channel
+     *
+     * Note: If replication is enabled, this will only contain entries
+     * for the users directly connected to this server instance. Requests
+     * for data for all users in the channel should be routed through
+     * ReplicationInterface.
+     *
+     * @var string[]
+     */
     protected $users = [];
 
     /**
@@ -18,23 +28,11 @@ class PresenceChannel extends Channel
      */
     public function getUsers(string $appId)
     {
-        if (config('websockets.replication.enabled') === true) {
         // Get the members list from the replication backend
         return app(ReplicationInterface::class)
             ->channelMembers($appId, $this->channelName);
     }
 
-        return $this->users;
-    }
-
-    /**
-     * @return array
-     */
-    public function getUserCount()
-    {
-        return count($this->users);
-    }
-
     /**
      * @link https://pusher.com/docs/pusher_protocol#presence-channel-events
      *
@@ -51,7 +49,6 @@ class PresenceChannel extends Channel
         $channelData = json_decode($payload->channel_data);
         $this->users[$connection->socketId] = $channelData;
 
-        if (config('websockets.replication.enabled') === true) {
         // Add the connection as a member of the channel
         app(ReplicationInterface::class)
             ->joinChannel(
@@ -73,14 +70,6 @@ class PresenceChannel extends Channel
                     'data' => json_encode($this->getChannelData($users)),
                 ]));
             });
-        } else {
-            // Send the success event
-            $connection->send(json_encode([
-                'event' => 'pusher_internal:subscription_succeeded',
-                'channel' => $this->channelName,
-                'data' => json_encode($this->getChannelData($this->users)),
-            ]));
-        }
 
         $this->broadcastToOthers($connection, (object) [
             'event' => 'pusher_internal:member_added',
@@ -97,7 +86,6 @@ class PresenceChannel extends Channel
             return;
         }
 
-        if (config('websockets.replication.enabled') === true) {
         // Remove the connection as a member of the channel
         app(ReplicationInterface::class)
             ->leaveChannel(
@@ -105,7 +93,6 @@ class PresenceChannel extends Channel
                 $this->channelName,
                 $connection->socketId
             );
-        }
 
         $this->broadcastToOthers($connection, (object) [
             'event' => 'pusher_internal:member_removed',
@@ -124,7 +111,6 @@ class PresenceChannel extends Channel
      */
     public function toArray(string $appId = null)
     {
-        if (config('websockets.replication.enabled') === true) {
         return app(ReplicationInterface::class)
             ->channelMembers($appId, $this->channelName)
             ->then(function ($users) {
@@ -134,11 +120,6 @@ class PresenceChannel extends Channel
             });
     }
 
-        return array_merge(parent::toArray(), [
-            'user_count' => count($this->users),
-        ]);
-    }
-
     protected function getChannelData(array $users): array
     {
         return [