-
Notifications
You must be signed in to change notification settings - Fork 0
/
domain.go
103 lines (90 loc) · 2.32 KB
/
domain.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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package repository
import (
"github.com/aerosystems/checkmail-service/internal/models"
"gorm.io/gorm"
)
type DomainRepo struct {
db *gorm.DB
}
func NewDomainRepo(db *gorm.DB) *DomainRepo {
return &DomainRepo{
db: db,
}
}
func (r *DomainRepo) FindAll() (*[]models.Domain, error) {
var domains []models.Domain
result := r.db.Find(&domains)
if result.Error != nil {
return nil, result.Error
}
return &domains, nil
}
func (r *DomainRepo) FindByID(id int) (*models.Domain, error) {
var domain models.Domain
result := r.db.First(&domain, id)
if result.Error != nil {
return nil, result.Error
}
return &domain, nil
}
func (r *DomainRepo) FindByName(name string) (*models.Domain, error) {
var domain models.Domain
result := r.db.First(&domain, "name = ?", name)
if result.Error != nil {
return nil, result.Error
}
return &domain, nil
}
func (r *DomainRepo) Create(domain *models.Domain) error {
result := r.db.Create(&domain)
if result.Error != nil {
return result.Error
}
return nil
}
func (r *DomainRepo) Update(domain *models.Domain) error {
result := r.db.Save(&domain)
if result.Error != nil {
return result.Error
}
return nil
}
func (r *DomainRepo) Delete(domain *models.Domain) error {
result := r.db.Delete(&domain)
if result.Error != nil {
return result.Error
}
return nil
}
func (r *DomainRepo) MatchEquals(name string) (*models.Domain, error) {
var domain models.Domain
result := r.db.First(&domain, "name = ? AND coverage = ?", name, "equals")
if result.Error != nil {
return nil, result.Error
}
return &domain, nil
}
func (r *DomainRepo) MatchContains(name string) (*models.Domain, error) {
var domain models.Domain
result := r.db.First(&domain, "name LIKE ? AND coverage = ?", "%"+name+"%", "contains")
if result.Error != nil {
return nil, result.Error
}
return &domain, nil
}
func (r *DomainRepo) MatchBegins(name string) (*models.Domain, error) {
var domain models.Domain
result := r.db.First(&domain, "name LIKE ? AND coverage = ?", name+"%", "begins")
if result.Error != nil {
return nil, result.Error
}
return &domain, nil
}
func (r *DomainRepo) MatchEnds(name string) (*models.Domain, error) {
var domain models.Domain
result := r.db.First(&domain, "name LIKE ? AND coverage = ?", "%"+name, "ends")
if result.Error != nil {
return nil, result.Error
}
return &domain, nil
}