jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-04-21 19:27:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

db optimize

Browse Source
This commit is contained in:
Simon Ding
2024-07-06 11:56:53 +08:00
parent 7d9dc17f87
commit 1fef91fe66
30 changed files with 2289 additions and 1573 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
db/db.go
View File

@@ -65,7 +65,7 @@ func (c *Client) GetLanguage() string {
return lang
}
func (c *Client) AddWatchlist(path string, detail *tmdb.TVDetails) (*ent.Series, error) {
func (c *Client) AddWatchlist(path string, detail *tmdb.TVDetails, episodes []int) (*ent.Series, error) {
count := c.ent.Series.Query().Where(series.TmdbID(int(detail.ID))).CountX(context.Background())
if (count > 0) {
return nil, fmt.Errorf("tv series %s already in watchlist", detail.Name)
@@ -74,9 +74,10 @@ func (c *Client) AddWatchlist(path string, detail *tmdb.TVDetails) (*ent.Series,
SetTmdbID(int(detail.ID)).
SetPath(path).
SetOverview(detail.Overview).
SetTitle(detail.Name).
SetName(detail.Name).
SetOriginalName(detail.OriginalName).
SetPosterPath(detail.PosterPath).
AddEpisodeIDs(episodes...).
Save(context.TODO())
return r, err
}
@@ -90,16 +91,15 @@ func (c *Client) GetWatchlist() []*ent.Series {
return list
}
func (c *Client) SaveEposideDetail(d *ent.Epidodes) error {
_, err := c.ent.Epidodes.Create().
func (c *Client) SaveEposideDetail(d *ent.Episode) (int, error) {
ep, err := c.ent.Episode.Create().
SetAirDate(d.AirDate).
SetSeasonNumber(d.SeasonNumber).
SetEpisodeNumber(d.EpisodeNumber).
SetSeriesID(d.SeriesID).
SetOverview(d.Overview).
SetTitle(d.Title).Save(context.TODO())
return err
return ep.ID,err
}
type TorznabSetting struct {