1 files changed, 15 insertions, 0 deletions
diff --git a/repositories/domain/domain.go b/repositories/domain/domain.go
index 406e272..5fe7778 100644
--- a/repositories/domain/domain.go
+++ b/repositories/domain/domain.go
@@ -13,6 +13,15 @@ func AllDomains() []domain.Domain {
 	return domains
 }
 
+func FindDomainByID(domainID uint) *domain.Domain {
+	var foundDomain domain.Domain
+	result := database.DB.Preload("TLD").First(&foundDomain, domainID)
+	if result.Error == gorm.ErrRecordNotFound {
+		return nil
+	}
+	return &foundDomain
+}
+
 func FindDomainByFullName(name string, tldName string) *domain.Domain {
 	var foundDomain domain.Domain
 	result := database.DB.
@@ -36,3 +45,9 @@ func UpdateDomain(updatedDomain *domain.Domain) error {
 func DeleteDomain(targetDomain *domain.Domain) error {
 	return database.DB.Delete(targetDomain).Error
 }
+
+func CountDomainsByTLDID(tldID uint) int64 {
+	var count int64
+	database.DB.Model(&domain.Domain{}).Where("tld_id = ?", tldID).Count(&count)
+	return count
+}