Mercurial > hg > release > thermostat-1.0
view vm-heap-analysis/agent/src/main/java/com/redhat/thermostat/vm/heap/analysis/agent/internal/HeapDumpReceiver.java @ 1368:b1f82879e6a9
Rename c.r.t.utils.management to c.r.t.agent.utils.management
Reviewed-by: vanaltj
Review-thread: http://icedtea.classpath.org/pipermail/thermostat/2013-November/008945.html
author | Omair Majid <omajid@redhat.com> |
---|---|
date | Mon, 25 Nov 2013 19:03:29 -0500 |
parents | b7c6db90e034 |
children | a0592d702416 |
line wrap: on
line source
/* * Copyright 2012, 2013 Red Hat, Inc. * * This file is part of Thermostat. * * Thermostat is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published * by the Free Software Foundation; either version 2, or (at your * option) any later version. * * Thermostat is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with Thermostat; see the file COPYING. If not see * <http://www.gnu.org/licenses/>. * * Linking this code with other modules is making a combined work * based on this code. Thus, the terms and conditions of the GNU * General Public License cover the whole combination. * * As a special exception, the copyright holders of this code give * you permission to link this code with independent modules to * produce an executable, regardless of the license terms of these * independent modules, and to copy and distribute the resulting * executable under terms of your choice, provided that you also * meet, for each linked independent module, the terms and conditions * of the license of that module. An independent module is a module * which is not derived from or based on this code. If you modify * this code, you may extend this exception to your version of the * library, but you are not obligated to do so. If you do not wish * to do so, delete this exception statement from your version. */ package com.redhat.thermostat.vm.heap.analysis.agent.internal; import java.io.File; import java.io.IOException; import java.nio.file.Files; import java.util.logging.Level; import java.util.logging.Logger; import com.redhat.thermostat.agent.command.RequestReceiver; import com.redhat.thermostat.agent.utils.management.MXBeanConnectionPool; import com.redhat.thermostat.common.command.Request; import com.redhat.thermostat.common.command.Response; import com.redhat.thermostat.common.command.Response.ResponseType; import com.redhat.thermostat.common.utils.LoggingUtils; import com.redhat.thermostat.storage.core.WriterID; import com.redhat.thermostat.vm.heap.analysis.common.HeapDAO; import com.redhat.thermostat.vm.heap.analysis.common.HistogramLoader; import com.redhat.thermostat.vm.heap.analysis.common.ObjectHistogram; import com.redhat.thermostat.vm.heap.analysis.common.model.HeapInfo; public class HeapDumpReceiver implements RequestReceiver { private static final Logger log = LoggingUtils.getLogger(HeapDumpReceiver.class); private final HeapDAO heapDao; private final JMXHeapDumper jmxHeapDumper; private final JMapHeapDumper jmapHeapDumper; private final HistogramLoader histogramLoader; private final WriterID writerId; public HeapDumpReceiver(HeapDAO heapDao, MXBeanConnectionPool pool, WriterID writerId) { this(heapDao, new JMXHeapDumper(pool), new JMapHeapDumper(), new HistogramLoader(), writerId); } HeapDumpReceiver(HeapDAO heapDao, JMXHeapDumper jmxHeapDumper, JMapHeapDumper jmapHeapDumper, HistogramLoader histogramLoader, WriterID writerId) { this.heapDao = heapDao; this.jmxHeapDumper = jmxHeapDumper; this.jmapHeapDumper = jmapHeapDumper; this.histogramLoader = histogramLoader; this.writerId = writerId; } @Override public Response receive(Request request) { String vmId = request.getParameter("vmId"); String strPid = request.getParameter("vmPid"); try { int vmPid = Integer.parseInt(strPid); File heapDumpFile = dumpHeap(vmPid); ObjectHistogram histogram = loadHistogram(heapDumpFile.getAbsolutePath()); saveHeapDumpInfo(vmId, heapDumpFile, histogram); } catch (IOException e) { log.log(Level.SEVERE, "Unexpected IO problem while writing heap dump", e); return new Response(ResponseType.ERROR); } catch (HeapDumpException e) { log.log(Level.SEVERE, "Unexpected problem while writing heap dump", e); return new Response(ResponseType.ERROR); } catch (NumberFormatException e) { log.log(Level.WARNING, "Invalid PID: " + strPid, e); return new Response(ResponseType.ERROR); } return new Response(ResponseType.OK); } private File dumpHeap(int pid) throws IOException, HeapDumpException { File tempFile = Files.createTempFile("thermostat-", "-heapdump").toFile(); String tempFileName = tempFile.getAbsolutePath(); tempFile.delete(); // Need to delete before dumping heap, jmap does not override existing file and stop with an error. dumpHeapUsingJMX(pid, tempFileName); return tempFile; } private void dumpHeapUsingJMX(int pid, String filename) throws HeapDumpException { try { jmxHeapDumper.dumpHeap(pid, filename); } catch (HeapDumpException e) { log.log(Level.WARNING, "Heap dump using JMX failed, trying jmap", e); dumpHeapUsingJMap(pid, filename); } } private void dumpHeapUsingJMap(int pid, String filename) throws HeapDumpException { jmapHeapDumper.dumpHeap(pid, filename); } private ObjectHistogram loadHistogram(String heapDumpFilename) throws IOException { return histogramLoader.load(heapDumpFilename); } private void saveHeapDumpInfo(String vmId, File tempFile, ObjectHistogram histogram) throws IOException { String wId = writerId.getWriterID(); HeapInfo heapInfo = new HeapInfo(wId, vmId, System.currentTimeMillis()); heapDao.putHeapInfo(heapInfo, tempFile, histogram); } }