func NewSubscriber(ctx context.Context, url string) (*Subscriber, error) { var sock mangos.Socket var err error sock, err = sub.NewSocket() if err != nil { return nil, err } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) err = sock.Dial(url) if err != nil { return nil, err } // subscribes to everything err = sock.SetOption(mangos.OptionSubscribe, []byte("")) if err != nil { return nil, err } subscriber := &Subscriber{ url: url, ctx: ctx, sock: sock, changes: make(chan []string, 8), } go subscriber.run() return subscriber, nil }
func client(url string, name string) { var sock mangos.Socket var err error var msg []byte if sock, err = sub.NewSocket(); err != nil { die("can't get new sub socket: %s", err.Error()) } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) if err = sock.Dial(url); err != nil { die("can't dial on sub socket: %s", err.Error()) } // Empty byte array effectively subscribes to everything err = sock.SetOption(mangos.OptionSubscribe, []byte("")) if err != nil { die("cannot subscribe: %s", err.Error()) } for { if msg, err = sock.Recv(); err != nil { die("Cannot recv: %s", err.Error()) } fmt.Printf("CLIENT(%s): RECEIVED %s\n", name, string(msg)) } }
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 client(url string, name string) { var sock mangos.Socket var err error var msg []byte if sock, err = respondent.NewSocket(); err != nil { die("can't get new respondent socket: %s", err.Error()) } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) if err = sock.Dial(url); err != nil { die("can't dial on respondent socket: %s", err.Error()) } for { if msg, err = sock.Recv(); err != nil { die("Cannot recv: %s", err.Error()) } fmt.Printf("CLIENT(%s): RECEIVED \"%s\" SURVEY REQUEST\n", name, string(msg)) d := date() fmt.Printf("CLIENT(%s): SENDING DATE SURVEY RESPONSE\n", name) if err = sock.Send([]byte(d)); err != nil { die("Cannot send: %s", err.Error()) } } }
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 node1(url string) { var sock mangos.Socket var err error if sock, err = pair.NewSocket(); err != nil { die("can't get new pair socket: %s", err.Error()) } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) if err = sock.Dial(url); err != nil { die("can't dial on pair socket: %s", err.Error()) } send_recv(sock, "node1") }
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 ClientWithSub(urlServer string, urlPubSub string, service string) (*DiscoveryClient, error) { var sock mangos.Socket var err error var subscriber *Subscriber ctx, cancel := context.WithCancel(context.Background()) if urlPubSub != "" { subCtx, _ := context.WithCancel(ctx) subscriber, err = NewSubscriber(subCtx, urlPubSub) if err != nil { return nil, err } } sock, err = respondent.NewSocket() if err != nil { return nil, err } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) err = sock.Dial(urlServer) if err != nil { return nil, err } client := &DiscoveryClient{ urlServer: urlServer, urlPubSub: urlPubSub, service: service, ctx: ctx, cancel: cancel, sock: sock, subscriber: subscriber, } go client.run() return client, nil }
func node(args []string) { var sock mangos.Socket var err error var msg []byte var x int if sock, err = bus.NewSocket(); err != nil { die("bus.NewSocket: %s", err) } sock.AddTransport(ipc.NewTransport()) sock.AddTransport(tcp.NewTransport()) if err = sock.Listen(args[2]); err != nil { die("sock.Listen: %s", err.Error()) } // wait for everyone to start listening time.Sleep(time.Second) for x = 3; x < len(args); x++ { if err = sock.Dial(args[x]); err != nil { die("socket.Dial: %s", err.Error()) } } // wait for everyone to join time.Sleep(time.Second) fmt.Printf("%s: SENDING '%s' ONTO BUS\n", args[1], args[1]) if err = sock.Send([]byte(args[1])); err != nil { die("sock.Send: %s", err.Error()) } for { if msg, err = sock.Recv(); err != nil { die("sock.Recv: %s", err.Error()) } fmt.Printf("%s: RECEIVED \"%s\" FROM BUS\n", args[1], string(msg)) } }
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() }