reCoreD-UI/controllers/record.go

package controllers

import (
	"fmt"
	"reCoreD-UI/database"
	"reCoreD-UI/models"
	"strconv"

	"gorm.io/gorm"
)

type recordsDAO struct {
	database.BaseDAO[models.IRecord]
}

func CreateRecord(r models.IRecord) (models.IRecord, error) {
	if r.GetType() != models.RecordTypeSOA {
		_, err := GetDomains(r.WithOutDotTail())
		if err != nil {
			return nil, err
		}
	}

	if err := r.CheckZone(); err != nil {
		return nil, err
	}

	res, err := (recordsDAO{}).Create(database.Client, r)
	return res, err
}

func CreateRecords(rs []models.IRecord) error {
	tx := database.Client.Begin()
	for _, r := range rs {
		if err := r.CheckZone(); err != nil {
			tx.Rollback()
			return err
		}

		if _, err := (recordsDAO{}).Create(tx, r); err != nil {
			tx.Rollback()
			return err
		}
	}

	return tx.Commit().Error
}

func GetRecords(cond models.IRecord) ([]models.IRecord, error) {
	return (recordsDAO{}).GetAll(database.Client, cond)
}

func UpdateRecord(r models.IRecord) error {
	if err := r.CheckZone(); err != nil {
		return err
	}

	if _, err := (recordsDAO{}).Update(database.Client, r); err != nil {
		return err
	}
	return nil
}

func DeleteRecord(domain, id string) error {
	ID, err := strconv.Atoi(id)
	if err != nil {
		return err
	}

	tx := database.Client.Begin()
	record, err := (recordsDAO{}).GetOne(tx, &models.Record[models.RecordContentDefault]{ID: uint(ID), Zone: fmt.Sprintf("%s.", domain)})
	if err != nil {
		tx.Rollback()
		return err
	}

	if record.GetType() == models.RecordTypeSOA {
		tx.Rollback()
		return gorm.ErrRecordNotFound
	}

	if err := (recordsDAO{}).Delete(tx, record); err != nil {
		tx.Rollback()
		return err
	}

	return tx.Commit().Error
}

// for metrics
func getRecordCounts() (map[string]float64, error) {
	rows, err := (recordsDAO{}).GetAll(database.Client, &models.Record[models.RecordContentDefault]{})
	if err != nil {
		return nil, err
	}

	result := make(map[string]float64)
	for _, row := range rows {
		record := &models.Record[models.RecordContentDefault]{}
		if err := record.FromEntity(row); err != nil {
			return nil, err
		}
		result[record.Zone] += 1
	}
	return result, nil
}