diff options
Diffstat (limited to 'database/db.go')
| -rw-r--r-- | database/db.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/database/db.go b/database/db.go index 6b579dd9..2e468587 100644 --- a/database/db.go +++ b/database/db.go @@ -38,6 +38,7 @@ func initModels() error { &model.InboundClientIps{}, &xray.ClientTraffic{}, &model.HistoryOfSeeders{}, + &model.CustomGeoResource{}, } for _, model := range models { if err := db.AutoMigrate(model); err != nil { @@ -175,9 +176,8 @@ func GetDB() *gorm.DB { return db } -// IsNotFound checks if the given error is a GORM record not found error. func IsNotFound(err error) bool { - return err == gorm.ErrRecordNotFound + return errors.Is(err, gorm.ErrRecordNotFound) } // IsSQLiteDB checks if the given file is a valid SQLite database by reading its signature. |