| 150 | } |
| 151 | |
| 152 | func runPublisher(nc *nats.Conn, startwg, donewg *sync.WaitGroup, numMsgs int, msgSize int) { |
| 153 | startwg.Done() |
| 154 | |
| 155 | args := flag.Args() |
| 156 | subj := args[0] |
| 157 | var msg []byte |
| 158 | if msgSize > 0 { |
| 159 | msg = make([]byte, msgSize) |
| 160 | } |
| 161 | |
| 162 | start := time.Now() |
| 163 | |
| 164 | for i := 0; i < numMsgs; i++ { |
| 165 | nc.Publish(subj, msg) |
| 166 | } |
| 167 | nc.Flush() |
| 168 | benchmark.AddPubSample(bench.NewSample(numMsgs, msgSize, start, time.Now(), nc)) |
| 169 | |
| 170 | donewg.Done() |
| 171 | } |
| 172 | |
| 173 | func runSubscriber(nc *nats.Conn, startwg, donewg *sync.WaitGroup, numMsgs int, msgSize int) { |
| 174 | args := flag.Args() |