author | dcubed |
Thu, 28 Jun 2018 10:46:05 -0400 | |
changeset 50886 | 1b346a59b5da |
parent 50881 | a21cad3fa448 |
child 50887 | fa91003ec489 |
permissions | -rw-r--r-- |
50113 | 1 |
/* |
2 |
* Copyright (c) 2012, 2018, Oracle and/or its affiliates. All rights reserved. |
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
|
4 |
* |
|
5 |
* This code is free software; you can redistribute it and/or modify it |
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
|
7 |
* published by the Free Software Foundation. Oracle designates this |
|
8 |
* particular file as subject to the "Classpath" exception as provided |
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
|
10 |
* |
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
|
15 |
* accompanied this code). |
|
16 |
* |
|
17 |
* You should have received a copy of the GNU General Public License version |
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
|
20 |
* |
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
|
22 |
* or visit www.oracle.com if you need additional information or have any |
|
23 |
* questions. |
|
24 |
*/ |
|
25 |
package jdk.jfr.internal.dcmd; |
|
26 |
||
27 |
import java.io.IOException; |
|
28 |
import java.io.PrintWriter; |
|
29 |
import java.io.StringWriter; |
|
50745 | 30 |
import java.nio.file.Files; |
50113 | 31 |
import java.nio.file.InvalidPathException; |
32 |
import java.nio.file.Path; |
|
50745 | 33 |
import java.nio.file.Paths; |
50113 | 34 |
import java.time.Duration; |
35 |
import java.util.ArrayList; |
|
36 |
import java.util.Collections; |
|
50226
408021edf22f
8195819: Remove recording=x from jcmd JFR.check output
egahlin
parents:
50113
diff
changeset
|
37 |
import java.util.Comparator; |
50113 | 38 |
import java.util.List; |
39 |
||
40 |
import jdk.jfr.FlightRecorder; |
|
41 |
import jdk.jfr.Recording; |
|
42 |
import jdk.jfr.internal.SecuritySupport; |
|
43 |
import jdk.jfr.internal.SecuritySupport.SafePath; |
|
44 |
import jdk.jfr.internal.Utils; |
|
45 |
||
46 |
/** |
|
47 |
* Base class for JFR diagnostic commands |
|
48 |
* |
|
49 |
*/ |
|
50 |
abstract class AbstractDCmd { |
|
51 |
||
52 |
private final StringWriter result; |
|
53 |
private final PrintWriter log; |
|
54 |
||
55 |
protected AbstractDCmd() { |
|
56 |
result = new StringWriter(); |
|
57 |
log = new PrintWriter(result); |
|
58 |
} |
|
59 |
||
60 |
protected final FlightRecorder getFlightRecorder() { |
|
61 |
return FlightRecorder.getFlightRecorder(); |
|
62 |
} |
|
63 |
||
64 |
public final String getResult() { |
|
65 |
return result.toString(); |
|
66 |
} |
|
67 |
||
50745 | 68 |
protected final SafePath resolvePath(Recording recording, String filename) throws InvalidPathException { |
69 |
if (filename == null) { |
|
50886
1b346a59b5da
8206004: [BACKOUT] errant changeset for JDK-8137164
dcubed
parents:
50881
diff
changeset
|
70 |
return makeGenerated(recording, Paths.get(".")); |
50113 | 71 |
} |
50745 | 72 |
Path path = Paths.get(filename); |
73 |
if (Files.isDirectory(path)) { |
|
74 |
return makeGenerated(recording, path); |
|
50113 | 75 |
} |
50745 | 76 |
return new SafePath(path.toAbsolutePath().normalize()); |
77 |
} |
|
78 |
||
79 |
private SafePath makeGenerated(Recording recording, Path directory) { |
|
80 |
return new SafePath(directory.toAbsolutePath().resolve(Utils.makeFilename(recording)).normalize()); |
|
50113 | 81 |
} |
82 |
||
83 |
protected final Recording findRecording(String name) throws DCmdException { |
|
84 |
try { |
|
85 |
return findRecordingById(Integer.parseInt(name)); |
|
86 |
} catch (NumberFormatException nfe) { |
|
87 |
// User specified a name, not an id. |
|
88 |
return findRecordingByName(name); |
|
89 |
} |
|
90 |
} |
|
91 |
||
50745 | 92 |
protected final void reportOperationComplete(String actionPrefix, String name, SafePath file) { |
50113 | 93 |
print(actionPrefix); |
50745 | 94 |
print(" recording"); |
95 |
if (name != null) { |
|
96 |
print(" \"" + name + "\""); |
|
97 |
} |
|
50113 | 98 |
if (file != null) { |
99 |
print(","); |
|
100 |
try { |
|
101 |
print(" "); |
|
102 |
long bytes = SecuritySupport.getFileSize(file); |
|
103 |
printBytes(bytes, " "); |
|
104 |
} catch (IOException e) { |
|
105 |
// Ignore, not essential |
|
106 |
} |
|
107 |
println(" written to:"); |
|
108 |
println(); |
|
109 |
printPath(file); |
|
110 |
} else { |
|
111 |
println("."); |
|
112 |
} |
|
113 |
} |
|
114 |
||
115 |
protected final List<Recording> getRecordings() { |
|
116 |
List<Recording> list = new ArrayList<>(getFlightRecorder().getRecordings()); |
|
50226
408021edf22f
8195819: Remove recording=x from jcmd JFR.check output
egahlin
parents:
50113
diff
changeset
|
117 |
Collections.sort(list, Comparator.comparing(Recording::getId)); |
50113 | 118 |
return list; |
119 |
} |
|
120 |
||
121 |
static String quoteIfNeeded(String text) { |
|
122 |
if (text.contains(" ")) { |
|
123 |
return "\\\"" + text + "\\\""; |
|
124 |
} else { |
|
125 |
return text; |
|
126 |
} |
|
127 |
} |
|
128 |
||
129 |
protected final void println() { |
|
130 |
log.println(); |
|
131 |
} |
|
132 |
||
133 |
protected final void print(String s) { |
|
134 |
log.print(s); |
|
135 |
} |
|
136 |
||
137 |
protected final void print(String s, Object... args) { |
|
138 |
log.printf(s, args); |
|
139 |
} |
|
140 |
||
141 |
protected final void println(String s, Object... args) { |
|
142 |
print(s, args); |
|
143 |
println(); |
|
144 |
} |
|
145 |
||
146 |
protected final void printBytes(long bytes, String separation) { |
|
50745 | 147 |
print(Utils.formatBytes(bytes, separation)); |
50113 | 148 |
} |
149 |
||
150 |
protected final void printTimespan(Duration timespan, String separator) { |
|
151 |
print(Utils.formatTimespan(timespan, separator)); |
|
152 |
} |
|
153 |
||
154 |
protected final void printPath(SafePath path) { |
|
155 |
if (path == null) { |
|
156 |
print("N/A"); |
|
157 |
return; |
|
158 |
} |
|
159 |
try { |
|
160 |
printPath(SecuritySupport.getAbsolutePath(path).toPath()); |
|
161 |
} catch (IOException ioe) { |
|
162 |
printPath(path.toPath()); |
|
163 |
} |
|
164 |
} |
|
165 |
||
166 |
protected final void printPath(Path path) { |
|
167 |
try { |
|
168 |
println(path.toAbsolutePath().toString()); |
|
169 |
} catch (SecurityException e) { |
|
170 |
// fall back on filename |
|
171 |
println(path.toString()); |
|
172 |
} |
|
173 |
} |
|
174 |
||
175 |
private Recording findRecordingById(int id) throws DCmdException { |
|
176 |
for (Recording r : getFlightRecorder().getRecordings()) { |
|
177 |
if (r.getId() == id) { |
|
178 |
return r; |
|
179 |
} |
|
180 |
} |
|
181 |
throw new DCmdException("Could not find %d.\n\nUse JFR.check without options to see list of all available recordings.", id); |
|
182 |
} |
|
183 |
||
184 |
private Recording findRecordingByName(String name) throws DCmdException { |
|
185 |
for (Recording recording : getFlightRecorder().getRecordings()) { |
|
186 |
if (name.equals(recording.getName())) { |
|
187 |
return recording; |
|
188 |
} |
|
189 |
} |
|
190 |
throw new DCmdException("Could not find %s.\n\nUse JFR.check without options to see list of all available recordings.", name); |
|
191 |
} |
|
192 |
} |