guestbook!

1 files changed, 13 insertions, 1 deletions

diff --git a/database/dns.go b/database/dns.go
index 568653d..fc01347 100644
--- a/database/dns.go
+++ b/database/dns.go
@@ -20,6 +20,18 @@ type DNSRecord struct {
 	CreatedAt time.Time `json:"created_at"`
 }
 
+func CountUserDNSRecords(db *sql.DB, userID string) (int, error) {
+	log.Println("counting dns records for user", userID)
+
+	row := db.QueryRow("SELECT COUNT(*) FROM dns_records WHERE user_id = ?", userID)
+	var count int
+	err := row.Scan(&count)
+	if err != nil {
+		return 0, err
+	}
+	return count, nil
+}
+
 func GetUserDNSRecords(db *sql.DB, userID string) ([]DNSRecord, error) {
 	log.Println("getting dns records for user", userID)
 
@@ -43,7 +55,7 @@ func GetUserDNSRecords(db *sql.DB, userID string) ([]DNSRecord, error) {
 }
 
 func SaveDNSRecord(db *sql.DB, record *DNSRecord) (*DNSRecord, error) {
-	log.Println("saving dns record", record)
+	log.Println("saving dns record", record.ID)
 
 	record.CreatedAt = time.Now()
 	_, err := db.Exec("INSERT OR REPLACE INTO dns_records (id, user_id, name, type, content, ttl, internal, created_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", record.ID, record.UserID, record.Name, record.Type, record.Content, record.TTL, record.Internal, record.CreatedAt)