diff --git a/backend/models/episode_torrents.go b/backend/models/episode_torrents.go index 0d3449a..da2c039 100644 --- a/backend/models/episode_torrents.go +++ b/backend/models/episode_torrents.go @@ -11,13 +11,13 @@ import ( const ( upsertEpisodeTorrentQuery = ` 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, - :seeders, :leechers) + :seeders, :leechers, :size) ON CONFLICT (imdb_id, season, episode, quality, source) DO UPDATE SET imdb_id=:imdb_id, url=:url, source=:source, quality=:quality, upload_user=:upload_user, season=:season, episode=:episode, - seeders=:seeders, leechers=:leechers + seeders=:seeders, leechers=:leechers, size=:size RETURNING id;` getEpisodeTorrentQuery = ` @@ -37,6 +37,7 @@ type episodeTorrentDB struct { Episode int `db:"episode"` Seeders int `db:"seeders"` Leechers int `db:"leechers"` + Size int `db:"size"` Created time.Time `db:"created_at"` Updated time.Time `db:"updated_at"` } @@ -52,6 +53,7 @@ func NewTorrentFromEpisodeTorrentDB(eDB *episodeTorrentDB) *polochon.Torrent { Leechers: eDB.Leechers, Source: eDB.Source, UploadUser: eDB.UploadUser, + Size: eDB.Size, } } @@ -68,6 +70,7 @@ func NewEpisodeTorrentDB(t *polochon.Torrent, imdbID string, season, episode int UploadUser: t.UploadUser, Seeders: t.Seeders, Leechers: t.Leechers, + Size: t.Size, } } diff --git a/backend/models/movie_torrents.go b/backend/models/movie_torrents.go index a62f945..7d5602b 100644 --- a/backend/models/movie_torrents.go +++ b/backend/models/movie_torrents.go @@ -11,12 +11,12 @@ import ( const ( upsertMovieTorrentQuery = ` 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, - :leechers) + :leechers, :size) ON CONFLICT (imdb_id, quality, source) 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;` getMovieTorrentQueryByImdbID = ` @@ -34,6 +34,7 @@ type movieTorrentDB struct { UploadUser string `db:"upload_user"` Seeders int `db:"seeders"` Leechers int `db:"leechers"` + Size int `db:"size"` Created time.Time `db:"created_at"` Updated time.Time `db:"updated_at"` } @@ -49,6 +50,7 @@ func NewTorrentFromMovieTorrentDB(mDB *movieTorrentDB) *polochon.Torrent { Leechers: mDB.Leechers, Source: mDB.Source, UploadUser: mDB.UploadUser, + Size: mDB.Size, } } @@ -63,6 +65,7 @@ func NewMovieTorrentDB(t *polochon.Torrent, imdbID string) movieTorrentDB { UploadUser: t.UploadUser, Seeders: t.Seeders, Leechers: t.Leechers, + Size: t.Size, } } diff --git a/migrations/0009_torrents_size.down.sql b/migrations/0009_torrents_size.down.sql new file mode 100644 index 0000000..9076ff9 --- /dev/null +++ b/migrations/0009_torrents_size.down.sql @@ -0,0 +1,2 @@ +ALTER TABLE movie_torrents DROP COLUMN size; +ALTER TABLE episode_torrents DROP COLUMN size; diff --git a/migrations/0009_torrents_size.up.sql b/migrations/0009_torrents_size.up.sql new file mode 100644 index 0000000..0f02504 --- /dev/null +++ b/migrations/0009_torrents_size.up.sql @@ -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;