Changeset 89 in code for trunk/downstream.go

Timestamp:

Mar 12, 2020, 5:33:03 PM (5 years ago)

Author:

contact

Message:

Update DB on JOIN and PART

File:

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

trunk/downstream.go

@@ -115 +115 @@
 }
 
+// upstream returns the upstream connection, if any. If there are zero or if
+// there are multiple upstream connections, it returns nil.
+func (dc *downstreamConn) upstream() *upstreamConn {
+        if dc.network == nil {
+                return nil
+        }
+
+        var upstream *upstreamConn
+        dc.forEachUpstream(func(uc *upstreamConn) {
+                upstream = uc
+        })
+        return upstream
+}
+
 func (dc *downstreamConn) unmarshalChannel(name string) (*upstreamConn, string, error) {
+        if uc := dc.upstream(); uc != nil {
+                return uc, name, nil
+        }
+
         // TODO: extract network name from channel name if dc.upstream == nil
         var channel *upstreamChannel
@@ -462 +480 @@
                         Params:  []string{upstreamName},
                 })
-                // TODO: add/remove channel from upstream config
+
+                switch msg.Command {
+                case "JOIN":
+                        err := dc.srv.db.StoreChannel(uc.network.ID, &Channel{
+                                Name: upstreamName,
+                        })
+                        if err != nil {
+                                dc.logger.Printf("failed to create channel %q in DB: %v", upstreamName, err)
+                        }
+                case "PART":
+                        if err := dc.srv.db.DeleteChannel(uc.network.ID, upstreamName); err != nil {
+                                dc.logger.Printf("failed to delete channel %q in DB: %v", upstreamName, err)
+                        }
+                }
         case "MODE":
                 if msg.Prefix == nil {