| 1 | /******************************************************************************* |
| 2 | * Copyright (c) 2009 Ericsson |
| 3 | * |
| 4 | * All rights reserved. This program and the accompanying materials are |
| 5 | * made available under the terms of the Eclipse Public License v1.0 which |
| 6 | * accompanies this distribution, and is available at |
| 7 | * http://www.eclipse.org/legal/epl-v10.html |
| 8 | * |
| 9 | * Contributors: |
| 10 | * Alvaro Sanchez-Leon (alvsan09@gmail.com) - Initial API and implementation |
| 11 | *******************************************************************************/ |
| 12 | |
| 13 | package org.eclipse.linuxtools.lttng.tests.state; |
| 14 | |
| 15 | import java.util.HashMap; |
| 16 | import java.util.Map; |
| 17 | |
| 18 | import org.eclipse.linuxtools.lttng.model.LTTngTreeNode; |
| 19 | import org.eclipse.linuxtools.lttng.state.LttngStateException; |
| 20 | import org.eclipse.linuxtools.lttng.state.trace.IStateTraceManager; |
| 21 | import org.eclipse.linuxtools.lttng.state.trace.StateTraceManager; |
| 22 | import org.eclipse.linuxtools.tmf.trace.ITmfTrace; |
| 23 | |
| 24 | /** |
| 25 | * @author alvaro |
| 26 | * |
| 27 | */ |
| 28 | public class StateManagerFactoryTestSupport { |
| 29 | // ======================================================================== |
| 30 | // Data |
| 31 | // ======================================================================= |
| 32 | |
| 33 | private static final Map<String, IStateTraceManager> instanceBook = new HashMap<String, IStateTraceManager>(); |
| 34 | |
| 35 | // ======================================================================== |
| 36 | // Methods |
| 37 | // ======================================================================= |
| 38 | |
| 39 | /** |
| 40 | * Provide a stateManager instance per trace |
| 41 | * |
| 42 | * @return |
| 43 | */ |
| 44 | public static IStateTraceManager getManager(ITmfTrace trace) { |
| 45 | String traceUniqueId = trace.getName(); |
| 46 | |
| 47 | if (traceUniqueId == null) { |
| 48 | return null; |
| 49 | } |
| 50 | |
| 51 | if (instanceBook.containsKey(traceUniqueId)) { |
| 52 | return instanceBook.get(traceUniqueId); |
| 53 | } |
| 54 | |
| 55 | // LttngTraceState traceModel = |
| 56 | // StateModelFactory.getStateEntryInstance(); |
| 57 | IStateTraceManager manager = null; |
| 58 | |
| 59 | // catch construction problems |
| 60 | Long id = 0L; |
| 61 | LTTngTreeNode parent = null; |
| 62 | |
| 63 | try { |
| 64 | manager = new StateTraceManager(id, parent, traceUniqueId, trace); |
| 65 | } catch (LttngStateException e) { |
| 66 | e.printStackTrace(); |
| 67 | } |
| 68 | |
| 69 | instanceBook.put(traceUniqueId, manager); |
| 70 | return manager; |
| 71 | } |
| 72 | |
| 73 | /** |
| 74 | * Remove previously registered managers |
| 75 | * @param traceUniqueId |
| 76 | */ |
| 77 | public static void removeManager(String traceUniqueId) { |
| 78 | if (traceUniqueId != null && instanceBook.containsKey(traceUniqueId)) { |
| 79 | instanceBook.remove(traceUniqueId); |
| 80 | } |
| 81 | } |
| 82 | |
| 83 | } |