fileSender code splitting

2025-09-07 22:52:59 +08:00
parent 99c927f5c7
commit 230a06b3fb
9 changed files with 1926 additions and 851 deletions
@@ -0,0 +1,232 @@
+import { SpeedCalculator } from "@/lib/speedCalculator";
+import { StateManager } from "./StateManager";
+import { postLogToBackend } from "@/app/config/api";
+
+/**
+ * 🚀 进度回调类型定义
+ */
+export type ProgressCallback = (fileId: string, progress: number, speed: number) => void;
+
+/**
+ * 🚀 进度跟踪器
+ * 负责文件和文件夹的进度计算、速度统计、回调触发
+ */
+export class ProgressTracker {
+  private speedCalculator = new SpeedCalculator();
+
+  constructor(private stateManager: StateManager) {}
+
+  /**
+   * 🎯 更新文件传输进度
+   */
+  async updateFileProgress(
+    byteLength: number,
+    fileId: string,
+    fileSize: number,
+    peerId: string,
+    wasActuallySent: boolean = true
+  ): Promise<void> {
+    const peerState = this.stateManager.getPeerState(peerId);
+    if (!peerState) return;
+
+    // 重要修复：只有成功发送的数据才更新统计
+    if (!wasActuallySent) {
+      postLogToBackend(
+        `[DEBUG] ⚠️ Data send failed, not updating progress - fileId: ${fileId}, size: ${byteLength}`
+      );
+      return;
+    }
+
+    // 更新文件已发送字节数
+    this.stateManager.updateFileBytesSent(peerId, fileId, byteLength);
+
+    // 计算进度ID和统计数据
+    let progressFileId = fileId;
+    let currentBytes = this.stateManager.getFileBytesSent(peerId, fileId);
+    let totalSize = fileSize;
+
+    // 如果文件属于文件夹，重新计算文件夹进度
+    if (peerState.currentFolderName) {
+      const folderName = peerState.currentFolderName;
+      const folderMeta = this.stateManager.getFolderMeta(folderName);
+      
+      progressFileId = folderName;
+      totalSize = folderMeta?.totalSize || 0;
+
+      // 重新计算文件夹进度（从其所有文件的进度总和）
+      // 这对于断点续传更加健壮和正确
+      currentBytes = this.stateManager.getFolderBytesSent(peerId, folderName);
+
+      postLogToBackend(
+        `[DEBUG] 📁 Folder progress update - folder: ${folderName}, file: ${fileId}, currentBytes: ${currentBytes}, totalSize: ${totalSize}`
+      );
+    }
+
+    // 更新速度计算器
+    this.speedCalculator.updateSendSpeed(peerId, currentBytes);
+    const speed = this.speedCalculator.getSendSpeed(peerId);
+    const progress = totalSize > 0 ? currentBytes / totalSize : 0;
+
+    // 持续更新网络性能（从传输速度学习）
+    this.stateManager.updateNetworkFromSpeed(peerId, speed);
+
+    // 触发进度回调
+    this.triggerProgressCallback(peerId, progressFileId, progress, speed);
+
+    postLogToBackend(
+      `[DEBUG] 📊 Progress updated - ${progressFileId}: ${(progress * 100).toFixed(2)}%, speed: ${speed.toFixed(2)} KB/s, bytes: ${currentBytes}/${totalSize}`
+    );
+  }
+
+  /**
+   * 🎯 更新文件夹传输进度
+   */
+  async updateFolderProgress(
+    folderName: string,
+    fileProgress: Record<string, number>,
+    peerId: string
+  ): Promise<void> {
+    const folderMeta = this.stateManager.getFolderMeta(folderName);
+    if (!folderMeta) {
+      postLogToBackend(`[DEBUG] ⚠️ Folder metadata not found: ${folderName}`);
+      return;
+    }
+
+    // 计算文件夹总进度
+    let totalSentBytes = 0;
+    folderMeta.fileIds.forEach((fileId) => {
+      totalSentBytes += this.stateManager.getFileBytesSent(peerId, fileId);
+    });
+
+    const progress = folderMeta.totalSize > 0 ? totalSentBytes / folderMeta.totalSize : 0;
+    const speed = this.speedCalculator.getSendSpeed(peerId);
+
+    // 触发文件夹进度回调
+    this.triggerProgressCallback(peerId, folderName, progress, speed);
+
+    postLogToBackend(
+      `[DEBUG] 📁 Folder progress - ${folderName}: ${(progress * 100).toFixed(2)}%, speed: ${speed.toFixed(2)} KB/s, bytes: ${totalSentBytes}/${folderMeta.totalSize}`
+    );
+  }
+
+  /**
+   * 🎯 设置进度回调函数
+   */
+  setProgressCallback(
+    callback: ProgressCallback,
+    peerId: string
+  ): void {
+    this.stateManager.updatePeerState(peerId, { progressCallback: callback });
+  }
+
+  /**
+   * 🎯 触发进度回调
+   */
+  private triggerProgressCallback(
+    peerId: string,
+    fileId: string,
+    progress: number,
+    speed: number
+  ): void {
+    const peerState = this.stateManager.getPeerState(peerId);
+    if (peerState.progressCallback) {
+      try {
+        peerState.progressCallback(fileId, progress, speed);
+      } catch (error) {
+        postLogToBackend(
+          `[DEBUG] ❌ Progress callback error - fileId: ${fileId}, error: ${error}`
+        );
+      }
+    }
+  }
+
+  /**
+   * 🎯 计算当前传输速度
+   */
+  getCurrentSpeed(peerId: string): number {
+    return this.speedCalculator.getSendSpeed(peerId);
+  }
+
+  /**
+   * 🎯 完成文件传输进度（设置为100%）
+   */
+  completeFileProgress(fileId: string, peerId: string): void {
+    this.triggerProgressCallback(peerId, fileId, 1.0, 0);
+    
+    postLogToBackend(`[DEBUG] ✅ File progress completed: ${fileId}`);
+  }
+
+  /**
+   * 🎯 完成文件夹传输进度（设置为100%）
+   */
+  completeFolderProgress(folderName: string, peerId: string): void {
+    this.triggerProgressCallback(peerId, folderName, 1.0, 0);
+    
+    postLogToBackend(`[DEBUG] ✅ Folder progress completed: ${folderName}`);
+  }
+
+  /**
+   * 📊 获取详细的进度统计信息
+   */
+  getProgressStats(peerId: string) {
+    const peerState = this.stateManager.getPeerState(peerId);
+    const currentSpeed = this.getCurrentSpeed(peerId);
+    
+    // 计算总的已发送字节数
+    let totalBytesSent = 0;
+    Object.values(peerState.totalBytesSent).forEach(bytes => {
+      totalBytesSent += bytes;
+    });
+
+    return {
+      peerId,
+      currentSpeed,
+      totalBytesSent,
+      activeTransfers: Object.keys(peerState.totalBytesSent).length,
+      currentFolderName: peerState.currentFolderName,
+      isSending: peerState.isSending,
+      hasProgressCallback: !!peerState.progressCallback,
+    };
+  }
+
+  /**
+   * 📊 获取文件夹的详细进度信息
+   */
+  getFolderProgressDetails(folderName: string, peerId: string) {
+    const folderMeta = this.stateManager.getFolderMeta(folderName);
+    if (!folderMeta) return null;
+
+    const fileProgresses: Record<string, { sent: number; total: number; progress: number }> = {};
+    let totalSent = 0;
+
+    folderMeta.fileIds.forEach(fileId => {
+      const sent = this.stateManager.getFileBytesSent(peerId, fileId);
+      // 注意：这里需要从pendingFiles获取文件大小，暂时使用0
+      const total = 0; // TODO: 需要从StateManager获取文件大小
+      totalSent += sent;
+      
+      fileProgresses[fileId] = {
+        sent,
+        total,
+        progress: total > 0 ? sent / total : 0,
+      };
+    });
+
+    return {
+      folderName,
+      totalSize: folderMeta.totalSize,
+      totalSent,
+      overallProgress: folderMeta.totalSize > 0 ? totalSent / folderMeta.totalSize : 0,
+      fileCount: folderMeta.fileIds.length,
+      fileProgresses,
+    };
+  }
+
+  /**
+   * 🧹 清理进度跟踪资源
+   */
+  cleanup(): void {
+    // SpeedCalculator 内部会自动清理过期数据
+    postLogToBackend("[DEBUG] 🧹 ProgressTracker cleaned up");
+  }
+}