decorator_cache.go

package ev

import (
	"fmt"
	"github.com/go-email-validator/go-email-validator/pkg/ev/evcache"
	"github.com/go-email-validator/go-email-validator/pkg/ev/utils"
	"github.com/go-email-validator/go-email-validator/pkg/log"
	"go.uber.org/zap"
)

// CacheKeyGetter is type for key generators
// To use complex keys you can use https://github.com/vmihailenco/msgpack/
type CacheKeyGetter func(input Input, results ...ValidationResult) interface{}

// EmailCacheKeyGetter generates key as full email
func EmailCacheKeyGetter(input Input, _ ...ValidationResult) interface{} {
	return input.Email().String()
}

// DomainCacheKeyGetter generates key as domain
func DomainCacheKeyGetter(input Input, _ ...ValidationResult) interface{} {
	return input.Email().Domain()
}

// NewCacheDecorator instantiates cache decorator
func NewCacheDecorator(validator Validator, cache evcache.Interface, getKey CacheKeyGetter) Validator {
	if getKey == nil {
		getKey = EmailCacheKeyGetter
	}

	return &cacheDecorator{
		validator: validator,
		cache:     cache,
		getKey:    getKey,
	}
}

type cacheDecorator struct {
	validator Validator
	cache     evcache.Interface
	getKey    CacheKeyGetter
}

func (c *cacheDecorator) GetDeps() []ValidatorName {
	return c.validator.GetDeps()
}

func (c *cacheDecorator) Validate(input Input, results ...ValidationResult) (result ValidationResult) {
	key := c.getKey(input, results...)
	resultInterface, err := c.cache.Get(key)
	if err == nil && resultInterface != nil {
		result = *resultInterface.(*ValidationResult)
	} else {
		result = c.validator.Validate(input, results...)
		if err := c.cache.Set(key, result); err != nil {
			log.Logger().Error(fmt.Sprintf("cache decorator %v", err),
				zap.String("validator", utils.StructName(c.validator)),
				zap.String("key", fmt.Sprint(key)),
				zap.String("email", input.Email().String()),
				zap.String("results", fmt.Sprint(results)),
			)
		}
	}

	return result
}