ddns-updater/internal/data/data.go

package data

import (
	"sync"

	"github.com/qdm12/ddns-updater/internal/models"
	"github.com/qdm12/ddns-updater/internal/persistence"
	"github.com/qdm12/ddns-updater/internal/records"
)

type Database interface {
	Close() error
	Insert(record records.Record) (id int)
	Select(id int) (record records.Record, err error)
	SelectAll() (records []records.Record)
	Update(id int, record records.Record) error
	// From persistence database
	GetEvents(domain, host string) (events []models.HistoryEvent, err error)
}

type database struct {
	data []records.Record
	sync.RWMutex
	persistentDB persistence.Database
}

// NewDatabase creates a new in memory database
func NewDatabase(data []records.Record, persistentDB persistence.Database) Database {
	return &database{
		data:         data,
		persistentDB: persistentDB,
	}
}

func (db *database) Close() error {
	db.Lock() // ensure write operation finishes
	defer db.Unlock()
	return db.persistentDB.Close()
}