From ffa8f4e170a89fdf19816127c7a00466e8413470 Mon Sep 17 00:00:00 2001 From: Mitch Dennett Date: Sat, 6 Feb 2021 16:51:18 -0500 Subject: [PATCH] refactored render method --- .../com/severell/plugins/jte/ViewJteDriver.java | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/src/main/java/com/severell/plugins/jte/ViewJteDriver.java b/src/main/java/com/severell/plugins/jte/ViewJteDriver.java index 05ca947..247ecbf 100644 --- a/src/main/java/com/severell/plugins/jte/ViewJteDriver.java +++ b/src/main/java/com/severell/plugins/jte/ViewJteDriver.java @@ -8,6 +8,7 @@ import gg.jte.TemplateOutput; import gg.jte.output.PrintWriterOutput; import gg.jte.output.StringOutput; +import gg.jte.output.WriterOutput; import gg.jte.resolve.DirectoryCodeResolver; import java.io.PrintWriter; @@ -50,15 +51,7 @@ public void render(String template, Map object, Writer writer) th @Override public void render(String template, Map object, String baseDir, Writer writer) throws ViewException { - if(writer instanceof PrintWriter) { - TemplateOutput output = new PrintWriterOutput((PrintWriter) writer); - templateEngine.render(template, object, output); - } else if(writer instanceof StringWriter) { - StringOutput output = new StringOutput(); - templateEngine.render(template, object, output); - ((StringWriter) writer).write(output.toString()); - } else { - throw new ViewException("Invalid writer. Needs to be instance of PrintWriter"); - } + TemplateOutput output = new WriterOutput(writer); + templateEngine.render(template, object, output); } }