Mercurial > hg > release > thermostat-1.6
view vm-profiler/client-swing/src/main/java/com/redhat/thermostat/vm/profiler/client/swing/internal/VmProfileController.java @ 2049:a92d602216ad
Update copyright license headers for 2017
PR3290
Reviewed-by: jerboaa
Review-thread: http://icedtea.classpath.org/pipermail/thermostat/2017-January/021974.html
author | Andrew Azores <aazores@redhat.com> |
---|---|
date | Tue, 17 Jan 2017 12:19:56 -0500 |
parents | 7e6a52dfd303 |
children |
line wrap: on
line source
/* * Copyright 2012-2017 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.profiler.client.swing.internal; import static com.redhat.thermostat.vm.profiler.client.swing.internal.VmProfileView.TabbedPaneAction.TABLE_TAB_SELECTED; import java.io.InputStream; import java.net.InetSocketAddress; import java.util.ArrayList; import java.util.Arrays; import java.util.Collections; import java.util.Comparator; import java.util.HashMap; import java.util.List; import java.util.Map; import java.util.Objects; import java.util.concurrent.TimeUnit; import com.redhat.thermostat.client.command.RequestQueue; import com.redhat.thermostat.client.core.controllers.InformationServiceController; import com.redhat.thermostat.client.core.progress.ProgressHandle; import com.redhat.thermostat.client.core.progress.ProgressNotifier; import com.redhat.thermostat.client.core.views.BasicView; import com.redhat.thermostat.client.core.views.BasicView.Action; import com.redhat.thermostat.client.core.views.UIComponent; import com.redhat.thermostat.client.swing.SwingComponent; import com.redhat.thermostat.common.ActionEvent; import com.redhat.thermostat.common.ActionListener; import com.redhat.thermostat.common.ApplicationService; import com.redhat.thermostat.common.Clock; import com.redhat.thermostat.common.SystemClock; import com.redhat.thermostat.common.Timer; import com.redhat.thermostat.common.Timer.SchedulingType; import com.redhat.thermostat.common.command.Request; import com.redhat.thermostat.common.command.RequestResponseListener; import com.redhat.thermostat.common.command.Response; import com.redhat.thermostat.common.model.Range; import com.redhat.thermostat.shared.locale.LocalizedString; import com.redhat.thermostat.shared.locale.Translate; import com.redhat.thermostat.storage.core.HostRef; import com.redhat.thermostat.storage.core.VmRef; import com.redhat.thermostat.storage.dao.AgentInfoDAO; import com.redhat.thermostat.storage.dao.VmInfoDAO; import com.redhat.thermostat.storage.model.AgentInformation; import com.redhat.thermostat.storage.model.VmInfo.AliveStatus; import com.redhat.thermostat.vm.profiler.client.core.ProfilingResult; import com.redhat.thermostat.vm.profiler.client.core.ProfilingResultParser; import com.redhat.thermostat.vm.profiler.client.swing.internal.VmProfileView.TabbedPaneAction; import com.redhat.thermostat.vm.profiler.client.swing.internal.VmProfileView.Profile; import com.redhat.thermostat.vm.profiler.client.swing.internal.VmProfileView.ProfileAction; import com.redhat.thermostat.vm.profiler.client.swing.internal.VmProfileView.ProfilingState; import com.redhat.thermostat.vm.profiler.common.ProfileDAO; import com.redhat.thermostat.vm.profiler.common.ProfileInfo; import com.redhat.thermostat.vm.profiler.common.ProfileRequest; import com.redhat.thermostat.vm.profiler.common.ProfileStatusChange; public class VmProfileController implements InformationServiceController<VmRef> { static final String STATE_MAP_KEY = "vmProfileControllerRestoreBundle"; private static final Translate<LocaleResources> translator = LocaleResources.createLocalizer(); private static final Map<VmRef, SaveState> STATE_BUNDLE_MAP = new HashMap<>(); private final ApplicationService service; private final ProgressNotifier notifier; private final ProfileDAO profileDao; private final AgentInfoDAO agentInfoDao; private final VmInfoDAO vmInfoDao; private final RequestQueue queue; private final VmRef vm; private VmProfileView view; private VmProfileTreeMapView treeMapView; private TabbedPaneAction lastTabSelection; private ProfilingResult selectedResult; private Timer updater; private Clock clock; private boolean profilingStartOrStopRequested = false; private ProfileStatusChange previousStatus; private ProfilingState profilingState = ProfilingState.STOPPED; private ProgressHandle progressDisplay; public VmProfileController(ApplicationService service, ProgressNotifier notifier, AgentInfoDAO agentInfoDao, VmInfoDAO vmInfoDao, ProfileDAO dao, RequestQueue queue, VmProfileTreeMapViewProvider treeMapViewProvider, VmRef vm) { this(service, notifier, agentInfoDao, vmInfoDao, dao, queue, new SystemClock(), new SwingVmProfileView(), vm, treeMapViewProvider); } VmProfileController(final ApplicationService service, ProgressNotifier notifier, AgentInfoDAO agentInfoDao, VmInfoDAO vmInfoDao, ProfileDAO dao, RequestQueue queue, Clock clock, final VmProfileView view, VmRef vm, VmProfileTreeMapViewProvider treeMapViewProvider) { this.service = service; this.notifier = notifier; this.agentInfoDao = agentInfoDao; this.vmInfoDao = vmInfoDao; this.profileDao = dao; this.queue = queue; this.clock = clock; this.view = view; this.vm = vm; if (service.getApplicationCache().getAttribute(STATE_MAP_KEY) == null) { service.getApplicationCache().addAttribute(STATE_MAP_KEY, STATE_BUNDLE_MAP); } this.treeMapView = Objects.requireNonNull(treeMapViewProvider.createView()); view.addTabToTabbedPane(translator.localize(LocaleResources.PROFILER_RESULTS_TREEMAP), ((SwingComponent) this.treeMapView).getUiComponent()); this.lastTabSelection = TABLE_TAB_SELECTED; // TODO dispose the timer when done updater = service.getTimerFactory().createTimer(); updater.setSchedulingType(SchedulingType.FIXED_DELAY); updater.setInitialDelay(0); updater.setDelay(5); updater.setTimeUnit(TimeUnit.SECONDS); updater.setAction(new Runnable() { @Override public void run() { updateViewWithCurrentProfilingStatus(); updateViewWithProfiledRuns(); } }); view.addActionListener(new ActionListener<BasicView.Action>() { @Override public void actionPerformed(ActionEvent<Action> actionEvent) { switch (actionEvent.getActionId()) { case HIDDEN: saveState(); updater.stop(); break; case VISIBLE: restoreState(); view.setViewControlsEnabled(isAlive()); updater.start(); break; default: throw new AssertionError("Unknown action event: " + actionEvent); } } }); view.addProfileActionListener(new ActionListener<VmProfileView.ProfileAction>() { @Override public void actionPerformed(ActionEvent<ProfileAction> actionEvent) { ProfileAction id = actionEvent.getActionId(); switch (id) { case START_PROFILING: startProfiling(); break; case STOP_PROFILING: stopProfiling(); break; case PROFILE_SELECTED: loadSelectedProfileRunData(); updateViewWithSelectedProfileRunData(); break; default: throw new AssertionError("Unknown event: " + id); } } }); view.setTabbedPaneActionListener(new ActionListener<TabbedPaneAction>() { @Override public void actionPerformed(ActionEvent<TabbedPaneAction> actionEvent) { TabbedPaneAction id = actionEvent.getActionId(); boolean validTabSelection = Arrays.asList(VmProfileView.TabbedPaneAction.values()).contains(id); if (validTabSelection && selectedResult != null) { lastTabSelection = id; updateViewWithSelectedProfileRunData(); } else if (validTabSelection) { lastTabSelection = id; } else { throw new AssertionError("Unknown event: " + id); } } }); view.setViewControlsEnabled(isAlive()); } private void saveState() { SaveState bundle = new SaveState(previousStatus, profilingStartOrStopRequested, profilingState); Map<VmRef, SaveState> map = ((Map<VmRef, SaveState>) service.getApplicationCache().getAttribute(STATE_MAP_KEY)); map.put(vm, bundle); service.getApplicationCache().addAttribute(STATE_MAP_KEY, map); } private void restoreState() { SaveState bundle = ((Map<VmRef, SaveState>) service.getApplicationCache().getAttribute(STATE_MAP_KEY)).get(vm); profilingStartOrStopRequested = bundle != null && bundle.isProfilingStartOrStopRequested(); previousStatus = bundle == null ? null : bundle.getProfileStatusChange(); profilingState = bundle == null ? ProfilingState.STOPPED : bundle.getProfilingState(); if (previousStatus != null) { profilingState = getProfilingState(previousStatus, profilingStartOrStopRequested); } view.setProfilingState(profilingState); } private void startProfiling() { profilingState = ProfilingState.STARTING; view.setProfilingState(profilingState); setProgressNotificationAndSendRequest(true); } private void stopProfiling() { profilingState = ProfilingState.STOPPING; view.setProfilingState(profilingState); setProgressNotificationAndSendRequest(false); } private void setProgressNotificationAndSendRequest(boolean start) { showProgressNotification(start); sendProfilingRequest(start); } private void sendProfilingRequest(final boolean start) { InetSocketAddress address = agentInfoDao.getAgentInformation(vm.getHostRef()).getRequestQueueAddress(); String action = start ? ProfileRequest.START_PROFILING : ProfileRequest.STOP_PROFILING; Request req = ProfileRequest.create(address, vm.getVmId(), action); req.addListener(new RequestResponseListener() { @Override public void fireComplete(Request request, Response response) { switch (response.getType()) { case OK: updateViewWithCurrentProfilingStatus(); break; default: // FIXME show message to user hideProgressNotificationIfVisible(); profilingStartOrStopRequested = false; profilingState = ProfilingState.DISABLED; view.setProfilingState(profilingState); view.setViewControlsEnabled(false); break; } } }); queue.putRequest(req); profilingStartOrStopRequested = true; } private void updateViewWithCurrentProfilingStatus() { ProfileStatusChange currentStatus = profileDao.getLatestStatus(vm); if (currentStatus != null) { profilingState = getProfilingState(currentStatus, profilingStartOrStopRequested); } if (!isAlive()) { view.setProfilingState(ProfilingState.DISABLED); } else if (profilingStartOrStopRequested) { boolean statusChanged = (previousStatus == null && currentStatus != null) || (currentStatus != null && !(currentStatus.equals(previousStatus))); if (statusChanged) { view.setProfilingState(profilingState); profilingStartOrStopRequested = false; hideProgressNotificationIfVisible(); } } else { view.setProfilingState(profilingState); } previousStatus = currentStatus; } private ProfilingState getProfilingState(ProfileStatusChange profileStatusChange, boolean profilingStartOrStopRequested) { ProfilingState profilingState; boolean currentlyActive = profileStatusChange.isStarted(); if (currentlyActive && profilingStartOrStopRequested) { profilingState = ProfilingState.STARTING; } else if (currentlyActive) { profilingState = ProfilingState.STARTED; } else if (profilingStartOrStopRequested) { profilingState = ProfilingState.STOPPING; } else { profilingState = ProfilingState.STOPPED; } return profilingState; } private void showProgressNotification(boolean start) { if (start) { progressDisplay = new ProgressHandle(translator.localize(LocaleResources.STARTING_PROFILING)); } else { progressDisplay = new ProgressHandle(translator.localize(LocaleResources.STOPPING_PROFILING)); } progressDisplay.setIndeterminate(true); notifier.register(progressDisplay); progressDisplay.start(); } private void hideProgressNotificationIfVisible() { if (progressDisplay != null) { progressDisplay.stop(); progressDisplay = null; } } private boolean isAlive() { HostRef agent = vm.getHostRef(); AgentInformation agentInfo = agentInfoDao.getAgentInformation(agent); if (!agentInfo.isAlive()) { return false; } return vmInfoDao.getVmInfo(vm).isAlive(agentInfo) == AliveStatus.RUNNING; } private void updateViewWithProfiledRuns() { long end = clock.getRealTimeMillis(); long start = end - TimeUnit.DAYS.toMillis(1); // FIXME hardcoded 1 day List<ProfileInfo> profileInfos = profileDao.getAllProfileInfo(vm, new Range<>(start, end)); List<Profile> profiles = new ArrayList<>(); for (ProfileInfo profileInfo : profileInfos) { Profile profile = new Profile(profileInfo.getProfileId(), profileInfo.getStartTimeStamp(), profileInfo.getStopTimeStamp()); profiles.add(profile); } Collections.sort(profiles, new ByTimeStamp()); view.setAvailableProfilingRuns(profiles); } private void loadSelectedProfileRunData() { String profileId = view.getSelectedProfile().name; InputStream in = profileDao.loadProfileDataById(vm, profileId); selectedResult = new ProfilingResultParser().parse(in); } private void updateViewWithSelectedProfileRunData() { switch (lastTabSelection) { case TABLE_TAB_SELECTED: view.setProfilingDetailData(selectedResult); break; case TREEMAP_TAB_SELECTED: treeMapView.display(selectedResult); break; default: throw new AssertionError("Unknown selection: " + lastTabSelection); } } @Override public UIComponent getView() { return view; } @Override public LocalizedString getLocalizedName() { return translator.localize(LocaleResources.PROFILER_TAB_NAME); } static class ByTimeStamp implements Comparator<Profile> { @Override public int compare(Profile o1, Profile o2) { return Long.compare(o1.startTimeStamp, o2.startTimeStamp); } } static class SaveState { private final ProfileStatusChange profileStatusChange; private final boolean profilingStartOrStopRequested; private final ProfilingState profilingState; public SaveState(ProfileStatusChange profileStatusChange, boolean profilingStartOrStopRequested, ProfilingState profilingState) { this.profileStatusChange = profileStatusChange; this.profilingStartOrStopRequested = profilingStartOrStopRequested; this.profilingState = profilingState; } public ProfileStatusChange getProfileStatusChange() { return profileStatusChange; } public boolean isProfilingStartOrStopRequested() { return profilingStartOrStopRequested; } public ProfilingState getProfilingState() { return profilingState; } } }