twitch-bot/internal/apimodules/customevent/database.go

package customevent

import (
	"bytes"
	"encoding/json"
	"time"

	"github.com/gofrs/uuid/v3"
	"github.com/pkg/errors"
	"gorm.io/gorm"

	"github.com/Luzifer/twitch-bot/v3/internal/helpers"
	"github.com/Luzifer/twitch-bot/v3/pkg/database"
	"github.com/Luzifer/twitch-bot/v3/plugins"
)

const cleanupTimeout = 15 * time.Minute

type (
	storedCustomEvent struct {
		ID          string `gorm:"primaryKey"`
		Channel     string
		Fields      string
		ScheduledAt time.Time
	}
)

func cleanupStoredEvents(db database.Connector) error {
	return errors.Wrap(
		helpers.RetryTransaction(db.DB(), func(tx *gorm.DB) error {
			return tx.Where("scheduled_at < ?", time.Now().Add(cleanupTimeout*-1).UTC()).
				Delete(&storedCustomEvent{}).
				Error
		}),
		"deleting past events",
	)
}

func getFutureEvents(db database.Connector) (out []storedCustomEvent, err error) {
	return out, errors.Wrap(
		helpers.Retry(func() error {
			return db.DB().
				Where("scheduled_at >= ?", time.Now().UTC()).
				Find(&out).
				Error
		}),
		"getting events from database",
	)
}

func storeEvent(db database.Connector, scheduleAt time.Time, channel string, fields *plugins.FieldCollection) error {
	fieldBuf := new(bytes.Buffer)
	if err := json.NewEncoder(fieldBuf).Encode(fields); err != nil {
		return errors.Wrap(err, "marshalling fields")
	}

	return errors.Wrap(
		helpers.RetryTransaction(db.DB(), func(tx *gorm.DB) error {
			return tx.Create(storedCustomEvent{
				ID:          uuid.Must(uuid.NewV4()).String(),
				Channel:     channel,
				Fields:      fieldBuf.String(),
				ScheduledAt: scheduleAt,
			}).Error
		}),
		"storing event",
	)
}