Package org.apache.qpid.management.ui.views.queue

Source Code of org.apache.qpid.management.ui.views.queue.QueueOperationsTabControl$TableSorter

/*
*
* 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.management.ui.views.queue;

import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredContentProvider;
import org.eclipse.jface.viewers.ITableLabelProvider;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.jface.viewers.TableViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.jface.viewers.ViewerSorter;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.KeyAdapter;
import org.eclipse.swt.events.KeyEvent;
import org.eclipse.swt.events.MouseEvent;
import org.eclipse.swt.events.MouseListener;
import org.eclipse.swt.events.SelectionAdapter;
import org.eclipse.swt.events.SelectionEvent;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Combo;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Display;
import org.eclipse.swt.widgets.Group;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.TabFolder;
import org.eclipse.swt.widgets.Table;
import org.eclipse.swt.widgets.TableColumn;
import org.eclipse.swt.widgets.Text;
import org.eclipse.ui.forms.widgets.FormToolkit;
import org.eclipse.ui.forms.widgets.ScrolledForm;

import org.apache.qpid.management.common.mbeans.ManagedQueue;
import org.apache.qpid.management.ui.ApiVersion;
import org.apache.qpid.management.ui.ApplicationRegistry;
import org.apache.qpid.management.ui.ManagedBean;
import org.apache.qpid.management.ui.jmx.JMXManagedObject;
import org.apache.qpid.management.ui.jmx.MBeanUtility;
import org.apache.qpid.management.ui.views.NumberVerifyListener;
import org.apache.qpid.management.ui.views.TabControl;
import org.apache.qpid.management.ui.views.ViewUtility;

import static org.apache.qpid.management.common.mbeans.ManagedQueue.MSG_AMQ_ID;
import static org.apache.qpid.management.common.mbeans.ManagedQueue.MSG_HEADER;
import static org.apache.qpid.management.common.mbeans.ManagedQueue.MSG_QUEUE_POS;
import static org.apache.qpid.management.common.mbeans.ManagedQueue.MSG_REDELIVERED;
import static org.apache.qpid.management.common.mbeans.ManagedQueue.MSG_SIZE;
import static org.apache.qpid.management.ui.Constants.CONSOLE_IMAGE;
import static org.apache.qpid.management.ui.Constants.RESULT;

import javax.management.MBeanServerConnection;
import javax.management.MBeanServerInvocationHandler;
import javax.management.openmbean.CompositeData;
import javax.management.openmbean.CompositeDataSupport;
import javax.management.openmbean.TabularDataSupport;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;


/**
* Control class for the Queue mbean Operations tab.
*/
public class QueueOperationsTabControl extends TabControl
{
    private FormToolkit _toolkit;
    private ScrolledForm _form;
    private Table _table = null;
    private TableViewer _tableViewer = null;
    private Composite _paramsComposite = null;
           
    private ApiVersion _ApiVersion;
   
    private Text _fromMsgText;
    private Text _toMsgText;
    private static final String FROM_DEFAULT = "1";
    private static final String TO_DEFAULT = "50";
    private long _interval = 50; //(to-from)+1
    private Long _fromMsg = new Long(FROM_DEFAULT);
    private Long _toMsg = new Long(TO_DEFAULT);
   
    private TabularDataSupport _messages = null;
    private ManagedQueue _qmb;
   
    private Button _previousButton;
    private Button _nextButton;
   
    public QueueOperationsTabControl(TabFolder tabFolder, JMXManagedObject mbean, MBeanServerConnection mbsc)
    {
        super(tabFolder);
        _mbean = mbean;
        _ApiVersion = ApplicationRegistry.getServerRegistry(mbean).getManagementApiVersion();
        _qmb = (ManagedQueue) MBeanServerInvocationHandler.newProxyInstance(mbsc,
                                mbean.getObjectName(), ManagedQueue.class, false);
        _toolkit = new FormToolkit(_tabFolder.getDisplay());
        _form = _toolkit.createScrolledForm(_tabFolder);
        _form.getBody().setLayout(new GridLayout());
        createComposites();
        createWidgets();
    }
   
