app/attribute/attribute.go

package attribute

import (
	"time"

	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"kumoly.io/kumoly/app/store"
)

type Attribute struct {
	ID        uint `gorm:"primaryKey"`
	CreatedAt time.Time
	UpdatedAt time.Time

	Key         string `gorm:"unique;not null"`
	Description string
	Default     string
	Value       string
}

var l zerolog.Logger

func Init() {
	l = log.With().Str("mod", "attribute").Logger()

	l.Debug().Str("service", "attribute.Service").
		Msg("Migrating database for attribute.Service ...")
	err := store.Migrate(&Attribute{})
	if err != nil {
		panic(err)
	}
}

func Add(Key, Description, Default string) error {
	result := store.DB.Exec(`select "value" from "attributes" where "key" = ?`, Key)
	if result.RowsAffected != 0 {
		return nil
	}
	a := &Attribute{
		Key:         Key,
		Description: Description,
		Default:     Default,
		Value:       Default,
	}
	return store.DB.Create(a).Error
}

func RestoreDefault(key string) error {
	result := store.DB.Exec(`update "attributes" set value = (
	select "default" from "attributes" where "key" = ?
) where "key" = ?`, key, key)
	if result.RowsAffected == 0 {
		return ErrorAttributeNotFound
	}
	return result.Error
}

func Get(key string) string {
	var value string
	store.DB.
		Raw(`select "value" from "attributes" where "key" = ?`, key).
		Scan(&value)
	return value
}

func Set(key, value string) error {
	result := store.DB.Exec(`update "attributes" set value = ? where "key" = ?`,
		value, key)
	if result.RowsAffected == 0 {
		return ErrorAttributeNotFound
	}
	return result.Error
}