feat: adjust indexer priority to the same as prowlarr

This commit is contained in:
Simon Ding
2025-04-23 16:24:20 +08:00
parent 2267b0450a
commit 8b11e72427
11 changed files with 62 additions and 5 deletions

View File

@@ -78,6 +78,8 @@ var (
DefaultEnableRss bool
// DefaultPriority holds the default value on creation for the "priority" field.
DefaultPriority int
// PriorityValidator is a validator for the "priority" field. It is called by the builders before save.
PriorityValidator func(int) error
// DefaultSeedRatio holds the default value on creation for the "seed_ratio" field.
DefaultSeedRatio float32
// DefaultDisabled holds the default value on creation for the "disabled" field.

View File

@@ -273,6 +273,11 @@ func (ic *IndexersCreate) check() error {
if _, ok := ic.mutation.Priority(); !ok {
return &ValidationError{Name: "priority", err: errors.New(`ent: missing required field "Indexers.priority"`)}
}
if v, ok := ic.mutation.Priority(); ok {
if err := indexers.PriorityValidator(v); err != nil {
return &ValidationError{Name: "priority", err: fmt.Errorf(`ent: validator failed for field "Indexers.priority": %w`, err)}
}
}
return nil
}

View File

@@ -289,7 +289,20 @@ func (iu *IndexersUpdate) ExecX(ctx context.Context) {
}
}
// check runs all checks and user-defined validators on the builder.
func (iu *IndexersUpdate) check() error {
if v, ok := iu.mutation.Priority(); ok {
if err := indexers.PriorityValidator(v); err != nil {
return &ValidationError{Name: "priority", err: fmt.Errorf(`ent: validator failed for field "Indexers.priority": %w`, err)}
}
}
return nil
}
func (iu *IndexersUpdate) sqlSave(ctx context.Context) (n int, err error) {
if err := iu.check(); err != nil {
return n, err
}
_spec := sqlgraph.NewUpdateSpec(indexers.Table, indexers.Columns, sqlgraph.NewFieldSpec(indexers.FieldID, field.TypeInt))
if ps := iu.mutation.predicates; len(ps) > 0 {
_spec.Predicate = func(selector *sql.Selector) {
@@ -662,7 +675,20 @@ func (iuo *IndexersUpdateOne) ExecX(ctx context.Context) {
}
}
// check runs all checks and user-defined validators on the builder.
func (iuo *IndexersUpdateOne) check() error {
if v, ok := iuo.mutation.Priority(); ok {
if err := indexers.PriorityValidator(v); err != nil {
return &ValidationError{Name: "priority", err: fmt.Errorf(`ent: validator failed for field "Indexers.priority": %w`, err)}
}
}
return nil
}
func (iuo *IndexersUpdateOne) sqlSave(ctx context.Context) (_node *Indexers, err error) {
if err := iuo.check(); err != nil {
return _node, err
}
_spec := sqlgraph.NewUpdateSpec(indexers.Table, indexers.Columns, sqlgraph.NewFieldSpec(indexers.FieldID, field.TypeInt))
id, ok := iuo.mutation.ID()
if !ok {

View File

@@ -120,7 +120,7 @@ var (
{Name: "implementation", Type: field.TypeString},
{Name: "settings", Type: field.TypeString, Nullable: true, Default: ""},
{Name: "enable_rss", Type: field.TypeBool, Default: true},
{Name: "priority", Type: field.TypeInt, Default: 50},
{Name: "priority", Type: field.TypeInt, Default: 25},
{Name: "seed_ratio", Type: field.TypeFloat32, Nullable: true, Default: 0},
{Name: "disabled", Type: field.TypeBool, Nullable: true, Default: false},
{Name: "tv_search", Type: field.TypeBool, Nullable: true, Default: true},

View File

@@ -95,6 +95,8 @@ func init() {
indexersDescPriority := indexersFields[4].Descriptor()
// indexers.DefaultPriority holds the default value on creation for the priority field.
indexers.DefaultPriority = indexersDescPriority.Default.(int)
// indexers.PriorityValidator is a validator for the "priority" field. It is called by the builders before save.
indexers.PriorityValidator = indexersDescPriority.Validators[0].(func(int) error)
// indexersDescSeedRatio is the schema descriptor for seed_ratio field.
indexersDescSeedRatio := indexersFields[5].Descriptor()
// indexers.DefaultSeedRatio holds the default value on creation for the seed_ratio field.

View File

@@ -1,6 +1,7 @@
package schema
import (
"errors"
"time"
"entgo.io/ent"
@@ -19,7 +20,15 @@ func (Indexers) Fields() []ent.Field {
field.String("implementation"),
field.String("settings").Optional().Default("").Comment("deprecated, use api_key and url"),
field.Bool("enable_rss").Default(true),
field.Int("priority").Default(50),
field.Int("priority").Default(25).Validate(func(i int) error {
if i > 50 {
return errors.ErrUnsupported
}
if i <= 0 {
return errors.ErrUnsupported
}
return nil
}),
field.Float32("seed_ratio").Optional().Default(0).Comment("minimal seed ratio requied, before removing torrent"),
field.Bool("disabled").Optional().Default(false),
field.Bool("tv_search").Optional().Default(true),