Jboss EAP：native management API学习

import com.sun.javafx.sg.PGShape;
import org.jboss.as.controller.client.ModelControllerClient;
import org.jboss.as.controller.client.helpers.ClientConstants;
import org.jboss.dmr.ModelNode;
import org.junit.Test;

import javax.security.auth.callback.*;
import javax.security.sasl.RealmCallback;
import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.List;

public class JBossClient {

    private String host = "172.16.38.***";
    private int port = 9999;
    private String userid = "jimmy";
    private String password = "*****";

    @Test
    public void testGetServers() {
        //相当于CLI命令行： ls /host=master/server-config
        List<String> servers = getServers("master");
        for (String s : servers) {
            System.out.println(s);
        }
    }

    @Test
    public void getServerStatus() {
        //相当于CLI命令行：/host=master/server=server-one:read-attribute(name=server-state)
        System.out.println(getServerStatus("master", "server-one"));
        //相当于CLI命令行：/host=master/server-config=server-one:read-attribute(name=status)
        System.out.println(getServerStatus2("master", "server-one"));
    }

    @Test
    public void testStartServer() {
        //相当于CLI命令行：/host=master/server-config=server-one:start
        System.out.println(startServer("master", "server-one"));
    }

    @Test
    public void testStopServer() {
        //相当于CLI命令行：/host=master/server-config=server-one:stop
        System.out.println(stopServer("master", "server-one"));
    }

    /**
     * 获取指定服务器运行状态
     * @param hostName
     * @param serverName
     * @return
     */
    public String getServerStatus(String hostName, String serverName) {
        String status = "unknown";
        ModelControllerClient client = null;
        try {
            client = createClient(InetAddress.getByName(host), port, userid, password.toCharArray(), "ManagementRealm");
        } catch (UnknownHostException uhe) {
            uhe.printStackTrace();
            System.out.println("UHE: " + uhe.getMessage());
        }
        try {
            ModelNode op = new ModelNode();
            op.get(ClientConstants.OP).set(ClientConstants.READ_ATTRIBUTE_OPERATION);
            op.get(ClientConstants.OP_ADDR).add("host", hostName);
            op.get(ClientConstants.OP_ADDR).add("server", serverName);
            op.get("name").set("server-state");

            status = client.execute(op).get(ClientConstants.RESULT).asString();

            if (client != null) client.close();
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("Exception: " + e.getMessage());
        }

        return status;
    }


    /**
     * 另一种获取服务器运行状态的方法
     * @param hostName
     * @param serverName
     * @return
     */
    public String getServerStatus2(String hostName, String serverName) {
        String status = "unknown";
        ModelControllerClient client = null;
        try {
            client = createClient(InetAddress.getByName(host), port, userid, password.toCharArray(), "ManagementRealm");
        } catch (UnknownHostException uhe) {
            uhe.printStackTrace();
            System.out.println("UHE: " + uhe.getMessage());
        }
        try {
            ModelNode op = new ModelNode();
            op.get(ClientConstants.OP).set(ClientConstants.READ_ATTRIBUTE_OPERATION);
            op.get(ClientConstants.OP_ADDR).add("host", hostName);
            op.get(ClientConstants.OP_ADDR).add("server-config", serverName);
            op.get("name").set("status");

            status = client.execute(op).get(ClientConstants.RESULT).asString();

            if (client != null) client.close();
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("Exception: " + e.getMessage());
        }

        return status;
    }

    /**
     * 启动指定服务器
     *
     * @param hostName
     * @param serverName
     */
    public ModelNode startServer(String hostName, String serverName) {

        ModelControllerClient client = null;
        ModelNode returnVal = null;
        try {
            client = createClient(InetAddress.getByName(host), port, userid, password.toCharArray(), "ManagementRealm");
        } catch (UnknownHostException uhe) {
            uhe.printStackTrace();
            System.out.println("UHE: " + uhe.getMessage());
        }
        try {
            ModelNode op = new ModelNode();
            op.get(ClientConstants.OP).set("start");
            op.get(ClientConstants.OP_ADDR).add("host", hostName);
            op.get(ClientConstants.OP_ADDR).add("server-config", serverName);

            returnVal = client.execute(op).get(ClientConstants.RESULT);

            if (client != null) client.close();
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("Exception: " + e.getMessage());
        }

        return returnVal;

    }


    /**
     * 停止指定服务器
     *
     * @param hostName
     * @param serverName
     */
    public ModelNode stopServer(String hostName, String serverName) {

        ModelControllerClient client = null;
        ModelNode returnVal = null;
        try {
            client = createClient(InetAddress.getByName(host), port, userid, password.toCharArray(), "ManagementRealm");
        } catch (UnknownHostException uhe) {
            uhe.printStackTrace();
            System.out.println("UHE: " + uhe.getMessage());
        }
        try {
            ModelNode op = new ModelNode();
            op.get(ClientConstants.OP).set("stop");
            op.get(ClientConstants.OP_ADDR).add("host", hostName);
            op.get(ClientConstants.OP_ADDR).add("server-config", serverName);
            returnVal = client.execute(op).get(ClientConstants.RESULT);
            if (client != null) client.close();
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("Exception: " + e.getMessage());
        }

        return returnVal;

    }


    /**
     * 获取指定host下的所有server
     *
     * @param hostName
     * @return
     */
    public List<String> getServers(String hostName) {
        List<String> servers = new ArrayList<String>();
        ModelControllerClient client = null;
        try {
            client = createClient(InetAddress.getByName(host), 9999, userid, password.toCharArray(), "ManagementRealm");
        } catch (UnknownHostException uhe) {
            uhe.printStackTrace();
            System.out.println("UHE: " + uhe.getMessage());
        }
        try {
            ModelNode op = new ModelNode();
            op.get(ClientConstants.OP).set(ClientConstants.READ_RESOURCE_OPERATION);
            op.get(ClientConstants.OP_ADDR).add("host", hostName);
            List<ModelNode> returnVal = client.execute(op).get(ClientConstants.RESULT).get("server-config").asList();

            for (ModelNode _ : returnVal) {
                servers.add(_.asProperty().getName());
            }

            if (client != null) client.close();
        } catch (Exception e) {
            e.printStackTrace();
            System.out.println("Exception: " + e.getMessage());
        }
        return servers;
    }


    private ModelControllerClient createClient(final InetAddress host, final int port, final String username, final char[] password, final String securityRealmName) {
        final CallbackHandler callbackHandler = new CallbackHandler() {
            public void handle(Callback[] callbacks) throws IOException, UnsupportedCallbackException {
                for (Callback current : callbacks) {
                    if (current instanceof NameCallback) {
                        NameCallback ncb = (NameCallback) current;
                        ncb.setName(username);
                    } else if (current instanceof PasswordCallback) {
                        PasswordCallback pcb = (PasswordCallback) current;
                        //pcb.setPassword("admin123".toCharArray());
                        pcb.setPassword(password);
                    } else if (current instanceof RealmCallback) {
                        RealmCallback rcb = (RealmCallback) current;
                        rcb.setText(rcb.getDefaultText());
                    } else {
                        throw new UnsupportedCallbackException(current);
                    }
                }
            }
        };
        return ModelControllerClient.Factory.create(host, port, callbackHandler);
    }
}