muni.fi.dp.jz.jbatch.cli.CliBatchManagerBean.java Source code

Introduction

Here is the source code for muni.fi.dp.jz.jbatch.cli.CliBatchManagerBean.java

Source

/*
 * Copyright 2016 JBoss by Red Hat.
 *
 * 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 muni.fi.dp.jz.jbatch.cli;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import javax.ejb.Stateless;
import org.apache.log4j.Logger;
import org.jboss.as.cli.CliInitializationException;
import org.jboss.as.cli.CommandContext;
import org.jboss.as.cli.CommandContextFactory;
import org.jboss.as.cli.CommandLineException;
import org.jboss.as.controller.client.ModelControllerClient;
import org.jboss.dmr.ModelNode;
import org.json.JSONArray;
import org.json.JSONObject;

/**
 *
 * @author Zorz
 */
@Stateless
public class CliBatchManagerBean {

    private static final Logger LOG = Logger.getLogger(CliBatchManagerBean.class.getName());

    public CliBatchManagerBean() {
    }

    public String startJobCli(String deploymentName, String jobName) {
        String startJobCli = "/deployment=" + deploymentName + "/subsystem=batch-jberet:start-job(job-xml-name="
                + jobName + ")";
        //        TODO - if runCommand == null throw exception and return null;
        return runCommand(startJobCli);
    }

    public String startJobCli(String deploymentName, String jobName, Properties properties) {
        String startJobCli = "/deployment=" + deploymentName + "/subsystem=batch-jberet:start-job(job-xml-name="
                + jobName + ",properties={" + properties + "})";
        //        TODO - if runCommand == null throw exception and return null;
        return runCommand(startJobCli);
    }

    public String getDeploymentInfo() {
        String getInfo = "deployment-info";
        String resp = runCommand(getInfo);
        //       LOG.info("Resp: " + resp);
        JSONObject json = new JSONObject(resp);
        JSONObject result = json.getJSONObject("result");
        return result.toString();
    }

    public String getBatchDeploymentsWithJobs() {
        String allDeployments = getDeploymentInfo();
        JSONObject json = new JSONObject(allDeployments);
        //       Go through json object and filter deployments with batch-jberet subsystem only
        List<String> batchDeploymentsList = new ArrayList<>();
        Iterator<?> keys = json.keys();
        while (keys.hasNext()) {
            String key = (String) keys.next();
            JSONObject subsystems = json.getJSONObject(key).getJSONObject("subsystem");
            if (subsystems.has("batch-jberet")) {
                batchDeploymentsList.add(key);
            }
        }
        //        Pair deployments with their available batch jobs
        Map<String, JSONArray> deploymentsJobsMap = new HashMap<>();
        for (String batchDeployment : batchDeploymentsList) {
            String jobs = getJobsFromDeployment(batchDeployment);
            JSONArray jsonJobs = new JSONArray(jobs);
            deploymentsJobsMap.put(batchDeployment, jsonJobs);
        }
        JSONObject jsonMap = new JSONObject(deploymentsJobsMap);
        return jsonMap.toString();
    }

    public String getJobsFromDeployment(String deployment) {
        String deploymentJobs = runCommand("/deployment=" + deployment + "/subsystem=batch-jberet:read-resource");
        JSONObject jsonJobs = new JSONObject(deploymentJobs);
        //        names() returns JSONArray of keys
        JSONArray jobNamesArray = jsonJobs.getJSONObject("result").getJSONObject("job").names();
        JSONObject res = new JSONObject();
        res.put(deployment, jobNamesArray);

        return jobNamesArray.toString();
    }

    //Help methods
    public String runCommand(String command) {
        final CommandContext ctx;
        try {
            ctx = CommandContextFactory.getInstance().newCommandContext();

            try {
                // connect to the server controller
                ctx.connectController();
                // execute commands and operations                    
                String response = executeCommand(ctx, ctx.buildRequest(command));
                return response;
            } catch (CommandLineException e) {
                //                    CommandLineException not found with scope provided
                System.out.println("Exception when submitting command to server:" + e.toString());
            }

        } catch (CliInitializationException e) {
            //               CliInitializationException not found with scope provided
            System.out.println("Exception when creating the ctx:" + e.toString());
        }
        return null;
    }

    public static String executeCommand(CommandContext ctx, ModelNode modelNode) {

        ModelControllerClient client = ctx.getModelControllerClient();
        if (client != null) {
            try {
                ModelNode response = client.execute(modelNode);
                System.out.println(response);
                return (response.toJSONString(true));
            } catch (IOException e) {
                System.out.println("IOException thrown when executing command: " + e.toString());
            }
        } else {
            System.out.println("Connection Error! The ModelControllerClient is not available.");
        }
        return null;
    }

}