⚡️ Quicker UI updates when adding participant

lukevella · Mar 2, 2024 · d0f12d9 · d0f12d9
1 parent f9d5d0a
commit d0f12d9
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/apps/web/src/components/poll/mutations.ts b/apps/web/src/components/poll/mutations.ts
@@ -19,7 +19,17 @@ export const useAddParticipantMutation = () => {
   const queryClient = trpc.useUtils();
 
   return trpc.polls.participants.add.useMutation({
-    onSuccess: (_, { pollId, name, email }) => {
+    onSuccess: (newParticipant, input) => {
+      const { pollId, name, email } = newParticipant;
+      queryClient.polls.participants.list.setData(
+        { pollId },
+        (existingParticipants = []) => {
+          return [
+            { ...newParticipant, votes: input.votes },
+            ...existingParticipants,
+          ];
+        },
+      );
       queryClient.polls.participants.list.invalidate({ pollId });
       posthog?.capture("add participant", {
         pollId,