marco/yt-dlp-webui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use message queue to restore download from session file

Browse Source
This commit is contained in:
marcobaobao
2024-05-13 11:15:08 +02:00
parent 6abfb57598
commit 62eadb52a6
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
server/internal/memory_db.go
View File

@@ -88,7 +88,7 @@ func (m *MemoryDB) Persist() error {
} }
// Restore a persisted state // Restore a persisted state
func (m *MemoryDB) Restore(logger *slog.Logger) { func (m *MemoryDB) Restore(mq *MessageQueue, logger *slog.Logger) {
fd, err := os.Open("session.dat") fd, err := os.Open("session.dat")
if err != nil { if err != nil {
return return
@@ -114,7 +114,7 @@ func (m *MemoryDB) Restore(logger *slog.Logger) {
m.table.Store(proc.Id, restored) m.table.Store(proc.Id, restored)
if restored.Progress.Percentage != "-1" { if restored.Progress.Percentage != "-1" {
go restored.Start() mq.Publish(restored)
} }
} }
} }

4
server/server.go
View File

@@ -77,8 +77,6 @@ func RunBlocking(cfg *RunConfig) {
slog.NewTextHandler(io.MultiWriter(logWriters...), &slog.HandlerOptions{}), slog.NewTextHandler(io.MultiWriter(logWriters...), &slog.HandlerOptions{}),
) )
mdb.Restore(logger)
db, err := sql.Open("sqlite", cfg.DBPath) db, err := sql.Open("sqlite", cfg.DBPath)
if err != nil { if err != nil {
logger.Error("failed to open database", slog.String("err", err.Error())) logger.Error("failed to open database", slog.String("err", err.Error()))
@@ -90,7 +88,9 @@ func RunBlocking(cfg *RunConfig) {
} }
mq := internal.NewMessageQueue(logger) mq := internal.NewMessageQueue(logger)
go mq.Subscriber() go mq.Subscriber()
go mdb.Restore(mq, logger)
srv := newServer(serverConfig{ srv := newServer(serverConfig{
frontend: cfg.App, frontend: cfg.App,