RC9/qpid/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.
 *
 */
 /*
 *
 * Copyright (c) 2006 The Apache Software Foundation
 *
 * Licensed 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 org.apache.qpid.commands.objects.ObjectNames;
import org.apache.qpid.commands.objects.QueueObject;
import org.apache.qpid.utils.JMXinfo;

import javax.management.MBeanServerConnection;
import javax.management.ObjectName;
import javax.management.MBeanAttributeInfo;
import javax.management.MBeanInfo;
import javax.management.openmbean.*;
import java.util.*;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.awt.font.OpenType;

/**
 * Created by IntelliJ IDEA.
 * User: lahiru
 * Date: Aug 8, 2008
 * Time: 11:33:05 PM
 * To change this template use File | Settings | File Templates.
 */
public class Commandview extends Command {
    private String object;
    private String name;
    private String vhost;
    private int number = 0;
    private QueueObject objname;
    private MBeanServerConnection mbsc;
    private String method1, method2;
    private ObjectName queue;

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

    }

    public void viewmessages() {
        Set set = null;
        Object temp[] = {null};
        objname.setQueryString(this.object, this.name, this.vhost);
        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 (isname()) {

                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");
                setvhost(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 setobject(String object) {
        this.object = object;
    }

    private void setname(String name) {
        this.name = name;
    }

    private boolean isname() {
        if (this.name == null)
            return false;

        else
            return true;
    }

    private void setvhost(String vhost) {
        this.vhost = vhost;
    }

    public String getvhost() {
        return this.vhost;
    }

    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 String getname() {
        return this.name;
    }

    public String getobject() {
        return this.object;
    }

    public int getnumber() {
        return this.number;
    }

}