Changeset 202 in code for trunk/user.go

Timestamp:

Apr 1, 2020, 1:40:20 PM (5 years ago)

Author:

contact

Message:

Add "network delete" service command

And add all the infrastructure required to stop and delete networks.

References: ​https://todo.sr.ht/~emersion/soju/17

File:

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

trunk/user.go

@@ -38 +38 @@
 type network struct {
         Network
-        user *user
-        ring *Ring
+        user    *user
+        ring    *Ring
+        stopped chan struct{}
 
         lock    sync.Mutex
@@ -51 +52 @@
                 user:    user,
                 ring:    NewRing(user.srv.RingCap),
+                stopped: make(chan struct{}),
                 history: make(map[string]uint64),
         }
@@ -58 +60 @@
         var lastTry time.Time
         for {
+                select {
+                case <-net.stopped:
+                        return
+                default:
+                        // This space is intentionally left blank
+                }
+
                 if dur := time.Now().Sub(lastTry); dur < retryConnectMinDelay {
                         delay := retryConnectMinDelay - dur
@@ -91 +100 @@
         defer net.lock.Unlock()
         return net.conn
+}
+
+func (net *network) Stop() {
+        select {
+        case <-net.stopped:
+                return
+        default:
+                close(net.stopped)
+        }
+
+        if uc := net.upstream(); uc != nil {
+                uc.Close()
+        }
 }
 
@@ -266 +288 @@
         return network, nil
 }
+
+func (u *user) deleteNetwork(id int64) error {
+        for i, net := range u.networks {
+                if net.ID != id {
+                        continue
+                }
+
+                if err := u.srv.db.DeleteNetwork(net.ID); err != nil {
+                        return err
+                }
+
+                u.forEachDownstream(func(dc *downstreamConn) {
+                        if dc.network != nil && dc.network == net {
+                                dc.Close()
+                        }
+                })
+
+                net.Stop()
+                u.networks = append(u.networks[:i], u.networks[i+1:]...)
+                return nil
+        }
+
+        panic("tried deleting a non-existing network")
+}