ddns/hosts/redis.go

package hosts

import (
	"github.com/garyburd/redigo/redis"
	"github.com/pboehm/ddns/config"
	"time"
)

type RedisBackend struct {
	expirationSeconds int
	pool              *redis.Pool
}

func NewRedisBackend(config *config.Config) *RedisBackend {
	return &RedisBackend{
		expirationSeconds: config.HostExpirationDays * 24 * 60 * 60,
		pool: &redis.Pool{
			MaxIdle:     3,
			IdleTimeout: 240 * time.Second,

			Dial: func() (redis.Conn, error) {
				c, err := redis.Dial("tcp", config.RedisHost)
				if err != nil {
					return nil, err
				}
				return c, err
			},

			TestOnBorrow: func(c redis.Conn, t time.Time) error {
				_, err := c.Do("PING")
				return err
			},
		},
	}
}

func (r *RedisBackend) Close() {
	r.pool.Close()
}

func (r *RedisBackend) GetHost(name string) (*Host, error) {
	conn := r.pool.Get()
	defer conn.Close()

	host := Host{Hostname: name}

	var err error
	var data []interface{}

	if data, err = redis.Values(conn.Do("HGETALL", host.Hostname)); err != nil {
		return nil, err
	}

	if err = redis.ScanStruct(data, &host); err != nil {
		return nil, err
	}

	return &host, nil
}

func (r *RedisBackend) SetHost(host *Host) error {
	conn := r.pool.Get()
	defer conn.Close()

	var err error

	if _, err = conn.Do("HMSET", redis.Args{}.Add(host.Hostname).AddFlat(host)...); err != nil {
		return err
	}

	if _, err = conn.Do("EXPIRE", host.Hostname, r.expirationSeconds); err != nil {
		return err
	}

	return nil
}