go-zero/core/stores/redis/conf.go

package redis

import (
	"errors"
	"time"

	"github.com/tal-tech/go-zero/core/conf"
)

var (
	// ErrEmptyHost is an error that indicates no redis host is set.
	ErrEmptyHost = errors.New("empty redis host")
	// ErrEmptyType is an error that indicates no redis type is set.
	ErrEmptyType = errors.New("empty redis type")
	// ErrEmptyKey is an error that indicates no redis key is set.
	ErrEmptyKey = errors.New("empty redis key")
)

type (
	// A RedisConf is a redis config.
	RedisConf struct {
		Host          string
		Type          string        `json:",default=node,options=node|cluster"`
		Pass          string        `json:",optional"`
		Tls           bool          `json:",default=false,options=true|false"`
		SlowThreshold time.Duration `json:",default=100ms"`
	}

	// A RedisKeyConf is a redis config with key.
	RedisKeyConf struct {
		RedisConf
		Key string `json:",optional"`
	}
)

// NewRedis returns a Redis.
func (rc RedisConf) NewRedis() *Redis {
	var opts []Option
	if rc.Type == ClusterType {
		opts = append(opts, Cluster())
	}
	if len(rc.Pass) > 0 {
		opts = append(opts, WithPass(rc.Pass))
	}
	if rc.SlowThreshold > 0 {
		opts = append(opts, WithSlowThreshold(conf.CheckedDuration(rc.SlowThreshold)))
	}
	if rc.Tls {
		opts = append(opts, WithTLS())
	}

	return New(rc.Host, opts...)
}

// Validate validates the RedisConf.
func (rc RedisConf) Validate() error {
	if len(rc.Host) == 0 {
		return ErrEmptyHost
	}

	if len(rc.Type) == 0 {
		return ErrEmptyType
	}

	return nil
}

// Validate validates the RedisKeyConf.
func (rkc RedisKeyConf) Validate() error {
	if err := rkc.RedisConf.Validate(); err != nil {
		return err
	}

	if len(rkc.Key) == 0 {
		return ErrEmptyKey
	}

	return nil
}