Source code

001/*
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *   http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing,
013 * software distributed under the License is distributed on an
014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015 * KIND, either express or implied.  See the License for the
016 * specific language governing permissions and limitations
017 * under the License.
018 */
019package org.apache.reef.runtime.common.launch;
020
021import org.apache.reef.runtime.common.launch.parameters.LaunchID;
022import org.apache.reef.tang.annotations.Parameter;
023import org.apache.reef.wake.EventHandler;
024import org.apache.reef.wake.profiler.WakeProfiler;
025import org.apache.reef.wake.time.event.StopTime;
026
027import javax.inject.Inject;
028import java.io.FileNotFoundException;
029import java.io.PrintWriter;
030import java.io.UnsupportedEncodingException;
031import java.util.logging.Level;
032import java.util.logging.Logger;
033
034/**
035 * An EventHandler that writes out the profiler results.
036 */
037public final class ProfilingStopHandler implements EventHandler<StopTime> {
038  private static final Logger LOG = Logger.getLogger(ProfilingStopHandler.class.getName());
039  private static WakeProfiler profiler;
040  private final String launchID;
041
042  @Inject
043  public ProfilingStopHandler(@Parameter(LaunchID.class) final String launchID) {
044    this.launchID = launchID;
045  }
046
047  public static void setProfiler(final WakeProfiler profiler) {
048    ProfilingStopHandler.profiler = profiler;
049  }
050
051  @Override
052  public void onNext(final StopTime stopTime) {
053    try (final PrintWriter out = new PrintWriter("profile-" + launchID + ".json", "UTF-8")) {
054      out.print(profiler.objectGraphToString());
055    } catch (final FileNotFoundException | UnsupportedEncodingException e) {
056      LOG.log(Level.WARNING, "Unable to write the profile", e);
057    }
058  }
059}