internet/asn2asd.go

package internet

import (
	"fmt"

	"github.com/garyburd/redigo/redis"
)

//NewASN2ASDescClient returns a new client for quering the. The caller is
//reposible for closing the conn when the client is not used anymore.
func NewASN2ASDescClient(conn redis.Conn) *ASN2ASDescClient {
	asnasd := ASN2ASDescClient{
		conn: conn,
	}
	return &asnasd
}

//ASN2ASDescClient is the query client.
type ASN2ASDescClient struct {
	conn redis.Conn
}

// importedDates fetches all imported dates from redis
func (i *ASN2ASDescClient) importedDates() ([]string, error) {
	return redis.Strings(i.conn.Do("SMEMBERS", "asd:imported_dates"))
}

// Current returns the latest known result for an IP2ASN lookup.
func (i *ASN2ASDescClient) Current(ASN int) (string, error) {
	allDates, err := i.importedDates()
	if err != nil {
		return "", err
	}
	if len(allDates) < 0 {
		return "", err
	}
	current := allDates[len(allDates)-1]
	result, err := redis.String(i.conn.Do("HGET", fmt.Sprintf("asd:%d", ASN), current))
	if err != nil {
		return "", err
	}
	return result, nil
}