Changeset 40 in code for trunk/downstream.go

Timestamp:

Feb 7, 2020, 10:56:36 AM (5 years ago)

Author:

contact

Message:

Add user.forEachDownstream

File:

• trunk/downstream.go (modified) (2 diffs)

trunk/downstream.go

@@ -116 +116 @@
                 return fmt.Errorf("downstream connection already closed")
         }
-        if err := c.net.Close(); err != nil {
-                return err
-        }
+
+        if u := c.user; u != nil {
+                u.lock.Lock()
+                for i := range u.downstreamConns {
+                        if u.downstreamConns[i] == c {
+                                u.downstreamConns = append(u.downstreamConns[:i], u.downstreamConns[i+1:]...)
+                        }
+                }
+                u.lock.Unlock()
+        }
+
         close(c.messages)
         c.closed = true
-        return nil
+
+        return c.net.Close()
 }
 
@@ -182 +191 @@
         c.registered = true
         c.user = u
+
+        u.lock.Lock()
+        u.downstreamConns = append(u.downstreamConns, c)
+        u.lock.Unlock()
 
         c.messages <- &irc.Message{