Changeset 29 in code for trunk/server.go

Timestamp:

Feb 6, 2020, 9:25:32 PM (5 years ago)

Author:

contact

Message:

Keep track of upstream connections in a list

File:

• trunk/server.go (modified) (3 diffs)

trunk/server.go

@@ -46 +46 @@
         lock            sync.Mutex
         downstreamConns []*downstreamConn
+        upstreamConns   []*upstreamConn
 }
 
@@ -62 +63 @@
                                 return
                         }
+
+                        s.lock.Lock()
+                        s.upstreamConns = append(s.upstreamConns, conn)
+                        s.lock.Unlock()
+
                         if err := conn.readMessages(); err != nil {
                                 conn.logger.Printf("failed to handle messages: %v", err)
                         }
+
+                        s.lock.Lock()
+                        for i, c := range s.upstreamConns {
+                                if c == conn {
+                                        s.upstreamConns = append(s.upstreamConns[:i], s.upstreamConns[i+1:]...)
+                                        break
+                                }
+                        }
+                        s.lock.Unlock()
                 }()
         }
@@ -81 +96 @@
                         s.downstreamConns = append(s.downstreamConns, conn)
                         s.lock.Unlock()
+
                         if err := conn.readMessages(); err != nil {
                                 conn.logger.Printf("failed to handle messages: %v", err)
                         }
+
                         s.lock.Lock()
                         for i, c := range s.downstreamConns {