View Javadoc

    1   /*
    2    *  Licensed to the Apache Software Foundation (ASF) under one
    3    *  or more contributor license agreements.  See the NOTICE file
    4    *  distributed with this work for additional information
    5    *  regarding copyright ownership.  The ASF licenses this file
    6    *  to you under the Apache License, Version 2.0 (the
    7    *  "License"); you may not use this file except in compliance
    8    *  with the License.  You may obtain a copy of the License at
    9    *
    10   *    http://www.apache.org/licenses/LICENSE-2.0
    11   *
    12   *  Unless required by applicable law or agreed to in writing,
    13   *  software distributed under the License is distributed on an
    14   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
    15   *  KIND, either express or implied.  See the License for the
    16   *  specific language governing permissions and limitations
    17   *  under the License.
    18   *
    19   */
    20  package org.apache.mina.example.udp;
    21
    22  import java.awt.BorderLayout;
    23  import java.awt.Dimension;
    24  import java.io.IOException;
    25  import java.net.InetSocketAddress;
    26  import java.net.SocketAddress;
    27  import java.util.concurrent.ConcurrentHashMap;
    28
    29  import javax.swing.JFrame;
    30  import javax.swing.JLabel;
    31  import javax.swing.JPanel;
    32  import javax.swing.JTabbedPane;
    33
    34  import org.apache.mina.core.filterchain.DefaultIoFilterChainBuilder;
    35  import org.apache.mina.filter.logging.LoggingFilter;
    36  import org.apache.mina.transport.socket.DatagramSessionConfig;
    37  import org.apache.mina.transport.socket.nio.NioDatagramAcceptor;
    38
    39  /**
    40   * The class that will accept and process clients in order to properly
    41   * track the memory usage.
    42   *
    43   * @author <a href="http://mina.apache.org">Apache MINA Project</a>
    44   */
    45  public class MemoryMonitor {
    46
    47      private static final long serialVersionUID = 1L;
    48
    49      public static final int PORT = 18567;
    50
    51      protected static final Dimension PANEL_SIZE = new Dimension(300, 200);
    52
    53      private JFrame frame;
    54
    55      private JTabbedPane tabbedPane;
    56
    57      private ConcurrentHashMap<SocketAddress, ClientPanel> clients;
    58
    59      public MemoryMonitor() throws IOException {
    60
    61          NioDatagramAcceptor acceptor = new NioDatagramAcceptor();
    62          acceptor.setHandler(new MemoryMonitorHandler(this));
    63
    64          DefaultIoFilterChainBuilder chain = acceptor.getFilterChain();
    65          chain.addLast("logger", new LoggingFilter());
    66
    67          DatagramSessionConfig dcfg = acceptor.getSessionConfig();
    68          dcfg.setReuseAddress(true);
    69
    70          frame = new JFrame("Memory monitor");
    71          tabbedPane = new JTabbedPane();
    72          tabbedPane.add("Welcome", createWelcomePanel());
    73          frame.add(tabbedPane, BorderLayout.CENTER);
    74          clients = new ConcurrentHashMap<SocketAddress, ClientPanel>();
    75          frame.pack();
    76          frame.setLocation(300, 300);
    77          frame.setVisible(true);
    78
    79          acceptor.bind(new InetSocketAddress(PORT));
    80          System.out.println("UDPServer listening on port " + PORT);
    81      }
    82
    83      private JPanel createWelcomePanel() {
    84          JPanel panel = new JPanel();
    85          panel.setPreferredSize(PANEL_SIZE);
    86          panel.add(new JLabel("Welcome to the Memory Monitor"));
    87          return panel;
    88      }
    89
    90      protected void recvUpdate(SocketAddress clientAddr, long update) {
    91          ClientPanel clientPanel = clients.get(clientAddr);
    92          if (clientPanel != null) {
    93              clientPanel.updateTextField(update);
    94          } else {
    95              System.err.println("Received update from unknown client");
    96          }
    97      }
    98
    99      protected void addClient(SocketAddress clientAddr) {
    100         if (!containsClient(clientAddr)) {
    101             ClientPanel clientPanel = new ClientPanel(clientAddr.toString());
    102             tabbedPane.add(clientAddr.toString(), clientPanel);
    103             clients.put(clientAddr, clientPanel);
    104         }
    105     }
    106
    107     protected boolean containsClient(SocketAddress clientAddr) {
    108         return clients.contains(clientAddr);
    109     }
    110
    111     protected void removeClient(SocketAddress clientAddr) {
    112         clients.remove(clientAddr);
    113     }
    114
    115     public static void main(String[] args) throws IOException {
    116         new MemoryMonitor();
    117     }
    118 }