refactor: use db selector

penguin-statistics · Feb 10, 2023 · 1cc9a59 · 1cc9a59
1 parent 7bb753d
commit 1cc9a59
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/internal/repo/time_range.go b/internal/repo/time_range.go
@@ -6,14 +6,16 @@ import (
 	"github.com/uptrace/bun"
 
 	"exusiai.dev/backend-next/internal/model"
+	"exusiai.dev/backend-next/internal/repo/selector"
 )
 
 type TimeRange struct {
-	db *bun.DB
+	db  *bun.DB
+	sel selector.S[model.TimeRange]
 }
 
 func NewTimeRange(db *bun.DB) *TimeRange {
-	return &TimeRange{db: db}
+	return &TimeRange{db: db, sel: selector.New[model.TimeRange](db)}
 }
 
 func (r *TimeRange) GetTimeRangesByServer(ctx context.Context, server string) ([]*model.TimeRange, error) {

diff --git a/internal/repo/trend_element.go b/internal/repo/trend_element.go
@@ -8,14 +8,16 @@ import (
 	"github.com/uptrace/bun"
 
 	"exusiai.dev/backend-next/internal/model"
+	"exusiai.dev/backend-next/internal/repo/selector"
 )
 
 type TrendElement struct {
-	db *bun.DB
+	db  *bun.DB
+	sel selector.S[model.TrendElement]
 }
 
 func NewTrendElement(db *bun.DB) *TrendElement {
-	return &TrendElement{db: db}
+	return &TrendElement{db: db, sel: selector.New[model.TrendElement](db)}
 }
 
 func (r *TrendElement) BatchSaveElements(ctx context.Context, elements []*model.TrendElement, server string) error {