java/management/tools/qpid-cli/src/org/apache/qpid/commands/Commandview.java

/*
 *
 * 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.commands;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;

import javax.management.MBeanServerConnection;
import javax.management.ObjectName;
import javax.management.openmbean.CompositeData;
import javax.management.openmbean.TabularDataSupport;

import org.apache.qpid.commands.objects.QueueObject;
import org.apache.qpid.utils.JMXinfo;

public class Commandview extends CommandImpl
{

    public static final String COMMAND_NAME = "view";

    private int number = 0;
    private QueueObject objname;
    private MBeanServerConnection mbsc;
    private String method1;
    private ObjectName queue;

    public Commandview(JMXinfo info)
    {
        super(info);
        this.mbsc = info.getmbserverconnector();
        this.objname = new QueueObject(mbsc);
        this.method1 = "viewMessages";

    }

    public void viewmessages()
    {
        objname.setQueryString(this.getObject(), this.getName(), this.getVirtualhost());
        Set set = objname.returnObjects();
        String header = "", temp_header = "", message_data = "", outline = "";

        if (objname.getSet().size() != 0)
        {
            Iterator it = set.iterator();
            this.queue = (ObjectName) it.next();
            try
            {
                if (objname.getmessagecount(this.queue) == 0)
                {
                    echo("Selected Queue doesn't contain any messages");
                    return;
                }
                if (this.number == 0)
                    this.number = objname.getmessagecount(this.queue);

                if (objname.getmessagecount(this.queue) < this.number)
                {
                    echo("Given number is Greater than the Queue Depth");
                    return;
                }
                else
                {
                    Object[] params = { 1, this.number };
                    String[] signature = { new String("int"), new String("int") };
                    TabularDataSupport data = (TabularDataSupport) this.mbsc.invoke(queue, this.method1, params,
                            signature);

                    Set entrySet = data.entrySet();
                    ArrayList<Map.Entry> list = new ArrayList<Map.Entry>(entrySet);
                    if (list.size() != 0)
                    {// no data}
                        for (int i = 0; i < list.size(); i++)
                        {
                            CompositeData compositedata = (CompositeData) (list.get(i)).getValue();
                            List<String> itemNames = new ArrayList<String>(compositedata.getCompositeType().keySet());
                            if (i == 0) // display the table headers
                            {
                                for (int j = 0; j < itemNames.size(); j++)
                                {
                                    temp_header = "";
                                    if (j != 1) // skipping header information
                                    {
                                        temp_header = itemNames.get(j);
                                        while (temp_header.length() < 15)
                                            temp_header = " " + temp_header;

                                        header = header + temp_header + "|";
                                    }
                                    else
                                        continue;
                                }
                                echo(header);
                                while (outline.length() < header.length())
                                    outline = outline + "-";
                                echo(outline);
                            }

                            for (int j = 0; j < itemNames.size(); j++)
                            {
                                temp_header = "";
                                if (j != 1)
                                {
                                    temp_header = String.valueOf(compositedata.get(itemNames.get(j)));
                                    while (temp_header.length() < 15)
                                        temp_header = " " + temp_header;
                                    message_data = message_data + temp_header + "|";
                                }
                                else
                                    // header information is not displaying
                                    // unless user specify header information is
                                    // needed
                                    continue;

                            }
                            echo(message_data);
                            header = "";
                            message_data = "";
                        }
                    }
                    else
                    {
                        System.out.println("No Data to Display");
                    }
                }
            }
            catch (Exception ex)
            {
                ex.printStackTrace();
            }

        }
        else
        {
            if (hasName())
            {

                echo("The Queue you have specified is not in the current broker");
                echo("");
            }
            else
            {
                printusage();
            }
        }
    }

    public void execute()
    {
        /*
         * In here you it's easy to handle any number of otpions which are going
         * to add with the list command which works with main option object or o
         */

        if (checkoptionsetting("object") || checkoptionsetting("o"))
        {
            String object = optionchecker("object");
            if (object == null)
            {
                object = optionchecker("o");
            }
            if (object.compareToIgnoreCase("queue") == 0)
                setObject(object);
            else
            {
                unrecognizeoption();
                echo("This command is only applicable for delete command so please start with queue");
            }
            if (checkoptionsetting("name") || checkoptionsetting("n"))
            {
                String name = optionchecker("name");
                if (name == null)
                    name = optionchecker("n");

                setName(name);
            }
            if (checkoptionsetting("virtualhost") || checkoptionsetting("v"))
            {
                String vhost = optionchecker("virtualhost");
                if (vhost == null)
                    vhost = optionchecker("v");
                setVirtualhost(vhost);
            }
            if (checkoptionsetting("top") || checkoptionsetting("t"))
            {
                String number = optionchecker("top");
                if (number == null)
                    number = optionchecker("t");

                setnumber(removeSpaces(number));
            }
            this.viewmessages();
        }
        else if (checkoptionsetting("h") || checkoptionsetting("help"))
            printusage();
        else
            unrecognizeoption();
    }

    public void printusage()
    {
        echo("");
        echo("Usage:view [OPTION] ... [OBJECT TYPE]...\n");
        echo("view the information about given number of messages from the given queue object\n");
        echo("To specify the desired queue you have to give the virtualhost name and the queue name with following commands\n");
        echo("Where possible options include:\n");
        echo("        -v      --virtualhost Give the virtuallhost name of the desired queue");
        echo("        -n      --name        Give the queue name of the desired queue you want to view messages");
        echo("        -t      --top         Give how many number of messages you want to delete from the top (Default = all the messages will be deleted");
        echo("        -h      --help        Display the help and back to the qpid-cli prompt\n");

    }

    private void setnumber(String number)
    {
        Integer i = new Integer(number);
        this.number = i.intValue();
    }

    private static String removeSpaces(String s)
    {
        StringTokenizer st = new StringTokenizer(s, " ", false);
        String t = "";
        while (st.hasMoreElements())
            t += st.nextElement();
        return t;
    }

    public int getnumber()
    {
        return number;
    }
}