jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-05-29 22:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: db

Browse Source
This commit is contained in:
Simon Ding
2025-02-02 13:09:22 +08:00
parent 88492b3922
commit 2a9fdd2a13
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
db/db.go
View File

@@ -17,6 +17,7 @@ import (
"polaris/ent/storage" "polaris/ent/storage"
"polaris/log" "polaris/log"
"polaris/pkg/utils" "polaris/pkg/utils"
"slices"
"strings" "strings"
"time" "time"
@@ -462,9 +463,9 @@ func (c *Client) SetDefaultStorageByName(name string) error {
} }
func (c *Client) SaveHistoryRecord(h ent.History) (*ent.History, error) { func (c *Client) SaveHistoryRecord(h ent.History) (*ent.History, error) {
return c.ent.History.Create().SetMediaID(h.MediaID).SetEpisodeID(h.EpisodeID).SetDate(time.Now()). return c.ent.History.Create().SetMediaID(h.MediaID).SetDate(time.Now()).
SetStatus(h.Status).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetIndexerID(h.IndexerID). SetStatus(h.Status).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetIndexerID(h.IndexerID).
SetDownloadClientID(h.DownloadClientID).SetSize(h.Size).SetSaved(h.Saved).SetSeasonNum(h.SeasonNum). SetDownloadClientID(h.DownloadClientID).SetSize(h.Size).SetSeasonNum(h.SeasonNum).
SetEpisodeNums(h.EpisodeNums).SetHash(h.Hash).SetLink(h.Link).Save(context.TODO()) SetEpisodeNums(h.EpisodeNums).SetHash(h.Hash).SetLink(h.Link).Save(context.TODO())
} }
@@ -520,8 +521,12 @@ func (c *Client) SetEpisodeStatus(id int, status episode.Status) error {
} }
func (c *Client) IsEpisodeDownloadingOrDownloaded(id int) bool { func (c *Client) IsEpisodeDownloadingOrDownloaded(id int) bool {
his := c.ent.History.Query().Where(history.EpisodeID(id)).AllX(context.Background()) ep, _ := c.GetEpisodeByID(id)
his := c.ent.History.Query().Where(history.EpisodeNumsNotNil()).AllX(context.Background())
for _, h := range his { for _, h := range his {
if !slices.Contains(h.EpisodeNums, ep.EpisodeNumber) {
continue
}
if h.Status != history.StatusFail { if h.Status != history.StatusFail {
return true return true
} }