nsconnector.go

package tsuki

import (
	"fmt"
	"log"
	"net/http"
	"strings"
)

const NSPORT = ":7071"

type NSConnector interface {
    ReceivedChunk(id string)

    SetNSAddr(addr string)
    GetNSAddr() string
    IsNS(addr string) bool

    Poller
}

type HTTPNSConnector struct {
    Addr string
    httpAddr string
    ip string
}

func (c *HTTPNSConnector) ReceivedChunk(id string) {
    url := fmt.Sprintf("%s/confirm/receivedChunk?chunkID=%s", c.httpAddr, id)
    log.Printf("ReceivedChunk: %s", url)
    go http.Get(url)
}

func (c *HTTPNSConnector) GetNSAddr() string {
    return c.Addr
}

func (c *HTTPNSConnector) SetNSAddr(addr string) {
    colon := strings.IndexRune(addr, ':')
    if colon == -1 {
        colon = len(addr)
    }

    c.ip = addr[:colon]

    c.Addr = c.ip + NSPORT
    c.httpAddr = "http://" + c.Addr

    log.Printf("SetNSAddr: %#v", c)
}

func (c *HTTPNSConnector) IsNS(addr string) bool {
    if c.ip == "" {
        return true
    }

    colon := strings.IndexRune(addr, ':')
    if colon == -1 {
        colon = len(addr)
    }

    ip := addr[:colon]

    return c.ip == ip
}

func (c *HTTPNSConnector) Poll() {
    _, err := http.Get(c.httpAddr + "/pulse")

    if err != nil {
        log.Printf("warning: couldn't send hertbeat to %s", c.httpAddr + "/pulse")
    }
}


type SpyNSConnector struct {
    receivedChunks []string
    Addr string
    PulseCount int
}

func (c *SpyNSConnector) ReceivedChunk(id string) {
    c.receivedChunks = append(c.receivedChunks, id)
}

func (c *SpyNSConnector) Reset() {
    c.receivedChunks = nil
}

func (c *SpyNSConnector) GetNSAddr() string {
    return c.Addr
}

func (c *SpyNSConnector) SetNSAddr(addr string) {
    c.Addr = addr
}

func (c *SpyNSConnector) IsNS(addr string) bool {
    if c.Addr == "" {
        return true
    }

    return c.Addr == addr
}

func (c *SpyNSConnector) Poll() {
    if c.Addr != "" {
        c.PulseCount++
    }       
}