Skip to content

Commit 65c5570

Browse files
Josh-Matsuokaandrewazores
authored andcommitted
Adjust for core changes
1 parent b6da1bb commit 65c5570

File tree

4 files changed

+15
-15
lines changed

4 files changed

+15
-15
lines changed

src/main/java/io/cryostat/events/EventTemplates.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@
4343
import org.jboss.resteasy.reactive.RestPath;
4444
import org.jboss.resteasy.reactive.RestResponse;
4545
import org.jboss.resteasy.reactive.multipart.FileUpload;
46-
import org.jsoup.nodes.Document;
4746

4847
@Path("")
4948
public class EventTemplates {
@@ -179,24 +178,24 @@ public Response getTargetTemplate(
179178
@RestPath long id, @RestPath TemplateType templateType, @RestPath String templateName)
180179
throws Exception {
181180
Target target = Target.find("id", id).singleResult();
182-
Document doc;
181+
String xml;
183182
switch (templateType) {
184183
case TARGET:
185-
doc =
184+
xml =
186185
targetTemplateServiceFactory
187186
.create(target)
188187
.getXml(templateName, templateType)
189188
.orElseThrow();
190189
break;
191190
case CUSTOM:
192-
doc = customTemplateService.getXml(templateName, templateType).orElseThrow();
191+
xml = customTemplateService.getXml(templateName, templateType).orElseThrow();
193192
break;
194193
default:
195194
throw new BadRequestException();
196195
}
197196
return Response.status(RestResponse.Status.OK)
198197
.header(HttpHeaders.CONTENT_TYPE, HttpMimeType.JFC.mime())
199-
.entity(doc.toString())
198+
.entity(xml)
200199
.build();
201200
}
202201
}

src/main/java/io/cryostat/events/S3TemplateService.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,6 @@
6262
import org.eclipse.microprofile.config.inject.ConfigProperty;
6363
import org.jboss.logging.Logger;
6464
import org.jsoup.Jsoup;
65-
import org.jsoup.nodes.Document;
6665
import org.jsoup.parser.Parser;
6766
import software.amazon.awssdk.core.sync.RequestBody;
6867
import software.amazon.awssdk.services.s3.S3Client;
@@ -156,12 +155,12 @@ public List<Template> getTemplates() throws FlightRecorderException {
156155
}
157156

158157
@Override
159-
public Optional<Document> getXml(String templateName, TemplateType unused)
158+
public Optional<String> getXml(String templateName, TemplateType unused)
160159
throws FlightRecorderException {
161160
try (var stream = getModel(templateName)) {
162161
Document doc =
163162
Jsoup.parse(stream, StandardCharsets.UTF_8.name(), "", Parser.xmlParser());
164-
return Optional.of(doc);
163+
return Optional.of(doc.outerHtml());
165164
} catch (IOException e) {
166165
logger.error(e);
167166
return Optional.empty();

src/main/java/io/cryostat/events/TargetTemplateService.java

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@
3030

3131
import jakarta.enterprise.context.ApplicationScoped;
3232
import jakarta.inject.Inject;
33-
import org.jsoup.nodes.Document;
3433

3534
public class TargetTemplateService implements TemplateService {
3635

@@ -62,11 +61,15 @@ public List<Template> getTemplates() throws FlightRecorderException {
6261
}
6362

6463
@Override
65-
public Optional<Document> getXml(String templateName, TemplateType unused)
64+
public Optional<String> getXml(String templateName, TemplateType unused)
6665
throws FlightRecorderException {
67-
return connectionManager.executeConnectedTask(
68-
target,
69-
conn -> conn.getTemplateService().getXml(templateName, TemplateType.TARGET));
66+
Optional doc =
67+
connectionManager.executeConnectedTask(
68+
target,
69+
conn ->
70+
conn.getTemplateService()
71+
.getXml(templateName, TemplateType.TARGET));
72+
return doc.isPresent() ? Optional.of(doc.toString()) : Optional.empty();
7073
}
7174

7275
@Override

src/main/java/io/cryostat/targets/AgentJFRService.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -260,8 +260,7 @@ public IRecordingDescriptor start(
260260
null,
261261
templateService
262262
.getXml(templateName, preferredTemplateType)
263-
.orElseThrow()
264-
.outerHtml(),
263+
.orElseThrow(),
265264
duration,
266265
maxSize,
267266
maxAge);

0 commit comments

Comments
 (0)