forked from indes/flowerss-bot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
source.go
74 lines (62 loc) · 1.94 KB
/
source.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package storage
import (
"context"
"errors"
"gorm.io/gorm"
"github.com/nerdneilsfield/flowerss-bot/internal/log"
"github.com/nerdneilsfield/flowerss-bot/internal/model"
)
type SourceStorageImpl struct {
db *gorm.DB
}
func NewSourceStorageImpl(db *gorm.DB) *SourceStorageImpl {
return &SourceStorageImpl{db: db}
}
func (s *SourceStorageImpl) Init(ctx context.Context) error {
return s.db.Migrator().AutoMigrate(&model.Source{})
}
func (s *SourceStorageImpl) AddSource(ctx context.Context, source *model.Source) error {
result := s.db.WithContext(ctx).Create(source)
if result.Error != nil {
return result.Error
}
return nil
}
func (s *SourceStorageImpl) GetSource(ctx context.Context, id uint) (*model.Source, error) {
var source = &model.Source{}
result := s.db.WithContext(ctx).Where(&model.Source{ID: id}).First(source)
if result.Error != nil {
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
return nil, ErrRecordNotFound
}
return nil, result.Error
}
return source, nil
}
func (s *SourceStorageImpl) GetSourceByURL(ctx context.Context, url string) (*model.Source, error) {
var source = &model.Source{}
result := s.db.WithContext(ctx).Where(&model.Source{Link: url}).First(source)
if result.Error != nil {
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
return nil, ErrRecordNotFound
}
return nil, result.Error
}
return source, nil
}
func (s *SourceStorageImpl) Delete(ctx context.Context, id uint) error {
result := s.db.WithContext(ctx).Where("id = ?", id).Delete(&model.Source{})
if result.Error != nil {
return result.Error
}
return nil
}
func (s *SourceStorageImpl) UpsertSource(ctx context.Context, sourceID uint, newSource *model.Source) error {
newSource.ID = sourceID
result := s.db.WithContext(ctx).Where("id = ?", sourceID).Save(newSource)
if result.Error != nil {
return result.Error
}
log.Debugf("update %d row, sourceID %d new %#v", result.RowsAffected, sourceID, newSource)
return nil
}