Method Open

backend/plugin/db/redis/redis.go:49–157

Open opens the redis driver.

(_ context.Context, _ storepb.Engine, config db.ConnectionConfig)

// Open opens the redis driver.
func (d *Driver) Open(_ context.Context, _ storepb.Engine, config db.ConnectionConfig) (db.Driver, error) {
	tlscfg, err := util.GetTLSConfig(config.DataSource)
	if err != nil {
		return nil, errors.Wrap(err, "redis: failed to get tls config")
	}
	db := 0
	if config.ConnectionContext.DatabaseName != "" {
		database, err := strconv.Atoi(config.ConnectionContext.DatabaseName)
		if err != nil {
			return nil, errors.Wrapf(err, "failed to convert database %s to int", config.ConnectionContext.DatabaseName)
		}
		db = database
	}
	d.databaseName = fmt.Sprintf("%d", db)
	switch config.DataSource.GetRedisType() {
	case storepb.DataSource_REDIS_TYPE_UNSPECIFIED, storepb.DataSource_STANDALONE:
		options := &redis.Options{
			Addr:      fmt.Sprintf("%s:%s", config.DataSource.Host, config.DataSource.Port),
			Username:  config.DataSource.Username,
			Password:  config.Password,
			TLSConfig: tlscfg,
			DB:        db,
		}
		if config.DataSource.GetSshHost() != "" {
			sshClient, err := util.GetSSHClient(config.DataSource)
			if err != nil {
				return nil, err
			}
			d.sshClient = sshClient

			options.Dialer = func(_ context.Context, network, addr string) (net.Conn, error) {
				conn, err := sshClient.Dial(network, addr)
				if err != nil {
					return nil, err
				}
				return &util.NoDeadlineConn{Conn: conn}, nil
			}
		}
		client := redis.NewClient(options)
		d.databaseName = fmt.Sprintf("%d", db)
		d.rdb = client
	case storepb.DataSource_SENTINEL:
		sentinelAddrs := make([]string, 0, 1+len(config.DataSource.GetAdditionalAddresses()))
		sentinelAddrs = append(sentinelAddrs, fmt.Sprintf("%s:%s", config.DataSource.Host, config.DataSource.Port))
		for _, sentinelAddr := range config.DataSource.GetAdditionalAddresses() {
			sentinelAddrs = append(sentinelAddrs, fmt.Sprintf("%s:%s", sentinelAddr.Host, sentinelAddr.Port))
		}
		options := &redis.FailoverOptions{
			MasterName:       config.DataSource.GetMasterName(),
			Username:         config.DataSource.GetMasterUsername(),
			Password:         config.DataSource.GetMasterPassword(),
			SentinelUsername: config.DataSource.Username,
			SentinelPassword: config.Password,
			SentinelAddrs:    sentinelAddrs,
			DB:               db,
			TLSConfig:        tlscfg,
		}
		if config.DataSource.GetSshHost() != "" {