jp.co.acroquest.endosnipe.web.explorer.controller.ProfileController.java Source code

Introduction

Here is the source code for jp.co.acroquest.endosnipe.web.explorer.controller.ProfileController.java

Source

/*
 * Copyright (c) 2004-2014 Acroquest Technology Co., Ltd. All Rights Reserved.
 * Please read the associated COPYRIGHTS file for more details.
 *
 * THE  SOFTWARE IS  PROVIDED BY  Acroquest Technology Co., Ltd., WITHOUT  WARRANTY  OF
 * ANY KIND,  EXPRESS  OR IMPLIED,  INCLUDING BUT  NOT LIMITED  TO THE
 * WARRANTIES OF  MERCHANTABILITY,  FITNESS FOR A  PARTICULAR  PURPOSE
 * AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDER BE LIABLE FOR ANY
 * CLAIM, DAMAGES SUFFERED BY LICENSEE AS A RESULT OF USING, MODIFYING
 * OR DISTRIBUTING THIS SOFTWARE OR ITS DERIVATIVES.
 */
package jp.co.acroquest.endosnipe.web.explorer.controller;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;

import jp.co.acroquest.endosnipe.web.explorer.dto.MethodModelDto;
import jp.co.acroquest.endosnipe.web.explorer.entity.ClassModel;
import jp.co.acroquest.endosnipe.web.explorer.entity.InvocationInfo;
import jp.co.acroquest.endosnipe.web.explorer.manager.ProfilerManager;
import jp.co.acroquest.endosnipe.web.explorer.service.ProfilerService;
import net.arnx.jsonic.JSON;

/**
 * ?
 *
 * @author hiramatsu
 */
@Controller
@RequestMapping("/profiler")
public class ProfileController {
    /**  */
    @Autowired
    protected ProfilerService profilerService_;

    /**
     * 
     */
    public ProfileController() {

    }

    /**
     * ????
     *
     * @param agentName ????
     */
    @RequestMapping(value = "/reload", method = RequestMethod.POST)
    @ResponseBody
    public void reload(@RequestParam(value = "agentName") final String agentName) {
        profilerService_.getStatus(agentName);
    }

    /**
     * ?
     *
     * @param agentName ??
     */
    @RequestMapping(value = "/reset", method = RequestMethod.POST)
    @ResponseBody
    public void reset(@RequestParam(value = "agentName") final String agentName) {
        profilerService_.resetProfile(agentName);
    }

    /**
     * ???
     *
     * @param agentName ??
     * @param invocations ?
     */
    @RequestMapping(value = "/update", method = RequestMethod.POST)
    @ResponseBody
    public void update(@RequestParam(value = "agentName") final String agentName,
            @RequestParam(value = "invocations") final String invocations) {
        MethodModelDto[] invocationArray = JSON.decode(invocations, MethodModelDto[].class);
        List<InvocationInfo> invocationList = new ArrayList<InvocationInfo>();
        for (MethodModelDto methodModel : invocationArray) {
            InvocationInfo info = new InvocationInfo(methodModel.getClassName(), methodModel.getMethodName(),
                    Boolean.valueOf(methodModel.getTarget()), Boolean.valueOf(methodModel.getTransactionGraph()),
                    methodModel.getAlarmThreshold(), methodModel.getAlarmCpuThreshold());
            invocationList.add(info);
        }
        profilerService_.updateTarget(agentName, invocationList);
    }

    /**
     * ?
     * @param response {@link HttpServletResponse}
     * @param agentName ??
     */
    @RequestMapping(value = "/download", method = RequestMethod.POST)
    public void download(final HttpServletResponse response,
            @RequestParam(value = "agentName") final String agentName) {
        ProfilerManager manager = ProfilerManager.getInstance();
        ClassModel[] classModels = manager.getProfilerData(agentName);

        String result = profilerService_.createCsvData(classModels);

        //?
        response.setContentType("application/octet-stream");
        response.setCharacterEncoding("Windows-31j");
        response.setHeader("Content-Disposition", "filename=\"profile.csv\"");

        try {
            IOUtils.write(result, response.getOutputStream());
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

    /**
     * ??
     * @param response {@link HttpServletResponse}
     * @param agentName ??
     */
    @RequestMapping(value = "/getProfile", method = RequestMethod.POST)
    @ResponseBody
    public ClassModel[] getProfile(final HttpServletResponse response,
            @RequestParam(value = "agentName") final String agentName) {
        ProfilerManager manager = ProfilerManager.getInstance();
        return manager.getProfilerData(agentName);
    }
}