/*
 * (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.Option;
import org.apache.commons.cli.Options;

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


/**
 * <pre>
 * �ر�һ��topic��ָ������
 * usage:
 *      ClosePartitionsTool -topic xxtopic -start 2 -end 5
 *      ClosePartitionsTool -topic xxtopic -start 2 -end 5 -host 10.2.2.3
 *      ClosePartitionsTool -topic xxtopic -start 2 -end 5 -port 9999
 *      ClosePartitionsTool -topic xxtopic -start 2 -end 5 -host 10.2.2.3 -port 9999
 * </pre>
 * 
 * @author �޻�
 * @since 2011-8-23 ����3:42:38
 */

public class ClosePartitionsTool extends ShellTool {

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


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


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


    @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"));
        String topic = commandLine.getOptionValue("topic");
        int start = Integer.parseInt(commandLine.getOptionValue("start"));
        int end = Integer.parseInt(commandLine.getOptionValue("end"));

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

        if (metaConfigInstance != null) {
            jmxClient.invoke(metaConfigInstance.getObjectName(), "closePartitions", new Object[] { topic, start, end },
                new String[] { "java.lang.String", "int", "int" });
            jmxClient.close();
            this.println("invoke " + metaConfigInstance.getClassName() + "#closePartitions success");
        }
        else {
            this.println("û���ҵ� " + METACONFIG_NAME);
        }
    }


    private CommandLine getCommandLine(String[] args) {
        Option topicOption = new Option("topic", true, "topic");
        topicOption.setRequired(true);
        Option startOption = new Option("start", true, "start partition num");
        startOption.setRequired(true);
        Option endOption = new Option("end", true, "end partition num");
        endOption.setRequired(true);
        return CommandLineUtils.parseCmdLine(args, new Options().addOption(topicOption).addOption(startOption)
            .addOption(endOption).addOption("host", true, "host").addOption("port", true, "port"));
    }
}