| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| using System; |
| using System.Diagnostics; |
| using System.Threading.Tasks; |
| using Apache.NMS; |
| |
| namespace PingPong |
| { |
| public class Ping: IDisposable |
| { |
| private readonly IConnection connection; |
| private readonly ISession session; |
| private readonly IMessageProducer messageProducer; |
| private readonly IMessageConsumer messageConsumer; |
| private readonly Stopwatch stopwatch; |
| private readonly ITextMessage pingMessage; |
| private TaskCompletionSource<Stats> tsc; |
| private int numberOfMessages; |
| private int skipMessages; |
| private int counter; |
| |
| public Ping(IConnectionFactory connectionFactory) |
| { |
| this.connection = connectionFactory.CreateConnection(); |
| this.session = this.connection.CreateSession(); |
| this.messageProducer = session.CreateProducer(session.GetTopic("ping")); |
| this.messageConsumer = session.CreateConsumer(session.GetTopic("pong")); |
| this.messageConsumer.Listener += OnMessage; |
| this.stopwatch = new Stopwatch(); |
| this.pingMessage = session.CreateTextMessage("Ping"); |
| this.connection.Start(); |
| } |
| |
| private void OnMessage(IMessage message) |
| { |
| if (skipMessages > 0) |
| skipMessages--; |
| else |
| counter++; |
| |
| if (counter == numberOfMessages) |
| { |
| stopwatch.Stop(); |
| this.tsc.SetResult(new Stats { MessagesCount = counter, Elapsed = stopwatch.Elapsed }); |
| } |
| else |
| { |
| messageProducer.Send(pingMessage); |
| } |
| } |
| |
| public Task<Stats> Start(int numberOfMessages, int skipMessages) |
| { |
| this.numberOfMessages = numberOfMessages; |
| this.skipMessages = skipMessages; |
| this.tsc = new TaskCompletionSource<Stats>(); |
| stopwatch.Start(); |
| messageProducer.Send(pingMessage); |
| return this.tsc.Task; |
| } |
| |
| public void Dispose() |
| { |
| connection?.Dispose(); |
| } |
| } |
| } |