| /* |
| * |
| * 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. |
| * |
| */ |
| package org.apache.qpid.transport; |
| |
| import java.io.IOException; |
| import java.nio.ByteBuffer; |
| |
| import org.apache.qpid.transport.network.ConnectionBinding; |
| import org.apache.qpid.transport.network.io.IoAcceptor; |
| |
| /** |
| * Sink |
| * |
| */ |
| |
| public class Sink implements SessionListener |
| { |
| |
| private static final String FORMAT_HDR = "%-12s %-18s %-18s %-18s"; |
| private static final String FORMAT_ROW = "SSN#%-8X %-18s %-18s %-18s"; |
| |
| private long interval = 100000; |
| private long start = System.currentTimeMillis(); |
| private long count = 0; |
| private long bytes = 0; |
| private long interval_start = start; |
| private long bytes_start = bytes; |
| private long time = start; |
| private int id = System.identityHashCode(this); |
| |
| public Sink() |
| { |
| } |
| |
| private double msg_rate() |
| { |
| return 1000 * (double) count / (double) (time - start); |
| } |
| |
| private double byte_rate() |
| { |
| return (1000 * (double) bytes / (double) (time - start)) / (1024*1024); |
| } |
| |
| private double msg_interval_rate() |
| { |
| return 1000 * (double) interval / (double) (time - interval_start); |
| } |
| |
| private double byte_interval_rate() |
| { |
| return (1000 * (double) (bytes - bytes_start) / (double) (time - interval_start)) / (1024*1024); |
| } |
| |
| private String rates() |
| { |
| return String.format("%.2f/%.2f", msg_rate(), byte_rate()); |
| } |
| |
| private String interval_rates() |
| { |
| return String.format("%.2f/%.2f", msg_interval_rate(), byte_interval_rate()); |
| } |
| |
| private String counts() |
| { |
| return String.format("%d/%.2f", count, ((double) bytes)/(1024*1024)); |
| } |
| |
| public void opened(Session ssn) {} |
| |
| public void message(Session ssn, MessageTransfer xfr) |
| { |
| count++; |
| bytes += xfr.getBody().remaining(); |
| if ((count % interval) == 0) |
| { |
| time = System.currentTimeMillis(); |
| System.out.println |
| (String.format |
| (FORMAT_ROW, id, counts(), rates(), interval_rates())); |
| interval_start = time; |
| bytes_start = bytes; |
| } |
| ssn.processed(xfr); |
| } |
| |
| public void exception(Session ssn, SessionException exc) |
| { |
| exc.printStackTrace(); |
| } |
| |
| public void closed(Session ssn) {} |
| |
| public static final void main(String[] args) throws IOException |
| { |
| ConnectionDelegate delegate = new ServerDelegate() |
| { |
| @Override public Session getSession(Connection conn, SessionAttach atc) |
| { |
| Session ssn = super.getSession(conn, atc); |
| ssn.setSessionListener(new Sink()); |
| return ssn; |
| } |
| }; |
| |
| IoAcceptor ioa = new IoAcceptor |
| ("0.0.0.0", 5672, ConnectionBinding.get(delegate)); |
| System.out.println |
| (String.format |
| (FORMAT_HDR, "Session", "Count/MBytes", "Cumulative Rate", "Interval Rate")); |
| System.out.println |
| (String.format |
| (FORMAT_HDR, "-------", "------------", "---------------", "-------------")); |
| ioa.start(); |
| } |
| |
| } |