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

refactor: db code

Browse Source
This commit is contained in:
Simon Ding
2025-04-22 15:39:47 +08:00
parent 26dc488d48
commit 7ccc73e044
8 changed files with 181 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
engine/client.go
View File

@@ -16,7 +16,7 @@ import (
"github.com/robfig/cron"
)
func NewEngine(db *db.Client, language string) *Engine {
func NewEngine(db db.Database, language string) *Engine {
return &Engine{
db: db,
cron: cron.New(),
@@ -31,7 +31,7 @@ type scheduler struct {
f func() error
}
type Engine struct {
db *db.Client
db db.Database
cron *cron.Cron
tasks utils.Map[int, *Task]
language string

6
engine/torrent.go
View File

@@ -71,7 +71,7 @@ func names2Query(media *ent.Media) []string {
return names
}
func SearchTvSeries(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
func SearchTvSeries(db1 db.Database, param *SearchParam) ([]torznab.Result, error) {
series, err := db1.GetMediaDetails(param.MediaId)
if err != nil {
return nil, fmt.Errorf("no tv series of id %v: %v", param.MediaId, err)
@@ -234,7 +234,7 @@ func isNoSeasonSeries(detail *db.MediaDetails) bool {
return hasSeason2 && !season2HasEpisode1 //only one 1st episode
}
func SearchMovie(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
func SearchMovie(db1 db.Database, param *SearchParam) ([]torznab.Result, error) {
movieDetail, err := db1.GetMediaDetails(param.MediaId)
if err != nil {
return nil, err
@@ -310,7 +310,7 @@ const (
SearchTypeMovie SearchType = 2
)
func searchWithTorznab(db *db.Client, t SearchType, queries ...string) []torznab.Result {
func searchWithTorznab(db db.Database, t SearchType, queries ...string) []torznab.Result {
var res []torznab.Result
allTorznab := db.GetAllIndexers()