From 96b8d20f7b97dfa572647cc8bf23d233f3e69a81 Mon Sep 17 00:00:00 2001 From: clz Date: Sat, 10 Jun 2023 15:18:44 +0800 Subject: [PATCH] refactor: refactor sql --- internal/label/reposiroty.go | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/internal/label/reposiroty.go b/internal/label/reposiroty.go index 16bf7af..9bd5ab5 100644 --- a/internal/label/reposiroty.go +++ b/internal/label/reposiroty.go @@ -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 }