func StartNanomsgJsonTest(address string, clients int, requestsPerClient int, messageSize int, timer metrics.Timer, requestSize *int) func(wg *sync.WaitGroup) {
return func(wg *sync.WaitGroup) {
var socket mangos.Socket
var err error
if socket, err = req.NewSocket(); err != nil {
log.Fatal(err)
}
defer socket.Close()
defer wg.Done()
socket.AddTransport(tcp.NewTransport())
if err = socket.Dial("tcp://" + address); err != nil {
log.Fatal(err)
}
request, _ := json.Marshal(Request{Method: "TEST", Payload: strings.Repeat("a", messageSize)})
*requestSize = len(request)
for i := 0; i < requestsPerClient; i++ {
timer.Time(func() {
if err = socket.Send(request); err != nil {
log.Fatal(err)
}
socket.Recv()
})
}
}
}
func main() {
flag.Parse()
var sock mangos.Socket
var err error
if sock, err = pub.NewSocket(); err != nil {
die("can't get new pub socket: %s", err)
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Listen("tcp://127.0.0.1:55555"); err != nil {
die("can't listen on pub socket: %s", err.Error())
}
//sub
var subscriber mangos.Socket
if subscriber, err = sub.NewSocket(); err != nil {
die("can't get new sub socket: %s", err.Error())
}
subscriber.AddTransport(ipc.NewTransport())
subscriber.AddTransport(tcp.NewTransport())
if err = subscriber.Dial("tcp://127.0.0.1:55555"); err != nil {
die("can't dial on sub socket: %s", err.Error())
}
// Empty byte array effectively subscribes to everything
err = subscriber.SetOption(mangos.OptionSubscribe, []byte(""))
start := time.Now()
//var msg []byte
for i := 1; i <= *numbPtr; i++ {
//conn.Publish("test", []byte(randSeq(320)))
sock.Send([]byte(randSeq(320)))
if _, err = subscriber.Recv(); err != nil {
die("Cannot recv: %s", err.Error())
}
//log.Println(string(msg))
/*
runtime.Gosched()
atomic.AddUint64(&ops, 1)
if ops == uint64(*numbPtr) {
elapsed := time.Since(start)
log.Printf("Time took %s", elapsed)
}
*/
}
elapsed := time.Since(start)
log.Printf("Time took %s", elapsed)
defer sock.Close()
fmt.Scanln()
}
func startNode(masterHost, name string) {
var sock mangos.Socket
var err error
var msg []byte
masterUrl := url.URL{Scheme: "tcp", Host: masterHost}
ip := getPrivateIP()
// Try to get new "respondent" socket
if sock, err = respondent.NewSocket(); err != nil {
utils.Die("Can't get new respondent socket: %s", err.Error())
}
defer sock.Close()
sock.AddTransport(tcp.NewTransport())
// Connect to master
if err = sock.Dial(masterUrl.String()); err != nil {
utils.Die("Can't dial on respondent socket: %s", err.Error())
}
// Wait for a survey request and send responses
for {
if msg, err = sock.Recv(); err != nil {
utils.Die("Cannot recv: %s", err.Error())
}
fmt.Printf("Client(%s): Received \"%s\" survey request\n", name, string(msg))
var loadAvg *load.LoadAvgStat
if loadAvg, err = load.LoadAvg(); err != nil {
utils.Die("Cannot get load average: %s", err.Error())
}
var cpuInfo []cpu.CPUInfoStat
if cpuInfo, err = cpu.CPUInfo(); err != nil {
utils.Die("Cannot get CPU info: %s", err.Error())
}
fmt.Printf("CPU INFO len: %d\n", len(cpuInfo))
// Get the normalized CPU load
avg := loadAvg.Load1
cores := int32(0)
for _, info := range cpuInfo {
fmt.Printf("Inner Cores: %d\n", info.Cores)
cores += info.Cores
}
fmt.Printf("Load avg: %f\n", avg)
fmt.Printf("Cores: %d\n", cores)
avg = avg / float64(cores)
fmt.Printf("Client(%s): Sending survey response\n", name)
if err = sock.Send([]byte(fmt.Sprintf("%s,%f", ip, avg))); err != nil {
utils.Die("Cannot send: %s", err.Error())
}
}
}
func simpleSend(t *testing.T, tx mangos.Socket, wg *sync.WaitGroup) {
var buf [256]byte
i := 0
for i = 0; i < 10000; i++ {
l := rand.Intn(255) + 1
buf[0] = uint8(l)
if e := tx.Send(buf[:l]); e != nil {
t.Fatalf("Send: %v", e)
break
}
}
t.Logf("Sent %d Msgs", i)
if e := tx.Close(); e != nil {
t.Fatalf("Tx Close: %v", e)
}
wg.Done()
}
func node1(url string, msg string) {
var sock mangos.Socket
var err error
if sock, err = push.NewSocket(); err != nil {
die("can't get new push socket: %s", err.Error())
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Dial(url); err != nil {
die("can't dial on push socket: %s", err.Error())
}
fmt.Printf("NODE1: SENDING \"%s\"\n", msg)
if err = sock.Send([]byte(msg)); err != nil {
die("can't send message on push socket: %s", err.Error())
}
sock.Close()
}
func main() {
how_rand := flag.Bool("rand", false, "")
flag.Parse()
if flag.NArg() < 1 {
log.Fatal("Missing arguments: at least one endpoint to bind to")
}
var err error
var out mangos.Socket
if out, err = push.NewSocket(); err != nil {
log.Fatal("Nanomsg socket creation failure: ", err)
} else {
defer out.Close()
out.AddTransport(ipc.NewTransport())
out.AddTransport(tcp.NewTransport())
out.SetOption(mangos.OptionWriteQLen, 16)
out.SetOption(mangos.OptionRetryTime, 0)
for i := 0; i < flag.NArg(); i++ {
if err := out.Listen(flag.Arg(i)); err != nil {
log.Fatal("Nanomsg listen() error: ", err)
}
}
}
switch {
case *how_rand:
Gen(out, func(tab []string) string {
if len(tab) <= 0 {
return ""
}
return tab[rand.Intn(len(tab))]
})
default:
i := 0
Gen(out, func(tab []string) string {
if len(tab) <= 0 {
return ""
}
i = (i + 1) % len(tab)
return tab[i]
})
}
}
func clientWorker(url string, id int) {
var sock mangos.Socket
var m *mangos.Message
var err error
if sock, err = req.NewSocket(); err != nil {
die("can't get new req socket: %s", err.Error())
}
// Leave this in Cooked mode!
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Dial(url); err != nil {
die("can't dial on req socket: %s", err.Error())
}
// send an empty messsage
m = mangos.NewMessage(1)
if err = sock.SendMsg(m); err != nil {
die("can't send request: %s", err.Error())
}
if m, err = sock.RecvMsg(); err != nil {
die("can't recv reply: %s", err.Error())
}
sock.Close()
if len(m.Body) != 4 {
die("bad response len: %d", len(m.Body))
}
worker := binary.BigEndian.Uint32(m.Body[0:])
L.Lock()
fmt.Printf("Client: %4d Server: %4d\n", id, worker)
L.Unlock()
}
func node1(url string) {
var sock mangos.Socket
var err error
var msg []byte
if sock, err = req.NewSocket(); err != nil {
die("can't get new req socket: %s", err.Error())
}
sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
if err = sock.Dial(url); err != nil {
die("can't dial on req socket: %s", err.Error())
}
fmt.Printf("NODE1: SENDING DATE REQUEST %s\n", "DATE")
if err = sock.Send([]byte("DATE")); err != nil {
die("can't send message on push socket: %s", err.Error())
}
if msg, err = sock.Recv(); err != nil {
die("can't receive date: %s", err.Error())
}
fmt.Printf("NODE1: RECEIVED DATE %s\n", string(msg))
sock.Close()
}
func simpleRecv(t *testing.T, rx mangos.Socket, wg *sync.WaitGroup) {
i := 0
for i = 0; i < 10000; i++ {
buf, e := rx.Recv()
if e != nil {
t.Fatalf("Recv: %v", e)
break
}
if len(buf) < 1 {
t.Fatalf("Recv: empty buf")
break
}
if len(buf) != int(buf[0]) {
t.Fatalf("Recv: length %d != expected %d",
len(buf), buf[0])
break
}
}
t.Logf("Recvd %d Msgs", i)
if e := rx.Close(); e != nil {
t.Fatalf("Rx Close: %v", e)
}
wg.Done()
}
func (m *Master) queryWorkers(c chan<- float64) {
var (
err error
sock mangos.Socket
)
// Try to get new "surveyor" socket
if sock, err = surveyor.NewSocket(); err != nil {
utils.Die("Can't get new surveyor socket: %s", err)
}
defer sock.Close()
sock.AddTransport(tcp.NewTransport())
// Begin listening on the URL
if err = sock.Listen(m.url.String()); err != nil {
utils.Die("Can't listen on surveyor socket: %s", err.Error())
}
// Set "deadline" for the survey and a timeout for receiving responses
if err = sock.SetOption(mangos.OptionSurveyTime, m.surveyDeadline); err != nil {
utils.Die("SetOption(mangos.OptionSurveyTime): %s", err.Error())
}
if err = sock.SetOption(mangos.OptionRecvDeadline, m.surveyDeadline+(1*time.Second)); err != nil {
utils.Die("SetOption(mangos.OptionRecvDeadline): %s", err.Error())
}
for {
fmt.Println("Sending master request")
if err = sock.Send([]byte("CPU")); err != nil {
utils.Die("Failed sending survey: %s", err.Error())
}
loadAvgs := []float64{}
for {
var msg []byte
if msg, err = sock.Recv(); err != nil {
break
}
parts := strings.Split(string(msg), ",")
if len(parts) != 2 {
continue
}
ip := parts[0]
loadAvgString := parts[1]
// Find the corresponding droplet
var worker *Worker
for _, w := range m.workers {
if w.privateAddr == ip {
worker = w
break
}
}
if worker == nil {
fmt.Printf("Message received from unknown worker '%s'. Skipping...\n", ip)
continue
}
var loadAvg float64
if loadAvg, err = strconv.ParseFloat(string(loadAvgString), 64); err != nil {
utils.Die("ParseFloat(): %s", err.Error())
}
// Set their load average and append this worker's load average to the list
worker.loadAvg = loadAvg
loadAvgs = append(loadAvgs, loadAvg)
}
// Compute the average loadAvg
var loadAvg float64
for _, avg := range loadAvgs {
loadAvg += avg
}
loadAvg /= float64(len(loadAvgs))
// Send the load averages
if !math.IsNaN(loadAvg) {
c <- loadAvg
}
// Wait
time.Sleep(m.queryInterval)
}
}