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) { func (s *sqliteRepository) GetLabels(ctx context.Context) ([]Label, error) {
var labels []Label var labels []Label
res := s.Find(&labels) if err := s.DB.Find(&labels).Error; err != nil {
if res.Error != nil { return nil, err
return nil, res.Error
} }
return labels, nil return labels, nil
} }
func (s *sqliteRepository) GetLabelById(ctx context.Context, id int) (*Label, error) { func (s *sqliteRepository) GetLabelById(ctx context.Context, id int) (*Label, error) {
var label Label var label Label
res := s.First(&label, id) if err := s.DB.Where("id =?", id).First(&label).Error; err != nil {
if res.Error != nil { return nil, err
return nil, res.Error
} }
return &label, nil 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 { func (s *sqliteRepository) DeleteLabel(ctx context.Context, id int) error {
return s.Delete(&Label{}, id).Error return s.Where("id =?", id).Delete(&Label{}).Error
} }