1 /*
2 * Copyright (c) 2016, 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
26 package jdk.jfr.cmd;
27
28 import java.nio.file.Path;
29
30 import javax.script.ScriptEngine;
31 import javax.script.ScriptEngineManager;
32
33 import jdk.jfr.ValueDescriptor;
34 import jdk.jfr.consumer.RecordedEvent;
35 import jdk.jfr.consumer.RecordedObject;
36 import jdk.jfr.consumer.RecordingFile;
37 import jdk.nashorn.api.scripting.JSObject;
38 import jdk.test.lib.Asserts;
39 import jdk.test.lib.process.OutputAnalyzer;
40
41 /*
42 * @test
43 * @key jfr
44 * @summary Tests print --json
45 *
46 * @library /test/lib /test/jdk
47 * @modules jdk.scripting.nashorn
48 * jdk.jfr
49 *
50 * @run main/othervm jdk.jfr.cmd.TestPrintJSON
51 */
52 public class TestPrintJSON {
53
54 public static void main(String... args) throws Exception {
55
56 Path recordingFile = ExecuteHelper.createProfilingRecording().toAbsolutePath();
57
58 OutputAnalyzer output = ExecuteHelper.run("print", "--json", recordingFile.toString());
59 String json = output.getStdout();
60
61 // Parse JSON using Nashorn
62 String statement = "var jsonObject = " + json;
63 ScriptEngineManager factory = new ScriptEngineManager();
64 ScriptEngine engine = factory.getEngineByName("nashorn");
65 engine.eval(statement);
66 JSObject o = (JSObject) engine.get("jsonObject");
67 JSObject recording = (JSObject) o.getMember("recording");
68 JSObject events = (JSObject) recording.getMember("events");
69
70 // Verify events are equal
71 try (RecordingFile rf = new RecordingFile(recordingFile)) {
72 for (Object jsonEvent : events.values()) {
73 RecordedEvent recordedEvent = rf.readEvent();
74 double typeId = recordedEvent.getEventType().getId();
75 String startTime = recordedEvent.getStartTime().toString();
76 String duration = recordedEvent.getDuration().toString();
77 Asserts.assertEquals(typeId, ((Number) ((JSObject) jsonEvent).getMember("typeId")).doubleValue());
78 Asserts.assertEquals(startTime, ((JSObject) jsonEvent).getMember("startTime"));
79 Asserts.assertEquals(duration, ((JSObject) jsonEvent).getMember("duration"));
80 assertEquals(jsonEvent, recordedEvent);
81 }
82 Asserts.assertFalse(rf.hasMoreEvents(), "Incorrect number of events");
83 }
84 }
85
86 private static void assertEquals(Object jsonObject, Object jfrObject) throws Exception {
87 // Check object
88 if (jfrObject instanceof RecordedObject) {
89 JSObject values = (JSObject) ((JSObject) jsonObject).getMember("values");
90 RecordedObject recObject = (RecordedObject) jfrObject;
91 Asserts.assertEquals(values.values().size(), recObject.getFields().size());
92 for (ValueDescriptor v : recObject.getFields()) {
93 String name = v.getName();
94 assertEquals(values.getMember(name), recObject.getValue(name));
95 return;
96 }
97 }
98 // Check array
99 if (jfrObject != null && jfrObject.getClass().isArray()) {
100 Object[] jfrArray = (Object[]) jfrObject;
101 JSObject jsArray = (JSObject) jsonObject;
102 for (int i = 0; i < jfrArray.length; i++) {
103 assertEquals(jsArray.getSlot(i), jfrArray[i]);
104 }
105 return;
106 }
107 String jsonText = String.valueOf(jsonObject);
108 // Double.NaN / Double.Inifinity is not supported by JSON format,
109 // use null
110 if (jfrObject instanceof Double) {
111 double expected = ((Double) jfrObject);
112 if (Double.isInfinite(expected) || Double.isNaN(expected)) {
113 Asserts.assertEquals("null", jsonText);
114 return;
115 }
116 double value = Double.parseDouble(jsonText);
117 Asserts.assertEquals(expected, value);
118 return;
119 }
120 // Float.NaN / Float.Inifinity is not supported by JSON format,
121 // use null
122 if (jfrObject instanceof Float) {
123 float expected = ((Float) jfrObject);
124 if (Float.isInfinite(expected) || Float.isNaN(expected)) {
125 Asserts.assertEquals("null", jsonText);
126 return;
127 }
128 float value = Float.parseFloat(jsonText);
129 Asserts.assertEquals(expected, value);
130 return;
131 }
132 if (jfrObject instanceof Integer) {
133 Integer expected = ((Integer) jfrObject);
134 double value = Double.parseDouble(jsonText);
135 Asserts.assertEquals(expected.doubleValue(), value);
136 return;
137 }
138 if (jfrObject instanceof Long) {
139 Long expected = ((Long) jfrObject);
140 double value = Double.parseDouble(jsonText);
141 Asserts.assertEquals(expected.doubleValue(), value);
142 return;
143 }
144
145 String jfrText = String.valueOf(jfrObject);
146 Asserts.assertEquals(jfrText, jsonText, "Primitive values don't match. JSON = " + jsonText);
147 }
148 }