Changeset 73 in code for trunk/upstream.go

Timestamp:

Mar 4, 2020, 2:44:13 PM (5 years ago)

Author:

contact

Message:

Allow clients to specify an upstream name in their username

File:

• trunk/upstream.go (modified) (5 diffs)

trunk/upstream.go

@@ -101 +101 @@
 }
 
+func (uc *upstreamConn) forEachDownstream(f func(*downstreamConn)) {
+        uc.user.forEachDownstream(func(dc *downstreamConn) {
+                if dc.upstream != nil && dc.upstream != uc.upstream {
+                        return
+                }
+                f(dc)
+        })
+}
+
 func (uc *upstreamConn) getChannel(name string) (*upstreamChannel, error) {
         ch, ok := uc.channels[name]
@@ -141 +150 @@
                         }
 
-                        uc.user.forEachDownstream(func(dc *downstreamConn) {
+                        uc.forEachDownstream(func(dc *downstreamConn) {
                                 dc.SendMessage(&irc.Message{
                                         Prefix:  dc.marshalUserPrefix(uc, msg.Prefix),
@@ -211 +220 @@
                         }
 
-                        uc.user.forEachDownstream(func(dc *downstreamConn) {
+                        uc.forEachDownstream(func(dc *downstreamConn) {
                                 dc.SendMessage(&irc.Message{
                                         Prefix:  dc.marshalUserPrefix(uc, msg.Prefix),
@@ -241 +250 @@
                         }
 
-                        uc.user.forEachDownstream(func(dc *downstreamConn) {
+                        uc.forEachDownstream(func(dc *downstreamConn) {
                                 dc.SendMessage(&irc.Message{
                                         Prefix:  dc.marshalUserPrefix(uc, msg.Prefix),
@@ -327 +336 @@
                 ch.complete = true
 
-                uc.user.forEachDownstream(func(dc *downstreamConn) {
+                uc.forEachDownstream(func(dc *downstreamConn) {
                         forwardChannel(dc, ch)
                 })