nanomsg.org/go/mangos/v2@v2.0.9-0.20200203084354-8a092611e461/perf/reqlatency.go (about) 1 // Copyright 2019 The Mangos Authors 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use file except in compliance with the License. 5 // You may obtain a copy of the license at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 // Package perf provides utilities to measure mangos peformance against 16 // libnanomsg' perf tools. 17 18 package main 19 20 import ( 21 "fmt" 22 "log" 23 "time" 24 25 "nanomsg.org/go/mangos/v2" 26 "nanomsg.org/go/mangos/v2/protocol/rep" 27 "nanomsg.org/go/mangos/v2/protocol/req" 28 "nanomsg.org/go/mangos/v2/transport/all" 29 ) 30 31 // ReqRepLatencyServer is the server side for REQ/REP latency testing. 32 func ReqRepLatencyServer(addr string, msgSize int, roundTrips int) { 33 s, err := rep.NewSocket() 34 if err != nil { 35 log.Fatalf("Failed to make new pair socket: %v", err) 36 } 37 defer func() { time.Sleep(10 * time.Microsecond); _ = s.Close() }() 38 39 all.AddTransports(s) 40 l, err := s.NewListener(addr, nil) 41 if err != nil { 42 log.Fatalf("Failed to make new listener: %v", err) 43 } 44 45 err = l.Listen() 46 if err != nil { 47 log.Fatalf("Failed to listen: %v", err) 48 } 49 for i := 0; i != roundTrips; i++ { 50 msg, err := s.RecvMsg() 51 if err != nil { 52 log.Fatalf("Failed to recv: %v", err) 53 } 54 if len(msg.Body) != msgSize { 55 log.Fatalf("Received wrong message size: %d != %d", len(msg.Body), msgSize) 56 } 57 if err = s.SendMsg(msg); err != nil { 58 log.Fatalf("Failed to send: %v", err) 59 } 60 } 61 } 62 63 // ReqRepLatencyClient is the client side of the latency test. It measures 64 // round trip times using REQ/REP protocol. 65 func ReqRepLatencyClient(addr string, msgSize int, roundTrips int) { 66 s, err := req.NewSocket() 67 if err != nil { 68 log.Fatalf("Failed to make new pair socket: %v", err) 69 } 70 defer func() { 71 _ = s.Close() 72 }() 73 74 all.AddTransports(s) 75 d, err := s.NewDialer(addr, nil) 76 if err != nil { 77 log.Fatalf("Failed to make new dialer: %v", err) 78 } 79 80 err = d.Dial() 81 if err != nil { 82 log.Fatalf("Failed to dial: %v", err) 83 } 84 85 // 100 milliseconds to give TCP a chance to establish 86 time.Sleep(time.Millisecond * 100) 87 msg := mangos.NewMessage(msgSize) 88 89 total := time.Duration(0) 90 for i := 0; i < roundTrips; i++ { 91 msg.Body = msg.Body[0:msgSize] 92 msg.Header = msg.Header[:0] 93 start := time.Now() 94 if err = s.SendMsg(msg); err != nil { 95 log.Fatalf("Failed SendMsg: %v", err) 96 } 97 if msg, err = s.RecvMsg(); err != nil { 98 log.Fatalf("Failed RecvMsg: %v", err) 99 } 100 total += time.Since(start) 101 } 102 msg.Free() 103 104 lat := float64(total/time.Microsecond) / float64(roundTrips) 105 fmt.Printf("message size: %d [B]\n", msgSize) 106 fmt.Printf("round trip count: %d\n", roundTrips) 107 fmt.Printf("average RTT: %.3f [us]\n", lat) 108 }