cfcompfetch.go

package main

import (
	"encoding/json"
	"fmt"
	"github.com/apcera/nats"
	"os"
	"strings"
	"time"
)

type Report struct {
	Type        string
	Host        string
	Uid         string
	Index       int
	Credentials [2]string
}

func match(rep Report, ip string, dtype string) bool {
	host := strings.Split(rep.Host, ":")
	if host[0] == ip && dtype == rep.Type {
		return true
	}
	return false
}

func decode(msg *nats.Msg) (Report, error) {
	var report Report
	err := json.Unmarshal(msg.Data, &report)
	return report, err
}

func main() {
	var DISCOVER = "vcap.component.discover"
	nc, _ := nats.Connect(os.Args[1])
	dtype := os.Args[2]
	ip := os.Args[3]

	inbox := nats.NewInbox()
	sub, _ := nc.SubscribeSync(inbox)

	err := nc.PublishRequest(DISCOVER, inbox, []byte(""))
	if err == nil {
		results := make(chan []byte, 1)
		go func() {
			for {
				m, _ := sub.NextMsg(time.Second)
				if m != nil {
					rep, _ := decode(m)
					if match(rep, ip, dtype) {
						results <- m.Data
					}
				}
			}
		}()

		select {
		case res := <-results:
			fmt.Println(string(res))
		case <-time.After(time.Second * 5):
			fmt.Println("TIMEOUT")
			os.Exit(1)
		}

	}
}