blob: c180bb6a8cbc5d76c45fa2d98859afcc1ac0598b (
plain)
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
|
package seed
import (
"dove/database"
"dove/models/domain"
"dove/utils/logger"
)
func Run() {
seededCount := 0
for _, tldName := range DefaultTLDs.All() {
if tldExists(tldName) {
continue
}
if seedTLD(tldName) {
seededCount++
}
}
if seededCount > 0 {
logger.Successf(LogPrefix, SeedComplete, seededCount)
}
}
func tldExists(name string) bool {
var count int64
database.DB.Model(&domain.TLD{}).Where("name = ?", name).Count(&count)
return count > 0
}
func seedTLD(name string) bool {
tld := &domain.TLD{
Name: name,
IsDefault: true,
}
if createError := database.DB.Create(tld).Error; createError != nil {
logger.Errorf(LogPrefix, SeedFailed, name, createError)
return false
}
return true
}
|
