Changeset 489 in code for trunk/db.go

Timestamp:

Mar 31, 2021, 4:04:13 PM (4 years ago)

Author:

contact

Message:

Save delivery receipts in DB

This avoids loosing history on restart for clients that don't
support chathistory.

Closes: ​https://todo.sr.ht/~emersion/soju/80

File:

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

trunk/db.go

@@ -119 +119 @@
         DetachAfter   time.Duration
         DetachOn      MessageFilter
+}
+
+type DeliveryReceipt struct {
+        ID            int64
+        Target        string // channel or nick
+        Client        string
+        InternalMsgID string
 }
 
@@ -161 +168 @@
         FOREIGN KEY(network) REFERENCES Network(id),
         UNIQUE(network, name)
+);
+
+CREATE TABLE DeliveryReceipt (
+        id INTEGER PRIMARY KEY,
+        network INTEGER NOT NULL,
+        target VARCHAR(255) NOT NULL,
+        client VARCHAR(255),
+        internal_msgid VARCHAR(255) NOT NULL,
+        FOREIGN KEY(network) REFERENCES Network(id),
+        UNIQUE(network, target, client)
 );
 `
@@ -218 +235 @@
                 ALTER TABLE Channel ADD COLUMN detach_on INTEGER NOT NULL DEFAULT 0;
         `,
+        `
+                CREATE TABLE DeliveryReceipt (
+                        id INTEGER PRIMARY KEY,
+                        network INTEGER NOT NULL,
+                        target VARCHAR(255) NOT NULL,
+                        client VARCHAR(255),
+                        internal_msgid VARCHAR(255) NOT NULL,
+                        FOREIGN KEY(network) REFERENCES Network(id),
+                        UNIQUE(network, target, client)
+                );
+        `,
 }
 
@@ -579 +607 @@
         return err
 }
+
+func (db *DB) ListDeliveryReceipts(networkID int64) ([]DeliveryReceipt, error) {
+        db.lock.RLock()
+        defer db.lock.RUnlock()
+
+        rows, err := db.db.Query(`SELECT id, target, client, internal_msgid
+                FROM DeliveryReceipt
+                WHERE network = ?`, networkID)
+        if err != nil {
+                return nil, err
+        }
+        defer rows.Close()
+
+        var receipts []DeliveryReceipt
+        for rows.Next() {
+                var rcpt DeliveryReceipt
+                var client sql.NullString
+                if err := rows.Scan(&rcpt.ID, &rcpt.Target, &client, &rcpt.InternalMsgID); err != nil {
+                        return nil, err
+                }
+                rcpt.Client = client.String
+                receipts = append(receipts, rcpt)
+        }
+        if err := rows.Err(); err != nil {
+                return nil, err
+        }
+
+        return receipts, nil
+}
+
+func (db *DB) StoreClientDeliveryReceipts(networkID int64, client string, receipts []DeliveryReceipt) 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 DeliveryReceipt WHERE network = ? AND client = ?",
+                networkID, toNullString(client))
+        if err != nil {
+                return err
+        }
+
+        for i := range receipts {
+                rcpt := &receipts[i]
+
+                res, err := tx.Exec("INSERT INTO DeliveryReceipt(network, target, client, internal_msgid) VALUES (?, ?, ?, ?)",
+                        networkID, rcpt.Target, toNullString(client), rcpt.InternalMsgID)
+                if err != nil {
+                        return err
+                }
+                rcpt.ID, err = res.LastInsertId()
+                if err != nil {
+                        return err
+                }
+        }
+
+        return tx.Commit()
+}