diff --git a/db/db.go b/db/db.go index 2bd23bd..14a93fd 100644 --- a/db/db.go +++ b/db/db.go @@ -472,7 +472,8 @@ func (c *Client) SetDefaultStorageByName(name string) 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()). - SetStatus(h.Status).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetSaved(h.Saved).Save(context.TODO()) + SetStatus(h.Status).SetTargetDir(h.TargetDir).SetSourceTitle(h.SourceTitle).SetIndexerID(h.IndexerID). + SetSaved(h.Saved).Save(context.TODO()) } func (c *Client) SetHistoryStatus(id int, status history.Status) error { diff --git a/server/core/scheduler.go b/server/core/scheduler.go index b165412..e3736b2 100644 --- a/server/core/scheduler.go +++ b/server/core/scheduler.go @@ -276,6 +276,7 @@ func (c *Client) downloadMovieSingleEpisode(ep *ent.Episode) error { Size: r1.Size, Saved: torrent.Save(), DownloadClientID: dlc.ID, + IndexerID: r1.IndexerId, }) if err != nil { log.Errorf("save history error: %v", err) @@ -336,7 +337,7 @@ func (c *Client) checkSeiesNewSeason(media *ent.Media) error { return nil } -func (c *Client) isSeedRatioLimitReached(indexId int, t pkg.Torrent) (bool) { +func (c *Client) isSeedRatioLimitReached(indexId int, t pkg.Torrent) bool { indexer, err := c.db.GetIndexer(indexId) if err != nil { return true