Merge pull request #453 from spiral/plugin/redis

Plugin/redis

1 files changed, 75 insertions, 0 deletions

diff --git a/plugins/redis/plugin.go b/plugins/redis/plugin.go
new file mode 100644
index 00000000..08bb7972
--- /dev/null
+++ b/plugins/redis/plugin.go
@@ -0,0 +1,75 @@
+package redis
+
+import (
+	"github.com/go-redis/redis/v8"
+	"github.com/spiral/errors"
+	"github.com/spiral/roadrunner/v2/interfaces/config"
+	"github.com/spiral/roadrunner/v2/interfaces/log"
+)
+
+const PluginName = "redis"
+
+type Plugin struct {
+	// config for RR integration
+	cfg *Config
+	// logger
+	log log.Logger
+	// redis universal client
+	universalClient redis.UniversalClient
+}
+
+func (s *Plugin) GetClient() redis.UniversalClient {
+	return s.universalClient
+}
+
+func (s *Plugin) Init(cfg config.Configurer, log log.Logger) error {
+	const op = errors.Op("redis plugin init")
+	s.cfg = &Config{}
+	s.cfg.InitDefaults()
+
+	err := cfg.UnmarshalKey(PluginName, &s.cfg)
+	if err != nil {
+		return errors.E(op, errors.Disabled, err)
+	}
+
+	s.log = log
+
+	s.universalClient = redis.NewUniversalClient(&redis.UniversalOptions{
+		Addrs:              s.cfg.Addrs,
+		DB:                 s.cfg.DB,
+		Username:           s.cfg.Username,
+		Password:           s.cfg.Password,
+		SentinelPassword:   s.cfg.SentinelPassword,
+		MaxRetries:         s.cfg.MaxRetries,
+		MinRetryBackoff:    s.cfg.MaxRetryBackoff,
+		MaxRetryBackoff:    s.cfg.MaxRetryBackoff,
+		DialTimeout:        s.cfg.DialTimeout,
+		ReadTimeout:        s.cfg.ReadTimeout,
+		WriteTimeout:       s.cfg.WriteTimeout,
+		PoolSize:           s.cfg.PoolSize,
+		MinIdleConns:       s.cfg.MinIdleConns,
+		MaxConnAge:         s.cfg.MaxConnAge,
+		PoolTimeout:        s.cfg.PoolTimeout,
+		IdleTimeout:        s.cfg.IdleTimeout,
+		IdleCheckFrequency: s.cfg.IdleCheckFreq,
+		ReadOnly:           s.cfg.ReadOnly,
+		RouteByLatency:     s.cfg.RouteByLatency,
+		RouteRandomly:      s.cfg.RouteRandomly,
+		MasterName:         s.cfg.MasterName,
+	})
+
+	return nil
+}
+
+func (s *Plugin) Serve() chan error {
+	errCh := make(chan error, 1)
+	return errCh
+}
+
+func (s Plugin) Stop() error {
+	return s.universalClient.Close()
+}
+
+func (s *Plugin) Name() string {
+	return PluginName
+}