agumon.go

package main

import (
	zmq "github.com/pebbe/zmq4"
	//"golang.org/x/net/websocket"

	"errors"
	"fmt"
	"strconv"
	"strings"
)

type Hist struct {
	Bins     []int64
	N        int64
	Min, Max float64
	BinWidth float64
}

func NewHist(n int64, min, max float64) (*Hist, error) {
	if min >= max {
		return nil, errors.New("Min has to be smaller than max.")
	}
	if n < 1 {
		return nil, errors.New("N has to be larger than 0.")
	}
	hist := new(Hist)
	hist.Min = min
	hist.Max = max
	hist.N = n
	hist.Bins = make([]int64, n)
	hist.BinWidth = (max - min) / float64(n)
	return hist, nil
}

func (hist *Hist) UnsafeAdd(val float64) {
	bin := int64((val - hist.Min) / hist.BinWidth)
	hist.Bins[bin]++
}

func (hist *Hist) Add(val float64) error {
	if val < hist.Min || val > hist.Max {
		return errors.New("Value has to be within limits.")
	}
	hist.UnsafeAdd(val)
	return nil
}

func (hist *Hist) Print() {
	binValue := hist.Min
	for _, bin := range hist.Bins {
		fmt.Printf("%f: %d\n", binValue, bin)
		binValue += hist.BinWidth
	}
}

func main() {
	sub, _ := zmq.NewSocket(zmq.SUB)
	defer sub.Close()
	sub.Connect("tcp://localhost:31337")

	filter := "10001 "
	sub.SetSubscribe(filter)

	update_nbr := 0
	h, err := NewHist(10, -3, 3)
	if err != nil {
		panic("LULWUT!")
	}

	for update_nbr <= 10 {
		msg, _ := sub.Recv(0)
		fmt.Printf("%s\n", msg)

		if msgs := strings.Fields(msg); len(msgs) > 1 {
			if temp, err := strconv.ParseFloat(msgs[1], 64); err == nil {
				herr := h.Add(temp)
				if herr != nil {
					fmt.Printf("Couldn't add value: %s", herr)
				}
				update_nbr++
			}
		}
	}
	h.Print()
}

agumon.py

#!/usr/bin/env python

import random
import time
import zmq

PORT = 31337

if __name__ == '__main__':
    ctx = zmq.Context()
    sck = ctx.socket(zmq.PUB)
    sck.bind("tcp://*:{}".format(PORT))
    while True:
        top = random.randrange(9999, 10005)
        msg = random.normalvariate(0, 1)
        msgstr = "{} {}".format(top, msg)
        print(msgstr)
        sck.send_string(msgstr)
        time.sleep(1)