feat: 实现消息分发推送确认撤回与启动恢复

internal/app/message/session.go

@@ -0,0 +1,75 @@
+package message
+
+import (
+	"context"
+	"database/sql"
+
+	"git.asio.asia/nixevol/NixMsg/internal/app/port"
+)
+
+// OnHandshakeComplete 握手完成：写 online_since、清空不 keep 的 expire_at，并推送。
+// 调用方须先把连接登记进 ConnRegistry（MemoryConns.Set）。
+func (a *App) OnHandshakeComplete(ctx context.Context, endpointID string, conn LiveConn) error {
+	nowMs := a.now().UnixMilli()
+	err := a.db.Queue.Do(ctx, func(tx *sql.Tx) error {
+		if _, err := tx.Exec(`UPDATE endpoints SET online_since = ? WHERE id = ?`, nowMs, endpointID); err != nil {
+			return err
+		}
+		_, err := tx.Exec(`
+UPDATE deliveries SET expire_at = NULL, updated_at = ?
+WHERE endpoint_id = ? AND state = 'pending' AND keep = 0`, nowMs, endpointID)
+		return err
+	})
+	if err != nil {
+		return err
+	}
+	return a.PushPending(ctx, endpointID, conn.ConnID)
+}
+
+// OnDisconnect 连接断开：当前连接则延长宽限；按代号清 pushed_conn。
+// 调用方负责从 ConnRegistry 移除连接。
+func (a *App) OnDisconnect(ctx context.Context, endpointID string, connID port.ConnID, isCurrent bool) error {
+	nowMs := a.now().UnixMilli()
+	graceMs := a.lim.GraceSeconds * 1000
+	deadline := nowMs + graceMs
+
+	err := a.db.Queue.Do(ctx, func(tx *sql.Tx) error {
+		if isCurrent {
+			if _, err := tx.Exec(`UPDATE endpoints SET offline_since = ? WHERE id = ?`, nowMs, endpointID); err != nil {
+				return err
+			}
+			if _, err := tx.Exec(`
+UPDATE deliveries SET
+  expire_at = CASE
+    WHEN expire_at IS NOT NULL AND expire_at > ? THEN expire_at
+    ELSE ?
+  END,
+  updated_at = ?
+WHERE endpoint_id = ? AND state = 'pending' AND keep = 0`, deadline, deadline, nowMs, endpointID); err != nil {
+				return err
+			}
+			if _, err := tx.Exec(`
+UPDATE deliveries SET
+  expire_at = CASE
+    WHEN expire_at IS NULL OR expire_at < ? THEN ?
+    ELSE expire_at
+  END,
+  updated_at = ?
+WHERE endpoint_id = ? AND state = 'pending' AND keep = 1 AND pushed_conn = ?`,
+				deadline, deadline, nowMs, endpointID, string(connID)); err != nil {
+				return err
+			}
+		}
+		_, err := tx.Exec(`
+UPDATE deliveries SET pushed_conn = NULL, updated_at = ?
+WHERE state = 'pending' AND pushed_conn = ?`, nowMs, string(connID))
+		return err
+	})
+	if err != nil {
+		return err
+	}
+	if live, ok := a.lookupConn(endpointID); ok && live.ConnID != connID {
+		a.WakePush(endpointID)
+	}
+	return nil
+}