clz/bill-server-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: refactor sql

Browse Source
This commit is contained in:
clz 2023-06-10 15:18:44 +08:00
parent b5e9dfa36b
commit 96b8d20f7b
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
internal/label/reposiroty.go
View File

@ -17,20 +17,17 @@ func NewLabelRepository(db *gorm.DB) LabelRepository {
func (s *sqliteRepository) GetLabels(ctx context.Context) ([]Label, error) {
var labels []Label
res := s.Find(&labels)
if res.Error != nil {
return nil, res.Error
if err := s.DB.Find(&labels).Error; err != nil {
return nil, err
}
return labels, nil
}
func (s *sqliteRepository) GetLabelById(ctx context.Context, id int) (*Label, error) {
var label Label
res := s.First(&label, id)
if res.Error != nil {
return nil, res.Error
if err := s.DB.Where("id =?", id).First(&label).Error; err != nil {
return nil, err
}
return &label, nil
}
@ -43,5 +40,5 @@ func (s *sqliteRepository) UpdateLabel(ctx context.Context, label *Label) error
}
func (s *sqliteRepository) DeleteLabel(ctx context.Context, id int) error {
return s.Delete(&Label{}, id).Error
return s.Where("id =?", id).Delete(&Label{}).Error
}