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

package redistest

import (
	"time"

	"git.i2edu.net/i2/go-zero/core/lang"
	"git.i2edu.net/i2/go-zero/core/stores/redis"
	"github.com/alicebob/miniredis/v2"
)

// CreateRedis returns a in process redis.Redis.
func CreateRedis() (r *redis.Redis, clean func(), err error) {
	mr, err := miniredis.Run()
	if err != nil {
		return nil, nil, err
	}

	return redis.NewRedis(mr.Addr(), redis.NodeType), func() {
		ch := make(chan lang.PlaceholderType)
		go func() {
			mr.Close()
			close(ch)
		}()
		select {
		case <-ch:
		case <-time.After(time.Second):
		}
	}, nil
}