ODWRTW/canape
2
2
Fork 0
Code Issues 3 Pull Requests 1 Releases Activity

Add torrent size in backend

Browse Source
This commit is contained in:
Lucas BEE 2019-07-26 10:48:07 +00:00
parent c48247daac
commit 89a2de2ce0
4 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/models/episode_torrents.go
View File

@ -11,13 +11,13 @@ import (
const ( const (
upsertEpisodeTorrentQuery = ` upsertEpisodeTorrentQuery = `
INSERT INTO episode_torrents (imdb_id, url, source, quality, upload_user, INSERT INTO episode_torrents (imdb_id, url, source, quality, upload_user,
season, episode, seeders, leechers) season, episode, seeders, leechers, size)
VALUES (:imdb_id, :url, :source, :quality, :upload_user, :season, :episode, VALUES (:imdb_id, :url, :source, :quality, :upload_user, :season, :episode,
:seeders, :leechers) :seeders, :leechers, :size)
ON CONFLICT (imdb_id, season, episode, quality, source) ON CONFLICT (imdb_id, season, episode, quality, source)
DO UPDATE SET imdb_id=:imdb_id, url=:url, source=:source, quality=:quality, DO UPDATE SET imdb_id=:imdb_id, url=:url, source=:source, quality=:quality,
upload_user=:upload_user, season=:season, episode=:episode, upload_user=:upload_user, season=:season, episode=:episode,
seeders=:seeders, leechers=:leechers seeders=:seeders, leechers=:leechers, size=:size
RETURNING id;` RETURNING id;`
getEpisodeTorrentQuery = ` getEpisodeTorrentQuery = `
@ -37,6 +37,7 @@ type episodeTorrentDB struct {
Episode int `db:"episode"` Episode int `db:"episode"`
Seeders int `db:"seeders"` Seeders int `db:"seeders"`
Leechers int `db:"leechers"` Leechers int `db:"leechers"`
Size int `db:"size"`
Created time.Time `db:"created_at"` Created time.Time `db:"created_at"`
Updated time.Time `db:"updated_at"` Updated time.Time `db:"updated_at"`
} }
@ -52,6 +53,7 @@ func NewTorrentFromEpisodeTorrentDB(eDB *episodeTorrentDB) *polochon.Torrent {
Leechers: eDB.Leechers, Leechers: eDB.Leechers,
Source: eDB.Source, Source: eDB.Source,
UploadUser: eDB.UploadUser, UploadUser: eDB.UploadUser,
Size: eDB.Size,
} }
} }
@ -68,6 +70,7 @@ func NewEpisodeTorrentDB(t *polochon.Torrent, imdbID string, season, episode int
UploadUser: t.UploadUser, UploadUser: t.UploadUser,
Seeders: t.Seeders, Seeders: t.Seeders,
Leechers: t.Leechers, Leechers: t.Leechers,
Size: t.Size,
} }
} }

9
backend/models/movie_torrents.go
View File

@ -11,12 +11,12 @@ import (
const ( const (
upsertMovieTorrentQuery = ` upsertMovieTorrentQuery = `
INSERT INTO movie_torrents (imdb_id, url, source, quality, upload_user, INSERT INTO movie_torrents (imdb_id, url, source, quality, upload_user,
seeders, leechers) seeders, leechers, size)
VALUES (:imdb_id, :url, :source, :quality, :upload_user, :seeders, VALUES (:imdb_id, :url, :source, :quality, :upload_user, :seeders,
:leechers) :leechers, :size)
ON CONFLICT (imdb_id, quality, source) ON CONFLICT (imdb_id, quality, source)
DO UPDATE SET imdb_id=:imdb_id, url=:url, source=:source, quality=:quality, DO UPDATE SET imdb_id=:imdb_id, url=:url, source=:source, quality=:quality,
upload_user=:upload_user, seeders=:seeders, leechers=:leechers upload_user=:upload_user, seeders=:seeders, leechers=:leechers, size=:size
RETURNING id;` RETURNING id;`
getMovieTorrentQueryByImdbID = ` getMovieTorrentQueryByImdbID = `
@ -34,6 +34,7 @@ type movieTorrentDB struct {
UploadUser string `db:"upload_user"` UploadUser string `db:"upload_user"`
Seeders int `db:"seeders"` Seeders int `db:"seeders"`
Leechers int `db:"leechers"` Leechers int `db:"leechers"`
Size int `db:"size"`
Created time.Time `db:"created_at"` Created time.Time `db:"created_at"`
Updated time.Time `db:"updated_at"` Updated time.Time `db:"updated_at"`
} }
@ -49,6 +50,7 @@ func NewTorrentFromMovieTorrentDB(mDB *movieTorrentDB) *polochon.Torrent {
Leechers: mDB.Leechers, Leechers: mDB.Leechers,
Source: mDB.Source, Source: mDB.Source,
UploadUser: mDB.UploadUser, UploadUser: mDB.UploadUser,
Size: mDB.Size,
} }
} }
@ -63,6 +65,7 @@ func NewMovieTorrentDB(t *polochon.Torrent, imdbID string) movieTorrentDB {
UploadUser: t.UploadUser, UploadUser: t.UploadUser,
Seeders: t.Seeders, Seeders: t.Seeders,
Leechers: t.Leechers, Leechers: t.Leechers,
Size: t.Size,
} }
} }

2
migrations/0009_torrents_size.down.sql Normal file
View File

@ -0,0 +1,2 @@
ALTER TABLE movie_torrents DROP COLUMN size;
ALTER TABLE episode_torrents DROP COLUMN size;

2
migrations/0009_torrents_size.up.sql Normal file
View File

@ -0,0 +1,2 @@
ALTER TABLE movie_torrents ADD COLUMN size bigint NOT NULL DEFAULT 0;
ALTER TABLE episode_torrents ADD COLUMN size bigint NOT NULL DEFAULT 0;