ca.simplegames.micro.controllers.StatsController.java Source code

Introduction

Here is the source code for ca.simplegames.micro.controllers.StatsController.java

Source

/*
 * Copyright (c)2012. Florin T.PATRASCU
 *
 * 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 ca.simplegames.micro.controllers;

import ca.simplegames.micro.Controller;
import ca.simplegames.micro.Globals;
import ca.simplegames.micro.MicroContext;
import com.sun.management.OperatingSystemMXBean;
import org.jrack.Rack;
import org.jrack.utils.Mime;
import org.json.JSONException;
import org.json.JSONObject;

import javax.management.MBeanServerConnection;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.lang.management.GarbageCollectorMXBean;
import java.lang.management.ManagementFactory;
import java.lang.management.ThreadMXBean;
import java.util.*;

/**
 * Provides a simple system info
 *
 * @author <a href="mailto:florin.patrascu@gmail.com">Florin T.PATRASCU</a>
 * @since $Revision$ (created: 2012-12-21 5:33 PM)
 */
public class StatsController implements Controller {
    private static final long MEGA_BYTE = 1048576;
    public static final String JSON_TYPE = ".json";
    public static final String POWERED_BY_MICRO = "Micro " + Globals.VERSION;

    public void execute(MicroContext context, Map configuration) throws ControllerException {
        Map<String, Object> systemInfo = new HashMap<String, Object>();
        Map<String, Object> osMap = new HashMap<String, Object>();
        MBeanServerConnection mbeanServer = ManagementFactory.getPlatformMBeanServer();

        OperatingSystemMXBean sunOperatingSystemMXBean = null;
        try {
            sunOperatingSystemMXBean = ManagementFactory.newPlatformMXBeanProxy(mbeanServer,
                    ManagementFactory.OPERATING_SYSTEM_MXBEAN_NAME, OperatingSystemMXBean.class);
        } catch (IOException e) {
            throw new ControllerException(e.getMessage());
        }

        Runtime rt = Runtime.getRuntime();
        long totalMemory = rt.totalMemory() / MEGA_BYTE;
        long freeMemory = rt.freeMemory() / MEGA_BYTE;
        long usedMemory = totalMemory - freeMemory;

        final long p100 = (int) Math.round(((double) freeMemory / (double) totalMemory) * 100);

        Map<String, Long> memInfo = new HashMap<String, Long>();

        memInfo.put("total", totalMemory);
        memInfo.put("used", usedMemory);
        memInfo.put("free", freeMemory);
        memInfo.put("percent_free", p100);

        systemInfo.put("memory", memInfo);
        systemInfo.put("powered_by", POWERED_BY_MICRO);

        //cpu usage in milli secs
        long currentCpuUsage = sunOperatingSystemMXBean.getProcessCpuTime() / 1000000;
        osMap.put("cpu_usage", currentCpuUsage);
        osMap.put("available_processors", sunOperatingSystemMXBean.getAvailableProcessors());
        osMap.put("system_load_average", sunOperatingSystemMXBean.getSystemLoadAverage());
        osMap.put("committed_virtual_memory_size", sunOperatingSystemMXBean.getCommittedVirtualMemorySize());
        osMap.put("free_physical_memory_size", sunOperatingSystemMXBean.getFreePhysicalMemorySize());
        osMap.put("total_physical_memory_size", sunOperatingSystemMXBean.getTotalPhysicalMemorySize());
        osMap.put("free_swap_space_size", sunOperatingSystemMXBean.getFreeSwapSpaceSize());
        osMap.put("total_swap_space_size", sunOperatingSystemMXBean.getTotalSwapSpaceSize());

        systemInfo.put("os", osMap);

        List<GarbageCollectorMXBean> gc = ManagementFactory.getGarbageCollectorMXBeans();
        List<Map> gcInfo = new ArrayList<Map>();

        for (GarbageCollectorMXBean aGc : gc) {
            Map<String, Object> gcMap = new HashMap<String, Object>();
            gcMap.put("name", aGc.getName());
            gcMap.put("collection_count", aGc.getCollectionCount());
            gcMap.put("collection_time", aGc.getCollectionTime());

            gcInfo.add(gcMap);
        }

        systemInfo.put("gc", gcInfo);

        ThreadMXBean threadMXBean = ManagementFactory.getThreadMXBean();
        Map<String, Object> threadInfoMap = new HashMap<String, Object>(); // more to come ;)
        threadInfoMap.put("peak_thread_count", threadMXBean.getPeakThreadCount());
        threadInfoMap.put("thread_count", threadMXBean.getThreadCount());
        threadInfoMap.put("total_started_thread_count", threadMXBean.getTotalStartedThreadCount());

        long[] deadlockedThreads = threadMXBean.findMonitorDeadlockedThreads();
        threadInfoMap.put("dead_locked_thread_count", deadlockedThreads != null ? deadlockedThreads.length : 0);
        systemInfo.put("thread_info", threadInfoMap);

        JSONObject sysinfoJson = new JSONObject(Collections.singletonMap("system_info", systemInfo));

        String sysinfoString = null;
        try {
            sysinfoString = context.getRequest().getParameter("pretty") != null ? sysinfoJson.toString(2)
                    : sysinfoJson.toString();
        } catch (JSONException e) {
            e.printStackTrace();
            throw new ControllerException(e.getMessage());
        }

        context.getRackResponse().withContentType(Mime.mimeType(JSON_TYPE)).withBody(sysinfoString)
                .withContentLength(sysinfoString.length()).with(Rack.MESSAGE_STATUS, HttpServletResponse.SC_OK);

        context.halt();
    }
}