    private void createComposites()
    {
        _paramsComposite = _toolkit.createComposite(_form.getBody(), SWT.NONE);
        _paramsComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));
        _paramsComposite.setLayout(new GridLayout());
    }
   
    /**
     * @see TabControl#getControl()
     */
    public Control getControl()
    {
        return _form;
    }
   
    /**
     * @see TabControl#setFocus()
     */
    public void setFocus()
    {
        _table.setFocus();
    }
   
    @Override
    public void refresh(ManagedBean mbean)
    {
        _messages = null;
        try
        {
            if(_ApiVersion.greaterThanOrEqualTo(1, 3))
            { //broker supports Qpid JMX API 1.3 and takes Long values
               
                //gather a list of all messages on the queue for display and selection
                _messages = (TabularDataSupport) _qmb.viewMessages(_fromMsg,_toMsg);
            }
            else
            { //broker supports Qpid JMX API 1.2 or below and takes int values

                if(_toMsg > Integer.MAX_VALUE || _toMsg > Integer.MAX_VALUE)
                {
                    ViewUtility.popupErrorMessage("Error", "This broker only supports viewing up to message " + Integer.MAX_VALUE);
                    _tableViewer.setInput(null);
                    return;
                }
               
                //gather a list of all messages on the queue for display and selection
                _messages = (TabularDataSupport) _qmb.viewMessages(_fromMsg.intValue(), _toMsg.intValue());
            }
        }
        catch (Exception e)
        {
            MBeanUtility.handleException(mbean,e);
        }

        _tableViewer.setInput(_messages);

        layout();
    }
   
    public void layout()
    {
        _form.layout(true);
        _form.getBody().layout(true, true);
    }
   
    private void createWidgets()
    {
        Group messagesGroup = new Group(_paramsComposite, SWT.SHADOW_NONE | SWT.SCROLL_LINE);
        messagesGroup.setBackground(_paramsComposite.getBackground());
        messagesGroup.setText("Messages");
        messagesGroup.setLayout(new GridLayout());
        GridData gridData = new GridData(SWT.FILL, SWT.FILL, true, true);
        messagesGroup.setLayoutData(gridData);
       
        //from and to fields for selecting the viewing range
        Composite viewMessageRangeComposite = _toolkit.createComposite(messagesGroup);
        gridData = new GridData(SWT.LEFT, SWT.FILL, false, false);
        viewMessageRangeComposite.setLayoutData(gridData);
        viewMessageRangeComposite.setLayout(new GridLayout(8,false));
       
        _toolkit.createLabel(viewMessageRangeComposite, "Queue pos: ");
        _fromMsgText = new Text(viewMessageRangeComposite, SWT.BORDER);
        _fromMsgText.setText(FROM_DEFAULT);
        gridData = new GridData(SWT.LEFT, SWT.FILL, false, false);
        gridData.widthHint = 75;
        _fromMsgText.setLayoutData(gridData);
        _fromMsgText.addVerifyListener(new NumberVerifyListener());
        _fromMsgText.addKeyListener(new KeyAdapter()
        {
            public void keyPressed(KeyEvent event)
            {
                if (event.character == SWT.CR)
                {
                    updateMessageInterval();
                }
            }
        });

        _toolkit.createLabel(viewMessageRangeComposite, "to");

        _toMsgText = new Text(viewMessageRangeComposite, SWT.BORDER);
        _toMsgText.setText(TO_DEFAULT);
        gridData = new GridData(SWT.LEFT, SWT.FILL, false, false);
        gridData.widthHint = 75;
        _toMsgText.setLayoutData(gridData);
        _toMsgText.addVerifyListener(new NumberVerifyListener());
        _toMsgText.addKeyListener(new KeyAdapter()
        {
            public void keyPressed(KeyEvent event)
            {
                if (event.character == SWT.CR)
                {
                    updateMessageInterval();
                }
            }
        });
       
        final Button setButton = _toolkit.createButton(viewMessageRangeComposite, "Set", SWT.PUSH);
        setButton.setLayoutData(new GridData(SWT.LEFT, SWT.FILL, false, false));
        setButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                updateMessageInterval();
            }
        });
       
        _toolkit.createLabel(viewMessageRangeComposite, "     "); //spacer
       
        _previousButton = _toolkit.createButton(viewMessageRangeComposite, "< Prev " + _interval, SWT.PUSH);
        _previousButton.setLayoutData(new GridData(SWT.LEFT, SWT.FILL, false, false));
        _previousButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                //make 'to' be 'from - 1' unless from is 1 (ie there are no previous messages)
                if(_fromMsg > 1)
                {
                    _toMsg = _fromMsg - 1;
                    _toMsgText.setText(_toMsg.toString());
                }
               
                //make 'from' be 'from - INTERVAL', or make it 1 if that would make it 0 or less
                _fromMsg = (_fromMsg - _interval < 1) ? 1 : _fromMsg - _interval;
                _fromMsgText.setText(_fromMsg.toString());
               
                refresh(_mbean);
            }
        });
       
        _nextButton = _toolkit.createButton(viewMessageRangeComposite, "Next " + _interval + " >", SWT.PUSH);
        _nextButton.setLayoutData(new GridData(SWT.LEFT, SWT.FILL, false, false));
        _nextButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                //make 'from' be 'to + 1' unless 'to' is already Long.MAX_VALUE
                if(_toMsg != Long.MAX_VALUE)
                {
                    _fromMsg = _toMsg + 1;
                    _fromMsgText.setText(_fromMsg.toString());
                }
               
                //make 'to' be 'to + INTERVAL', or make it Long.MAX_VALUE if that would too large
                _toMsg = (Long.MAX_VALUE - _toMsg > _interval) ? _toMsg + _interval : Long.MAX_VALUE;
                _toMsgText.setText(_toMsg.toString());
               
                refresh(_mbean);
            }
        });
       
        //message table
        Composite tableAndButtonsComposite = _toolkit.createComposite(messagesGroup);
        gridData = new GridData(SWT.FILL, SWT.FILL, true, true);
        gridData.minimumHeight = 180;
        gridData.heightHint = 180;
        tableAndButtonsComposite.setLayoutData(gridData);
        tableAndButtonsComposite.setLayout(new GridLayout(2,false));
              
        _table = new Table (tableAndButtonsComposite, SWT.MULTI | SWT.SCROLL_LINE | SWT.BORDER | SWT.FULL_SELECTION);
        _table.setLinesVisible (true);
        _table.setHeaderVisible (true);
        GridData data = new GridData(SWT.FILL, SWT.FILL, true, true);
        _table.setLayoutData(data);
       
        _tableViewer = new TableViewer(_table);
        final TableSorter tableSorter = new TableSorter();
       
        String[] titles = {"AMQ ID", "Size(bytes)"};
        if(_ApiVersion.greaterThanOrEqualTo(1, 3))
        {
           //if server management API is >= 1.3, show message's queue position
           titles = new String[]{"AMQ ID", "Size(bytes)", "Queue Position"};
        }

        int[] bounds = { 175, 175, 140 };
        for (int i = 0; i < titles.length; i++)
        {
            final int index = i;
            final TableColumn column = new TableColumn (_table, SWT.NONE);

            column.setText(titles[i]);
            column.setWidth(bounds[i]);
            column.setResizable(true);

            //Setting the right sorter
            column.addSelectionListener(new SelectionAdapter()
            {
                @Override
                public void widgetSelected(SelectionEvent e)
                {
                    tableSorter.setColumn(index);
                    final TableViewer viewer = _tableViewer;
                    int dir = viewer .getTable().getSortDirection();
                    if (viewer.getTable().getSortColumn() == column)
                    {
                        dir = dir == SWT.UP ? SWT.DOWN : SWT.UP;
                    }
                    else
                    {
                        dir = SWT.UP;
                    }
                    viewer.getTable().setSortDirection(dir);
                    viewer.getTable().setSortColumn(column);
                    viewer.refresh();
                }
            });

        }
       
        _tableViewer.setContentProvider(new ContentProviderImpl());
        _tableViewer.setLabelProvider(new LabelProviderImpl());
        _tableViewer.setSorter(tableSorter);
        _table.setSortColumn(_table.getColumn(0));
        _table.setSortDirection(SWT.UP);
       
        //Side Buttons
        Composite buttonsComposite = _toolkit.createComposite(tableAndButtonsComposite);
        gridData = new GridData(SWT.FILL, SWT.FILL, false, true);
        buttonsComposite.setLayoutData(gridData);
        buttonsComposite.setLayout(new GridLayout());
       
        final Button viewSelectedMsgButton = _toolkit.createButton(buttonsComposite, "View Message Content ...", SWT.PUSH);
        viewSelectedMsgButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false));
        viewSelectedMsgButton.setEnabled(false);
        viewSelectedMsgButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                if (_table.getSelectionIndex() == -1)
                {
                    return;
                }
               
                viewMessageContent();
            }
        });
       
        if(_ApiVersion.lessThan(1, 3)) //if the server predates Qpid JMX API 1.3
        {
            final Button deleteFirstMessageButton = _toolkit.createButton(buttonsComposite, "Delete 1st Unacquired Msg", SWT.PUSH);
            deleteFirstMessageButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false));
            deleteFirstMessageButton.addSelectionListener(new SelectionAdapter()
            {
                public void widgetSelected(SelectionEvent se)
                {
                    int response = ViewUtility.popupOkCancelConfirmationMessage("Delete 1st unacquired message",
                                                                                "Delete 1st unacquired message on the queue?\n\n"
                                                                              + "NOTE: Any ongoing consumer activity may mean this is "
                                                                              + "not the first message listed in the table.");
                    if (response == SWT.OK)
                    {
                        try
                        {
                            _qmb.deleteMessageFromTop();
                            ViewUtility.operationResultFeedback(null, "Deleted 1st unacquired message on the queue", null);
                        }
                        catch (Exception e)
                        {
                            ViewUtility.operationFailedStatusBarMessage("Error deleting 1st unacquired message on queue");
                            MBeanUtility.handleException(_mbean, e);
                        }

                        refresh(_mbean);;
                    }
                }
            });
        }
       
        final Button moveMessagesButton;
        if(_ApiVersion.greaterThanOrEqualTo(1, 3))
        {
            //If the server supports Qpid JMX API 1.3, show the move message button.
            //This is being disabled for earlier brokers due to bugs affecting the result appearance
            //and impacting on the ability of the source queues to deliver further messages.
           
            moveMessagesButton = _toolkit.createButton(buttonsComposite, "Move Message(s) ...", SWT.PUSH);

            moveMessagesButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false));
            moveMessagesButton.setEnabled(false);
            moveMessagesButton.addSelectionListener(new SelectionAdapter()
            {
                public void widgetSelected(SelectionEvent e)
                {
                    if (_table.getSelectionIndex() == -1)
                    {
                        return;
                    }

                    moveOrCopyMessages(moveMessagesButton.getShell(), QueueOperations.MOVE);
                }
            });
        }
        else
        {
            moveMessagesButton = null;
        }
       
        final Button copyMessagesButton;
        if(_ApiVersion.greaterThanOrEqualTo(1, 3))//if the server supports Qpid JMX API 1.3
        {
            copyMessagesButton= _toolkit.createButton(buttonsComposite, "Copy Message(s) ...", SWT.PUSH);
            copyMessagesButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false));
            copyMessagesButton.setEnabled(false);
            copyMessagesButton.addSelectionListener(new SelectionAdapter()
            {
                public void widgetSelected(SelectionEvent e)
                {
                    if (_table.getSelectionIndex() == -1)
                    {
                        return;
                    }

                    moveOrCopyMessages(copyMessagesButton.getShell(), QueueOperations.COPY);
                }
            });
        }
        else
        {
            copyMessagesButton = null;
        }

        final Button deleteMessagesButton;
        if(_ApiVersion.greaterThanOrEqualTo(1, 3))//if the server supports Qpid JMX API 1.3
        {
            deleteMessagesButton = _toolkit.createButton(buttonsComposite, "Delete Message(s) ...", SWT.PUSH);
            deleteMessagesButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false));
            deleteMessagesButton.setEnabled(false);
            deleteMessagesButton.addSelectionListener(new SelectionAdapter()
            {
                public void widgetSelected(SelectionEvent e)
                {
                    if (_table.getSelectionIndex() == -1)
                    {
                        return;
                    }

                    deleteMessages(deleteMessagesButton.getShell());
                }
            });
        }
        else
        {
            deleteMessagesButton = null;
        }
               
        final Button clearQueueButton = _toolkit.createButton(buttonsComposite, "Clear Queue", SWT.PUSH);
        clearQueueButton.setLayoutData(new GridData(SWT.FILL, SWT.TOP, false, false));
        clearQueueButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                int response = ViewUtility.popupOkCancelConfirmationMessage("Clear Queue",
                        "Clear queue ?");
                if (response == SWT.OK)
                {
                    try
                    {
                        if(_ApiVersion.greaterThanOrEqualTo(1, 3))
                        {
                            //Qpid JMX API 1.3+, returns the number of messages deleted
                            Long numDeleted = _qmb.clearQueue();
                            String message = "Queue cleared of " + numDeleted +
                                             " non-acquired message" + (numDeleted == 1 ? "": "s");
                           
                            ViewUtility.operationResultFeedback(null, message, null);
                        }
                        else
                        {
                            //Qpid JMX API 1.2 or below, void return
                            _qmb.clearQueue();
                            ViewUtility.operationResultFeedback(null, "Queue cleared of non-acquired messages", null);
                        }
                    }
                    catch (Exception e2)
                    {
                        ViewUtility.operationFailedStatusBarMessage("Error clearing Queue");
                        MBeanUtility.handleException(_mbean, e2);
                    }

                    refresh(_mbean);;
                }
            }
        });
       
        _toolkit.createLabel(messagesGroup, "Message Header: ");
       
        //Redelivered status and header
        Composite headerEtcComposite = _toolkit.createComposite(messagesGroup);
        gridData = new GridData(SWT.FILL, SWT.FILL, true, false);
        headerEtcComposite.setLayoutData(gridData);
        headerEtcComposite.setLayout(new GridLayout());
       
        final Text headerText = new Text(headerEtcComposite, SWT.WRAP | SWT.BORDER | SWT.V_SCROLL);
        headerText.setText("Select a message to view its header.");
        headerText.setEditable(false);
        data = new GridData(SWT.LEFT, SWT.TOP, false, false);
        data.minimumHeight = 210;
        data.heightHint = 210;
        data.minimumWidth = 500;
        data.widthHint = 500;
        headerText.setLayoutData(data);
       
        Composite redeliveryComposite = _toolkit.createComposite(headerEtcComposite);
        redeliveryComposite.setLayout(new GridLayout(2,false));
        data = new GridData(SWT.LEFT, SWT.FILL, false, false);
        data.minimumWidth = 150;
        data.widthHint = 150;
        redeliveryComposite.setLayoutData(data);
       
        _toolkit.createLabel(redeliveryComposite, "Redelivered: ");
        final Text redeliveredText = new Text(redeliveryComposite, SWT.BORDER);
        redeliveredText.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false));
        redeliveredText.setText("-");
        redeliveredText.setEditable(false);
       
        //listener for double clicking to view message content
        _table.addMouseListener(new MouseListener()                                             
        {
            // MouseListener implementation
            public void mouseDoubleClick(MouseEvent event)
            {
                viewMessageContent();
            }

            public void mouseDown(MouseEvent e){}
            public void mouseUp(MouseEvent e){}
        });
       
        //selection listener to enable and disable buttons, update header and redelivered info
        _tableViewer.addSelectionChangedListener(new ISelectionChangedListener(){
            public void selectionChanged(SelectionChangedEvent evt)
            {
                int selectionIndex = _table.getSelectionIndex();

                if (selectionIndex == -1)
                {
                    headerText.setText("Select a message to view its header.");
                    redeliveredText.setText("-");
                    viewSelectedMsgButton.setEnabled(false);
                    if(moveMessagesButton != null)
                    {
                        moveMessagesButton.setEnabled(false);
                    }
                    if(copyMessagesButton != null)
                    {
                        copyMessagesButton.setEnabled(false);
                    }
                    if(deleteMessagesButton != null)
                    {
                        deleteMessagesButton.setEnabled(false);
                    }

                    return;
                }
                else
                {  
                    if(moveMessagesButton != null)
                    {
                        moveMessagesButton.setEnabled(true);
                    }
                    if(copyMessagesButton != null)
                    {
                        copyMessagesButton.setEnabled(true);
                    }
                    if(deleteMessagesButton != null)
                    {
                        deleteMessagesButton.setEnabled(true);
                    }
                   
                    final CompositeData selectedMsg = (CompositeData)_table.getItem(selectionIndex).getData();
                    Boolean redelivered = (Boolean) selectedMsg.get(MSG_REDELIVERED);
                    redeliveredText.setText(redelivered.toString());

                    String[] msgHeader = (String[]) selectedMsg.get(MSG_HEADER);
                    headerText.setText("");
                    String lineSeperator = System.getProperty("line.separator");
                    int size = msgHeader.length;
                    for(int i=0; i < size; i++)
                    {
                        headerText.append(msgHeader[i]);
                        if(!(i == size - 1))
                        {
                            headerText.append(lineSeperator);
                        }
                    }
                    headerText.setSelection(0);
                }
               
                if (_table.getSelectionCount() > 1)
                {
                    viewSelectedMsgButton.setEnabled(false);
                }
                else
                {
                    viewSelectedMsgButton.setEnabled(true);
                }

            }
        });

    }
   
    /**
     * Content Provider class for the table viewer
     */
    private static class ContentProviderImpl  implements IStructuredContentProvider
    {
       
        public void inputChanged(Viewer v, Object oldInput, Object newInput)
        {
           
        }
       
        public void dispose()
        {
           
        }
       
        public Object[] getElements(Object parent)
        {
            Collection<Object> rowCollection = ((TabularDataSupport) parent).values();
          
            return rowCollection.toArray();
        }
    }
   
    /**
     * Label Provider class for the table viewer
     */
    private static class LabelProviderImpl extends LabelProvider implements ITableLabelProvider
    {
        public String getColumnText(Object element, int columnIndex)
        {
            switch (columnIndex)
            {
                case 0 : // msg id column
                    return String.valueOf(((CompositeDataSupport) element).get(MSG_AMQ_ID));
                case 1 : // msg size column
                    return String.valueOf(((CompositeDataSupport) element).get(MSG_SIZE));
                case 2 : // msg position in queue
                    return String.valueOf(((CompositeDataSupport) element).get(MSG_QUEUE_POS));
                default :
                    return "-";
            }
        }
       
        public Image getColumnImage(Object element, int columnIndex)
        {
            return null;
        }
       
    }

    /**
     * Sorter class for the table viewer.
     *
     */
    public static class TableSorter extends ViewerSorter
    {
        private int column;
        private static final int ASCENDING = 0;
        private static final int DESCENDING = 1;

        private int direction = DESCENDING;

        public TableSorter()
        {
            this.column = 0;
            direction = ASCENDING;
        }

        public void setColumn(int column)
        {
            if (column == this.column)
            {
                // Same column as last sort; toggle the direction
                direction = 1 - direction;
            }
            else
            {
                // New column; do an ascending sort
                this.column = column;
                direction = ASCENDING;
            }
        }

        @Override
        public int compare(Viewer viewer, Object e1, Object e2)
        {
            CompositeData msg1 = (CompositeData) e1;
            CompositeData msg2 = (CompositeData) e2;
           
            int comparison = 0;
            switch(column)
            {
                case 0:
                    comparison = ((Long) msg1.get(MSG_AMQ_ID)).compareTo((Long)msg2.get(MSG_AMQ_ID));
                    break;
                case 1:
                    comparison = ((Long) msg1.get(MSG_SIZE)).compareTo((Long)msg2.get(MSG_SIZE));
                    break;
                case 2:
                    comparison = ((Long) msg1.get(MSG_QUEUE_POS)).compareTo((Long)msg2.get(MSG_QUEUE_POS));
                    break;
                default:
                    comparison = 0;
            }
            // If descending order, flip the direction
            if(direction == DESCENDING)
            {
                comparison = -comparison;
            }
            return comparison;
        }
    }
   
    private void updateMessageInterval()
    {
        Long from;
        try
        {
            from = Long.valueOf(_fromMsgText.getText());
        }
        catch(Exception e1)
        {
            ViewUtility.popupErrorMessage("Invalid Value", "Please enter a valid 'from' number");
            return;
        }
       
        Long to;
        try
        {
            to = Long.valueOf(_toMsgText.getText());
        }
        catch(Exception e1)
        {
            ViewUtility.popupErrorMessage("Invalid Value", "Please enter a valid 'to' number");
            return;
        }

        _fromMsg = from;
        _toMsg = to;
       
        _interval = (to - from) + 1;
        _previousButton.setText("< Prev " + _interval);
        _nextButton.setText("Next " + _interval + " >");
       
        refresh(_mbean);
    }
   
    private void viewMessageContent()
    {
        int selectionIndex = _table.getSelectionIndex();

        if (selectionIndex != -1)
        {
            try
            {
                final CompositeData selectedMsg = (CompositeData)_table.getItem(selectionIndex).getData();
                Long msgId = (Long) selectedMsg.get(MSG_AMQ_ID);

                Object result = _qmb.viewMessageContent(msgId);

                populateResults(result);
            }
            catch (Exception e3)
            {
                MBeanUtility.handleException(_mbean, e3);
            }
        }
    }
   
    private void populateResults(Object result)
    {
        Display display = Display.getCurrent();
        int width = 610;
        int height = 400;
        Shell shell = ViewUtility.createPopupShell(RESULT, width, height);
        shell.setImage(ApplicationRegistry.getImage(CONSOLE_IMAGE));
        ViewUtility.populateCompositeWithData(_toolkit, shell, result);
       
        shell.open();
        while (!shell.isDisposed())
        {
            if (!display.readAndDispatch())
            {
                display.sleep();
            }
        }
        shell.dispose();
    }
   
    private void moveOrCopyMessages(final Shell parent, final QueueOperations op)
    {
        final ArrayList<Long> rangeStarts = new ArrayList<Long>();
        final ArrayList<Long> rangeEnds = new ArrayList<Long>();

        gatherSelectedAMQMsgIDRanges(rangeStarts,rangeEnds);
        String rangeString = getRangesString(rangeStarts,rangeEnds);
       
        String windowTitle;
        String dialogueMessage;
        final String feedBackMessage;
        final String failureFeedBackMessage;
       
        if(op.equals(QueueOperations.MOVE))
        {
            windowTitle = "Move Messages";
            dialogueMessage = "Move message(s) with AMQ ID:";
            feedBackMessage = "Messages moved";
            failureFeedBackMessage = "Error moving messages";
        }
        else
        {
            windowTitle = "Copy Messages";
            dialogueMessage = "Copy message(s) with AMQ ID:";
            feedBackMessage = "Messages copied";
            failureFeedBackMessage = "Error copying messages";
        }
       
        final Shell shell = ViewUtility.createModalDialogShell(parent, windowTitle);

        Composite idComposite = _toolkit.createComposite(shell, SWT.NONE);
        idComposite.setBackground(shell.getBackground());
        idComposite.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
        idComposite.setLayout(new GridLayout());
       
        _toolkit.createLabel(idComposite,dialogueMessage).setBackground(shell.getBackground());
        _toolkit.createLabel(idComposite,rangeString).setBackground(shell.getBackground());

        Composite destinationComposite = _toolkit.createComposite(shell, SWT.NONE);
        destinationComposite.setBackground(shell.getBackground());
        destinationComposite.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
        destinationComposite.setLayout(new GridLayout(2,false));
       
        _toolkit.createLabel(destinationComposite,"To Queue:").setBackground(shell.getBackground());
        final Combo destinationCombo = new Combo(destinationComposite,SWT.NONE | SWT.READ_ONLY);
        destinationCombo.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
       
        Composite okCancelButtonsComp = _toolkit.createComposite(shell);
        okCancelButtonsComp.setBackground(shell.getBackground());
        okCancelButtonsComp.setLayoutData(new GridData(SWT.RIGHT, SWT.FILL, true, true));
        okCancelButtonsComp.setLayout(new GridLayout(2,false));
       
        Button okButton = _toolkit.createButton(okCancelButtonsComp, "OK", SWT.PUSH);
        okButton.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, false, false));
        Button cancelButton = _toolkit.createButton(okCancelButtonsComp, "Cancel", SWT.PUSH);
        cancelButton.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, false, false));
       
        List<String> queueList = ApplicationRegistry.getServerRegistry(_mbean).getQueueNames(_mbean.getVirtualHostName());
        queueList.remove(_mbean.getName());
       
        if(queueList.size() == 0)
        {
            destinationCombo.setItems(new String[]{"No other queues available"});
            okButton.setEnabled(false);
        }
        else
        {
            Collections.sort(queueList);
            destinationCombo.setItems(queueList.toArray(new String[0]));
        }
        destinationCombo.select(0);
       
        okButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                String destQueue = destinationCombo.getItem(destinationCombo.getSelectionIndex()).toString();
                shell.dispose();

                try
                {
                    for(int i=0 ; i < rangeStarts.size() ; i++)
                    {
                        Long from = rangeStarts.get(i);
                        Long to = rangeEnds.get(i);
                       
                        switch(op)
                        {
                            case COPY:
                                _qmb.copyMessages(Long.valueOf(from), Long.valueOf(to), destQueue);
                                break;
                            case MOVE:
                                _qmb.moveMessages(Long.valueOf(from), Long.valueOf(to), destQueue);
                                break;
                        }
                    }
                   
                    ViewUtility.operationResultFeedback(null, feedBackMessage, null);
                }
                catch (Exception e4)
                {
                    ViewUtility.operationFailedStatusBarMessage(failureFeedBackMessage);
                    MBeanUtility.handleException(_mbean, e4);
                }

                refresh(_mbean);
            }
        });

        cancelButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                shell.dispose();
            }
        });

        shell.setDefaultButton(okButton);
        shell.pack();
        ViewUtility.centerChildInParentShell(parent, shell);
       
        shell.open();
    }
   
    private void deleteMessages(final Shell parent)
    {
        final ArrayList<Long> rangeStarts = new ArrayList<Long>();
        final ArrayList<Long> rangeEnds = new ArrayList<Long>();

        gatherSelectedAMQMsgIDRanges(rangeStarts,rangeEnds);
        String rangeString = getRangesString(rangeStarts,rangeEnds);
       
        final Shell shell = ViewUtility.createModalDialogShell(parent, "Delete Messages");

        Composite idComposite = _toolkit.createComposite(shell, SWT.NONE);
        idComposite.setBackground(shell.getBackground());
        idComposite.setLayoutData(new GridData(SWT.FILL, SWT.TOP, true, false));
        idComposite.setLayout(new GridLayout());
       
        _toolkit.createLabel(idComposite,"Delete message(s) with AMQ ID:").setBackground(shell.getBackground());
        _toolkit.createLabel(idComposite,rangeString).setBackground(shell.getBackground());

        Composite okCancelButtonsComp = _toolkit.createComposite(shell);
        okCancelButtonsComp.setBackground(shell.getBackground());
        okCancelButtonsComp.setLayoutData(new GridData(SWT.RIGHT, SWT.FILL, true, true));
        okCancelButtonsComp.setLayout(new GridLayout(2,false));
       
        Button okButton = _toolkit.createButton(okCancelButtonsComp, "OK", SWT.PUSH);
        okButton.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, false, false));
        Button cancelButton = _toolkit.createButton(okCancelButtonsComp, "Cancel", SWT.PUSH);
        cancelButton.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, false, false));
       
        okButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                shell.dispose();
               
                try
                {
                    for(int i=0 ; i < rangeStarts.size() ; i++)
                    {
                        Long from = rangeStarts.get(i);
                        Long to = rangeEnds.get(i);
                       
                        _qmb.deleteMessages(Long.valueOf(from), Long.valueOf(to));
                    }

                    ViewUtility.operationResultFeedback(null, "Messages deleted", null);
                }
                catch (Exception e4)
                {
                    ViewUtility.operationFailedStatusBarMessage("Error deleting messages");
                    MBeanUtility.handleException(_mbean, e4);
                }

                refresh(_mbean);
            }
        });

        cancelButton.addSelectionListener(new SelectionAdapter()
        {
            public void widgetSelected(SelectionEvent e)
            {
                shell.dispose();
            }
        });

        shell.setDefaultButton(okButton);
        shell.pack();
        ViewUtility.centerChildInParentShell(parent, shell);
       
        shell.open();
    }
   
    private void gatherSelectedAMQMsgIDRanges(ArrayList<Long> starts, ArrayList<Long> ends)
    {
        SortedSet<Long> amqIDs = new TreeSet<Long>();
       
        for(Integer i : _table.getSelectionIndices())
        {
            CompositeData selectedMsg = (CompositeData)_table.getItem(i).getData();
            amqIDs.add((Long)selectedMsg.get(MSG_AMQ_ID));
        }
               
        //initialise the first range
        Long start = amqIDs.first();
        Long end = amqIDs.first();
       
        for(Long id : amqIDs)
        {
            if(id == amqIDs.first())
            {
                //skip first check, already initialised range
                continue;
            }
           
            if(id == end +1)
            {
                //part of previous range, append
                end = id;
            }
            else
            {
                //not in previous range, record existing start and end msg id values
                starts.add(start);
                ends.add(end);
               
                //begin new range with this msg id
                start = id;
                end = id;
            }
        }
       
        //record the last range created
        starts.add(start);
        ends.add(end);
    }
   
    private String getRangesString(ArrayList<Long> starts, ArrayList<Long> ends)
    {
        String idRangesString = "";

        for(int i=0 ; i < starts.size() ; i++)
        {
            long start = starts.get(i);
            long end = ends.get(i);
           
            if(i != 0)
            {
                idRangesString = idRangesString.concat(", ");
            }
           
            if(start == end)
            {
                idRangesString = idRangesString.concat(String.valueOf(starts.get(i)));
            }
            else
            {
                idRangesString = idRangesString.concat(starts.get(i) + "-" + ends.get(i));
            }
        }
       
        return idRangesString.concat(".");
    }
   
    private enum QueueOperations
    {
        MOVE,
        COPY;
    }
}
TOP

Related Classes of org.apache.qpid.management.ui.views.queue.QueueOperationsTabControl$TableSorter

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.