From 3d397d952bc013a7c400931c9c150838a710c503 Mon Sep 17 00:00:00 2001
From: Rostislav Antonov <antonov.qwertylox004@yandex.ru>
Date: Wed, 28 Jun 2023 11:55:58 +0200
Subject: [PATCH] fix sockets on frontend side

---
 .../frontend/create-sample.tsx                | 46 +++++++++----------
 .../frontend/update-sample.tsx                |  6 +--
 2 files changed, 24 insertions(+), 28 deletions(-)

diff --git a/examples/aidbox-subscription/frontend/create-sample.tsx b/examples/aidbox-subscription/frontend/create-sample.tsx
index 6853a76..dd49ace 100644
--- a/examples/aidbox-subscription/frontend/create-sample.tsx
+++ b/examples/aidbox-subscription/frontend/create-sample.tsx
@@ -1,6 +1,6 @@
 import { Button, Card, Container, Grid, Link, Progress, Text } from '@nextui-org/react'
 import { Timeline } from 'antd'
-import { useState } from 'react'
+import { useEffect, useState } from 'react'
 import { Plus, TickSquare } from 'react-iconly'
 
 import { aidboxClient } from '../backend/aidbox-client.js'
@@ -65,31 +65,27 @@ export const CreateSample = () => {
     setPatientsCreated(true)
   }
 
-  socketIo.on('subs_notification_patient', function (data) {
-    if (!subsNotifications.includes(data)) {
-      setSubsNotifications(subsNotifications => [...subsNotifications, data])
-    }
-  })
+  useEffect(() => {
+    socketIo.on('subs_notification_patient', function (data) {
+      setTimeout(() => setSubsNotifications(subsNotifications => [...subsNotifications, data]), 400)
+    })
+    socketIo.on('push_patient', function (data) {
+       setTimeout(() => setPushedPatients(pushedPatients => [...pushedPatients, data]), 400)
+    })
 
-  socketIo.on('push_patient', function (data) {
-    console.log(data)
-    if (!pushedPatients.includes(data)) {
-      setPushedPatients(pushedPatients => [...pushedPatients, data])
-    }
-  })
+    socketIo.on('pull_patient', function (data) {
+      setTimeout(() => setPulledPatients(pulledPatients => [...pulledPatients, data]), 400)
+    })
 
-  socketIo.on('pull_patient', function (data) {
-    if (!pulledPatients.includes(data)) {
-      setPulledPatients(pulledPatients => [...pulledPatients, data])
-    }
-  })
+    socketIo.on('create_task_patient', function (data) {
+      setTimeout(() => setCreatedTasks(createdTasks => [...createdTasks, data]), 400)
+    })
 
-  socketIo.on('create_task_patient', function (data) {
-    if (!createdTasks.includes(data)) {
-      setCreatedTasks(createdTasks => [...createdTasks, data])
+    return () => {
+      socketIo.off()
     }
-  })
-console.log(pushedPatients, 'pushedPatients')
+}, [])
+
   return (
     <>
       <Text h2>Create Patient Subscription</Text>
@@ -102,7 +98,7 @@ console.log(pushedPatients, 'pushedPatients')
             direction='column'
           >
             <Text css={{ 'text-align': 'center', 'margin-top': 0 }}>
-              Create new patients to trigger subscriptions.
+              We can create 10 patients and observe how the subscription feature works.
             </Text>
             {patientsCreated
             ? <>
@@ -208,13 +204,13 @@ console.log(pushedPatients, 'pushedPatients')
                 }
                 ]}
               />}
-            </>
+              </>
             : <Button
                 onPress={createPatients}
                 icon={<Plus set='light' />}
               >
               Create 10 patients
-            </Button>
+              </Button>
           }
           </Container>
         </Card.Body>
diff --git a/examples/aidbox-subscription/frontend/update-sample.tsx b/examples/aidbox-subscription/frontend/update-sample.tsx
index 1099f13..a7322ca 100644
--- a/examples/aidbox-subscription/frontend/update-sample.tsx
+++ b/examples/aidbox-subscription/frontend/update-sample.tsx
@@ -133,13 +133,13 @@ export const UpdateSample = () => {
 
   socketIo.on('subs_notification_appointment', function (data) {
     if (data === appointment?.id) {
-    setTimeout(() => setSubsNotifications(true), 1400)
+      setTimeout(() => setSubsNotifications(true), 1400)
     }
   })
 
   socketIo.on('push_appointment', function (data) {
     if (data === appointment?.id) {
-    setTimeout(() => setPushedAppointment(true), 2400)
+      setTimeout(() => setPushedAppointment(true), 2400)
     }
   })
 
@@ -151,7 +151,7 @@ export const UpdateSample = () => {
 
   socketIo.on('create_task_appointment', function (data) {
     if (data === appointment?.id) {
-    setTimeout(() => setCreatedTasks(true), 500)
+      setTimeout(() => setCreatedTasks(true), 500)
     }
   })