/*
 * (C) 2007-2012 Alibaba Group Holding Limited.
 * 
 * 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.
 * Authors:
 *   wuhua <wq163@163.com> , boyan <killme2008@gmail.com>
 */
package com.taobao.metamorphosis.tools.shell;

import java.io.PrintStream;
import java.io.PrintWriter;

import javax.management.ObjectInstance;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.Options;

import com.taobao.metamorphosis.tools.utils.CommandLineUtils;
import com.taobao.metamorphosis.tools.utils.JMXClient;


/**
 * <pre>
 * Stop broker
 * usage:
 *      StopBrokerTool -host 10.2.2.3 -port 9999
 * </pre>
 * 
 * @author �޻�
 * @since 2011-8-23 ����3:42:38
 */

public class StopBrokerTool extends ShellTool {

    public StopBrokerTool(PrintStream out) {
        super(out);
    }


    public StopBrokerTool(PrintWriter writer) {
        super(writer);
    }


    public static void main(String[] args) throws Exception {
        new StopBrokerTool(System.out).doMain(args);
    }

    final static protected String METABROKER_NAME =
            "com.taobao.metamorphosis.server.assembly:type=MetaMorphosisBroker,*";


    @Override
    public void doMain(String[] args) throws Exception {
        CommandLine commandLine = this.getCommandLine(args);

        String host = commandLine.getOptionValue("host", "127.0.0.1");
        int port = Integer.parseInt(commandLine.getOptionValue("port", "9999"));

        JMXClient jmxClient = JMXClient.getJMXClient(host, port);
        this.println("connected to " + jmxClient.getAddressAsString());
        ObjectInstance brokerInstance = jmxClient.queryMBeanForOne(METABROKER_NAME);

        if (brokerInstance != null) {
            jmxClient.invoke(brokerInstance.getObjectName(), "stop", new Object[0], new String[0]);
            jmxClient.close();
            this.println("invoke " + brokerInstance.getClassName() + "#stop success");
        }
        else {
            this.println("û���ҵ� " + METABROKER_NAME);
        }
    }


    private CommandLine getCommandLine(String[] args) {
        return CommandLineUtils.parseCmdLine(args,
            new Options().addOption("host", true, "host").addOption("port", true, "port"));
    }
}