Changeset 375 in code

Timestamp:

Aug 3, 2020, 4:43:39 PM (5 years ago)

Author:

contact

Message:

Add DB.DeleteUser

File:

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

trunk/db.go

@@ -249 +249 @@
 
         return err
+}
+
+func (db *DB) DeleteUser(username string) error {
+        db.lock.Lock()
+        defer db.lock.Unlock()
+
+        tx, err := db.db.Begin()
+        if err != nil {
+                return err
+        }
+        defer tx.Rollback()
+
+        _, err = tx.Exec(`DELETE FROM Channel
+                WHERE id IN (
+                        SELECT Channel.id
+                        FROM Channel
+                        JOIN Network ON Channel.network = Network.id
+                        WHERE Network.user = ?
+                )`, username)
+        if err != nil {
+                return err
+        }
+
+        _, err = tx.Exec("DELETE FROM Network WHERE user = ?", username)
+        if err != nil {
+                return err
+        }
+
+        _, err = tx.Exec("DELETE FROM User WHERE username = ?", username)
+        if err != nil {
+                return err
+        }
+
+        return tx.Commit()
 }
 
@@ -360 +394 @@
         defer tx.Rollback()
 
+        _, err = tx.Exec("DELETE FROM Channel WHERE network = ?", id)
+        if err != nil {
+                return err
+        }
+
         _, err = tx.Exec("DELETE FROM Network WHERE id = ?", id)
-        if err != nil {
-                return err
-        }
-
-        _, err = tx.Exec("DELETE FROM Channel WHERE network = ?", id)
         if err != nil {
                 return err