Chore: project init
This commit is contained in:
24
internal/download/repository.go
Normal file
24
internal/download/repository.go
Normal file
@@ -0,0 +1,24 @@
|
||||
package download
|
||||
|
||||
import "gorm.io/gorm"
|
||||
|
||||
type Repository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewRepository(db *gorm.DB) *Repository {
|
||||
return &Repository{db: db}
|
||||
}
|
||||
|
||||
func (r *Repository) GetLatest() (*Info, error) {
|
||||
var info Info
|
||||
err := r.db.Last(&info).Error
|
||||
return &info, err
|
||||
}
|
||||
|
||||
func (r *Repository) Save(info *Info) error {
|
||||
if info.ID == 0 {
|
||||
return r.db.Create(info).Error
|
||||
}
|
||||
return r.db.Save(info).Error
|
||||
}
|
||||
Reference in New Issue
Block a user