feat: only retrieve 50 download records

ltaoo · ltaoo · commit de5b26166668 · 2026-03-03T00:06:34.000+08:00
diff --git a/README.md b/README.md
@@ -69,7 +69,7 @@ CGO_ENABLED=1 GOOS=darwin GOARCH=amd64 SDKROOT=$(xcrun --sdk macosx --show-sdk-p
 ### Linux
 
 ```bash
-# CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -trimpath -ldflags="-s -w" -o wx_video_download
+CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -trimpath -ldflags="-s -w" -o wx_video_download
 
 $env:CGO_ENABLED=0; $env:GOOS="linux"; $env:GOARCH="amd64"; go build -trimpath -ldflags="-s -w" -o wx_video_download
 
diff --git a/cmd/server.go b/cmd/server.go
@@ -52,6 +52,7 @@ func init() {
 	root_cmd.AddCommand(serve_cmd)
 	serve_cmd.AddCommand(server_status_cmd)
 	serve_cmd.AddCommand(mp_stop_cmd)
+	serve_cmd.AddCommand(mp_restart_cmd)
 }
 
 func serve_command() {
@@ -207,34 +208,47 @@ var mp_stop_cmd = &cobra.Command{
 	Use:   "stop",
 	Short: "停止公众号服务",
 	Run: func(cmd *cobra.Command, args []string) {
-		pid, err := read_wx_pidfile()
-		if err != nil || pid == 0 {
-			color.Red("未发现守护进程")
-			return
-		}
-		if !system.IsProcessRunning(pid) {
-			color.Green("进程已停止")
-			_ = remove_wx_pidfile()
-			return
-		}
-		_ = system.TerminateProcess(pid)
-		expire := time.After(8 * time.Second)
-		tick := time.NewTicker(200 * time.Millisecond)
-		defer tick.Stop()
-		for {
-			select {
-			case <-tick.C:
-				if !system.IsProcessRunning(pid) {
-					_ = remove_wx_pidfile()
-					color.Green("服务已关闭")
-					return
-				}
-			case <-expire:
-				color.Red("关闭超时")
+		stop_daemon()
+	},
+}
+
+var mp_restart_cmd = &cobra.Command{
+	Use:   "restart",
+	Short: "重启公众号服务",
+	Run: func(cmd *cobra.Command, args []string) {
+		stop_daemon()
+		start_daemon()
+	},
+}
+
+func stop_daemon() {
+	pid, err := read_wx_pidfile()
+	if err != nil || pid == 0 {
+		color.Red("未发现守护进程")
+		return
+	}
+	if !system.IsProcessRunning(pid) {
+		color.Green("进程已停止")
+		_ = remove_wx_pidfile()
+		return
+	}
+	_ = system.TerminateProcess(pid)
+	expire := time.After(8 * time.Second)
+	tick := time.NewTicker(200 * time.Millisecond)
+	defer tick.Stop()
+	for {
+		select {
+		case <-tick.C:
+			if !system.IsProcessRunning(pid) {
+				_ = remove_wx_pidfile()
+				color.Green("服务已关闭")
 				return
 			}
+		case <-expire:
+			color.Red("关闭超时")
+			return
 		}
-	},
+	}
 }
 
 func port_listening(addr string) bool {
diff --git a/internal/api/client.go b/internal/api/client.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"net/http"
 	"os"
+	"sort"
 	"strconv"
 	"time"
 
@@ -41,13 +42,69 @@ func NewAPIClient(cfg *APIConfig, parent_logger *zerolog.Logger) *APIClient {
 	official_cfg := officialaccount.NewOfficialAccountConfig(cfg.Original, cfg.RemoteServerMode)
 	officialaccount_client := officialaccount.NewOfficialAccountClient(official_cfg, parent_logger)
 	channels_client = channels.NewChannelsClient()
+
+	get_sorted_tasks := func() []*downloadpkg.Task {
+		tasks := downloader.GetTasks()
+		sort.Slice(tasks, func(i, j int) bool {
+			return tasks[i].CreatedAt.After(tasks[j].CreatedAt)
+		})
+		return tasks
+	}
+
 	channels_client.OnConnected = func(client *channels.Client) {
 		// Initial tasks
-		tasks := downloader.GetTasks()
-		if data, err := json.Marshal(APIClientWSMessage{Type: "tasks", Data: tasks}); err == nil {
+		all_tasks := get_sorted_tasks()
+		limit := 50
+		if limit > len(all_tasks) {
+			limit = len(all_tasks)
+		}
+		tasks := all_tasks[:limit]
+		if data, err := json.Marshal(APIClientWSMessage{Type: "tasks", Data: map[string]interface{}{
+			"list":  tasks,
+			"total": len(all_tasks),
+		}}); err == nil {
 			client.Send <- data
 		}
 	}
+
+	channels_client.OnMessage = func(client *channels.Client, message []byte) {
+		var req struct {
+			Type  string `json:"type"`
+			Page  int    `json:"page"`
+			Limit int    `json:"limit"`
+		}
+		if err := json.Unmarshal(message, &req); err == nil && req.Type == "fetch_tasks" {
+			allTasks := get_sorted_tasks()
+			start := (req.Page - 1) * req.Limit
+			if start < 0 {
+				start = 0
+			}
+			if req.Limit <= 0 {
+				req.Limit = 50
+			}
+
+			if start >= len(allTasks) {
+				if data, err := json.Marshal(APIClientWSMessage{Type: "tasks", Data: map[string]interface{}{
+					"list":  []*downloadpkg.Task{},
+					"total": len(allTasks),
+				}}); err == nil {
+					client.Send <- data
+				}
+				return
+			}
+			end := start + req.Limit
+			if end > len(allTasks) {
+				end = len(allTasks)
+			}
+			tasks := allTasks[start:end]
+			if data, err := json.Marshal(APIClientWSMessage{Type: "tasks", Data: map[string]interface{}{
+				"list":  tasks,
+				"total": len(allTasks),
+			}}); err == nil {
+				client.Send <- data
+			}
+		}
+	}
 	logger := parent_logger.With().Str("Client", "api_client").Logger()
 	client := &APIClient{
 		downloader: downloader,
diff --git a/internal/api/handler.go b/internal/api/handler.go
@@ -11,6 +11,7 @@ import (
 	"net/url"
 	"os"
 	"path/filepath"
+	"sort"
 	"strconv"
 	"strings"
 	"time"
@@ -269,10 +270,15 @@ func (c *APIClient) handleCreateFeedDownloadTask(ctx *gin.Context) {
 		result.Err(ctx, 500, "创建任务失败："+err.Error())
 		return
 	}
-	c.channels.Broadcast(APIClientWSMessage{
-		Type: "tasks",
-		Data: c.downloader.GetTasks(),
-	})
+	task := c.downloader.GetTask(id)
+	if task != nil {
+		c.channels.Broadcast(APIClientWSMessage{
+			Type: "event",
+			Data: map[string]interface{}{
+				"task": task,
+			},
+		})
+	}
 	result.Ok(ctx, gin.H{"id": id})
 }
 
@@ -328,6 +334,9 @@ func (c *APIClient) handleFetchTaskList(ctx *gin.Context) {
 		filter.Statuses = []base.Status{base.Status(status)}
 	}
 	list := c.downloader.GetTasksByFilter(filter)
+	sort.Slice(list, func(i, j int) bool {
+		return list[i].CreatedAt.After(list[j].CreatedAt)
+	})
 	total := len(list)
 	page_num, err := strconv.Atoi(page_str)
 	if err != nil {
@@ -429,10 +438,15 @@ func (c *APIClient) handleCreateLiveTask(ctx *gin.Context) {
 		result.Err(ctx, 500, "创建任务失败: "+err.Error())
 		return
 	}
-	c.channels.Broadcast(APIClientWSMessage{
-		Type: "tasks",
-		Data: c.downloader.GetTasks(),
-	})
+	task := c.downloader.GetTask(id)
+	if task != nil {
+		c.channels.Broadcast(APIClientWSMessage{
+			Type: "event",
+			Data: map[string]interface{}{
+				"task": task,
+			},
+		})
+	}
 	result.Ok(ctx, gin.H{"id": id})
 }
 
@@ -474,10 +488,17 @@ func (c *APIClient) handleBatchCreateTask(ctx *gin.Context) {
 	// 	Int("count", len(task.Reqs)).
 	// 	Dur("cost", time.Since(start)).
 	// 	Msg("批量创建任务完成")
-	c.channels.Broadcast(APIClientWSMessage{
-		Type: "tasks",
-		Data: c.downloader.GetTasks(),
-	})
+	for _, id := range ids {
+		task := c.downloader.GetTask(id)
+		if task != nil {
+			c.channels.Broadcast(APIClientWSMessage{
+				Type: "event",
+				Data: map[string]interface{}{
+					"task": task,
+				},
+			})
+		}
+	}
 	result.Ok(ctx, gin.H{"ids": ids})
 }
 
@@ -682,10 +703,15 @@ func (c *APIClient) handleCreateChannelsTask(ctx *gin.Context) {
 		result.Err(ctx, 500, "下载失败")
 		return
 	}
-	c.channels.Broadcast(APIClientWSMessage{
-		Type: "tasks",
-		Data: c.downloader.GetTasks(),
-	})
+	task := c.downloader.GetTask(id)
+	if task != nil {
+		c.channels.Broadcast(APIClientWSMessage{
+			Type: "event",
+			Data: map[string]interface{}{
+				"task": task,
+			},
+		})
+	}
 	result.Ok(ctx, gin.H{"id": id})
 }
 
diff --git a/internal/channels/client.go b/internal/channels/client.go
@@ -42,6 +42,7 @@ type ChannelsClient struct {
 	cache       *cache.Cache
 	req_seq     uint64
 	OnConnected func(client *Client)
+	OnMessage   func(client *Client, message []byte)
 }
 
 func NewChannelsClient() *ChannelsClient {
@@ -92,8 +93,13 @@ func (c *ChannelsClient) HandleChannelsWebsocket(ctx *gin.Context) {
 			c.requests_mu.RUnlock()
 			if ok {
 				ch <- resp
+				continue
 			}
 		}
+
+		if c.OnMessage != nil {
+			c.OnMessage(client, message)
+		}
 	}
 }
 func (c *ChannelsClient) Stop() {
diff --git a/internal/interceptor/inject/src/components.js b/internal/interceptor/inject/src/components.js
@@ -303,7 +303,7 @@ function total_speed(tasks) {
   return sum;
 }
 
-function __wx_refresh_downloader(selector, tasks) {
+function __wx_refresh_downloader(selector, tasks, total) {
   const container = document.querySelector(selector);
   if (!container) return;
   container.innerHTML = "";
@@ -313,7 +313,8 @@ function __wx_refresh_downloader(selector, tasks) {
 
   const countEl = document.getElementById("wx-dl-count");
   if (countEl) {
-    countEl.innerText = list.length > 0 ? `(${list.length})` : "";
+    const count = total !== undefined ? total : list.length;
+    countEl.innerText = count > 0 ? `(${count})` : "";
   }
 
   if (list.length === 0) {
@@ -470,12 +471,12 @@ function __wx_refresh_downloader(selector, tasks) {
   });
 
   if (list.length > 0) {
-    const footer = document.createElement("div");
-    footer.className = "weui-loadmore weui-loadmore_line";
-    footer.style.marginTop = "20px";
-    footer.innerHTML =
-      '<span class="weui-loadmore__tips">没有更多内容了</span>';
-    container.appendChild(footer);
+    // const footer = document.createElement("div");
+    // footer.className = "weui-loadmore weui-loadmore_line";
+    // footer.style.marginTop = "20px";
+    // footer.innerHTML =
+    //   '<span class="weui-loadmore__tips">没有更多内容了</span>';
+    // container.appendChild(footer);
   }
 }
 
diff --git a/internal/interceptor/inject/src/downloader.js b/internal/interceptor/inject/src/downloader.js
diff --git a/internal/interceptor/inject/src/downloaderv2.js b/internal/interceptor/inject/src/downloaderv2.js
diff --git a/pkg/util/fs.go b/pkg/util/fs.go

Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,7 @@ type ChannelsClient struct {`
`42`	`42`	`cache *cache.Cache`
`43`	`43`	`req_seq uint64`
`44`	`44`	`OnConnected func(client *Client)`
	`45`	`+ OnMessage func(client *Client, message []byte)`
`45`	`46`	`}`
`46`	`47`
`47`	`48`	`func NewChannelsClient() *ChannelsClient {`
`@@ -92,8 +93,13 @@ func (c ChannelsClient) HandleChannelsWebsocket(ctx gin.Context) {`
`92`	`93`	`c.requests_mu.RUnlock()`
`93`	`94`	`if ok {`
`94`	`95`	`ch <- resp`
	`96`	`+ continue`
`95`	`97`	`}`
`96`	`98`	`}`
	`99`	`+`
	`100`	`+ if c.OnMessage != nil {`
	`101`	`+ c.OnMessage(client, message)`
	`102`	`+ }`
`97`	`103`	`}`
`98`	`104`	`}`
`99`	`105`	`func (c *ChannelsClient) Stop() {`