Java tutorial
/* * #%L * ACS AEM Commons Bundle * %% * Copyright (C) 2013 Adobe * %% * 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. * #L% */ package com.adobe.acs.commons.workflow.bulk.execution.impl.servlets; import com.adobe.acs.commons.fam.ActionManager; import com.adobe.acs.commons.fam.ActionManagerFactory; import com.adobe.acs.commons.fam.impl.ThrottledTaskRunnerStats; import com.adobe.acs.commons.workflow.bulk.execution.BulkWorkflowEngine; import com.adobe.acs.commons.workflow.bulk.execution.model.Status; import com.adobe.acs.commons.workflow.bulk.execution.impl.runners.AEMWorkflowRunnerImpl; import com.adobe.acs.commons.workflow.bulk.execution.model.Config; import com.adobe.acs.commons.workflow.bulk.execution.model.Failure; import com.adobe.acs.commons.workflow.bulk.execution.model.Payload; import com.adobe.acs.commons.workflow.bulk.execution.model.Workspace; import org.apache.commons.lang.StringUtils; import org.apache.felix.scr.annotations.Reference; import org.apache.felix.scr.annotations.sling.SlingServlet; import org.apache.sling.api.SlingHttpServletRequest; import org.apache.sling.api.SlingHttpServletResponse; import org.apache.sling.api.servlets.SlingAllMethodsServlet; import org.apache.sling.commons.json.JSONException; import org.apache.sling.commons.json.JSONObject; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import javax.management.InstanceNotFoundException; import javax.management.ReflectionException; import javax.servlet.ServletException; import java.io.IOException; import java.text.MessageFormat; import java.text.SimpleDateFormat; import java.util.Calendar; /** * ACS AEM Commons - Bulk Workflow Manager - Status Servlet */ @SuppressWarnings("serial") @SlingServlet(methods = { "GET" }, resourceTypes = { BulkWorkflowEngine.SLING_RESOURCE_TYPE }, selectors = { "status" }, extensions = { "json" }) public class StatusServlet extends SlingAllMethodsServlet { private static final Logger log = LoggerFactory.getLogger(StatusServlet.class); private static final int DECIMAL_TO_PERCENT = 100; @Reference private ThrottledTaskRunnerStats ttrs; @Reference private ActionManagerFactory actionManagerFactory; @Override @SuppressWarnings({ "squid:S3776", "squid:S1192" }) protected final void doGet(SlingHttpServletRequest request, SlingHttpServletResponse response) throws ServletException, IOException { SimpleDateFormat sdf = new SimpleDateFormat("EEE, d MMM yyyy hh:mm:ss aaa"); response.setContentType("application/json"); response.setCharacterEncoding("UTF-8"); Config config = request.getResource().adaptTo(Config.class); Workspace workspace = config.getWorkspace(); final JSONObject json = new JSONObject(); try { json.put("initialized", workspace.isInitialized()); json.put("status", workspace.getStatus()); if (workspace.getSubStatus() != null) { json.put("subStatus", workspace.getSubStatus()); } json.put("runnerType", config.getRunnerType()); json.put("queryType", config.getQueryType()); json.put("queryStatement", config.getQueryStatement()); json.put("workflowModel", StringUtils.removeEnd(config.getWorkflowModelId(), "/jcr:content/model")); json.put("batchSize", config.getBatchSize()); json.put("autoThrottle", config.isAutoThrottle()); json.put("purgeWorkflow", config.isPurgeWorkflow()); json.put("interval", config.getInterval()); json.put("retryCount", config.getRetryCount()); json.put("timeout", config.getTimeout()); json.put("throttle", config.getThrottle()); json.put("message", workspace.getMessage()); if (config.isUserEventData()) { json.put("userEventData", config.getUserEventData()); } ActionManager actionManager = actionManagerFactory.getActionManager(workspace.getActionManagerName()); if (actionManager != null && !Status.COMPLETED.equals(workspace.getStatus())) { // If Complete, then look to JCR for final accounts as ActionManager may be gone addActionManagerTrackedCounts(workspace.getActionManagerName(), json); for (com.adobe.acs.commons.fam.Failure failure : actionManager.getFailureList()) { JSONObject failureJSON = new JSONObject(); failureJSON.put(Failure.PN_PAYLOAD_PATH, failure.getNodePath()); failureJSON.put(Failure.PN_FAILED_AT, sdf.format(failure.getTime().getTime())); json.accumulate("failures", failureJSON); } } else { addWorkspaceTrackedCounts(workspace, json); // Failures for (Failure failure : workspace.getFailures()) { json.accumulate("failures", failure.toJSON()); } } // Times if (workspace.getStartedAt() != null) { json.put("startedAt", sdf.format(workspace.getStartedAt().getTime())); json.put("timeTakenInMillis", (Calendar.getInstance().getTime().getTime() - workspace.getStartedAt().getTime().getTime())); } if (workspace.getStoppedAt() != null) { json.put("stoppedAt", sdf.format(workspace.getStoppedAt().getTime())); json.put("timeTakenInMillis", (workspace.getStoppedAt().getTime().getTime() - workspace.getStartedAt().getTime().getTime())); } if (workspace.getCompletedAt() != null) { json.put("completedAt", sdf.format(workspace.getCompletedAt().getTime())); json.put("timeTakenInMillis", (workspace.getCompletedAt().getTime().getTime() - workspace.getStartedAt().getTime().getTime())); } if (AEMWorkflowRunnerImpl.class.getName().equals(config.getRunnerType())) { for (Payload payload : config.getWorkspace().getActivePayloads()) { json.accumulate("activePayloads", payload.toJSON()); } } json.put("systemStats", getSystemStats()); response.getWriter().write(json.toString()); } catch (JSONException e) { log.error("Could not collect Bulk Workflow status due to: {}", e); JSONErrorUtil.sendJSONError(response, SlingHttpServletResponse.SC_INTERNAL_SERVER_ERROR, "Could not collect Bulk Workflow status.", e.getMessage()); } } private void addActionManagerTrackedCounts(String name, JSONObject json) throws JSONException { final ActionManager actionManager = actionManagerFactory.getActionManager(name); int failureCount = actionManager.getErrorCount(); int completeCount = actionManager.getSuccessCount(); int totalCount = actionManager.getAddedCount(); int remainingCount = actionManager.getRemainingCount(); json.put("totalCount", totalCount); json.put("completeCount", completeCount); json.put("remainingCount", remainingCount); json.put("failCount", failureCount); json.put("percentComplete", Math.round(((totalCount - remainingCount) / (totalCount * 1F)) * DECIMAL_TO_PERCENT)); } private void addWorkspaceTrackedCounts(Workspace workspace, JSONObject json) throws JSONException { // Counts int remainingCount = workspace.getTotalCount() - (workspace.getCompleteCount() + workspace.getFailCount()); json.put("totalCount", workspace.getTotalCount()); json.put("completeCount", workspace.getCompleteCount()); json.put("remainingCount", remainingCount); json.put("failCount", workspace.getFailCount()); json.put("percentComplete", Math.round(((workspace.getTotalCount() - remainingCount) / (workspace.getTotalCount() * 1F)) * DECIMAL_TO_PERCENT)); } @SuppressWarnings("squid:S1192") private JSONObject getSystemStats() throws JSONException { JSONObject json = new JSONObject(); try { json.put("cpu", MessageFormat.format("{0,number,#%}", ttrs.getCpuLevel())); } catch (InstanceNotFoundException e) { log.error("Could not collect CPU stats", e); json.put("cpu", -1); } catch (ReflectionException e) { log.error("Could not collect CPU stats", e); json.put("cpu", -1); } json.put("mem", MessageFormat.format("{0,number,#%}", ttrs.getMemoryUsage())); json.put("maxCpu", MessageFormat.format("{0,number,#%}", ttrs.getMaxCpu())); json.put("maxMem", MessageFormat.format("{0,number,#%}", ttrs.getMaxHeap())); return json; } }