42 lines
773 B
Go
42 lines
773 B
Go
package filter
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hireme/hireme/internal/db"
|
|
)
|
|
|
|
type FilterResult struct {
|
|
Keep bool
|
|
Score float64
|
|
Reason string
|
|
}
|
|
|
|
type JobFilter interface {
|
|
Name() string
|
|
Process(ctx context.Context, job *db.RawJob) (FilterResult, error)
|
|
}
|
|
|
|
type FilterPipeline struct {
|
|
filters []JobFilter
|
|
}
|
|
|
|
func NewPipeline(filters ...JobFilter) *FilterPipeline {
|
|
return &FilterPipeline{filters: filters}
|
|
}
|
|
|
|
func (p *FilterPipeline) Execute(ctx context.Context, job *db.RawJob) (bool, float64, string) {
|
|
var totalScore float64
|
|
for _, f := range p.filters {
|
|
res, err := f.Process(ctx, job)
|
|
if err != nil {
|
|
return false, 0, err.Error()
|
|
}
|
|
if !res.Keep {
|
|
return false, 0, res.Reason
|
|
}
|
|
totalScore += res.Score
|
|
}
|
|
return true, totalScore, ""
|
|
}
|