|
10 | 10 | import io.jooby.MessageEncoder; |
11 | 11 | import io.jooby.ModelAndView; |
12 | 12 | import io.jooby.TemplateEngine; |
13 | | -import io.jooby.internal.handler.SendAttachment; |
14 | | -import io.jooby.internal.handler.SendByteArray; |
15 | | -import io.jooby.internal.handler.SendByteBuf; |
16 | | -import io.jooby.internal.handler.SendByteBuffer; |
17 | | -import io.jooby.internal.handler.SendCharSequence; |
18 | | -import io.jooby.internal.handler.SendDirect; |
19 | | -import io.jooby.internal.handler.SendFileChannel; |
20 | | -import io.jooby.internal.handler.SendStream; |
21 | 13 | import io.netty.buffer.ByteBuf; |
22 | 14 |
|
23 | 15 | import javax.annotation.Nonnull; |
|
27 | 19 | import java.nio.channels.FileChannel; |
28 | 20 | import java.nio.charset.StandardCharsets; |
29 | 21 | import java.nio.file.Path; |
| 22 | +import java.util.ArrayList; |
30 | 23 | import java.util.Iterator; |
31 | | -import java.util.LinkedList; |
| 24 | +import java.util.List; |
32 | 25 |
|
33 | 26 | public class CompositeMessageEncoder implements MessageEncoder { |
34 | 27 |
|
35 | | - private LinkedList<MessageEncoder> list = new LinkedList<>(); |
| 28 | + private List<MessageEncoder> decoders = new ArrayList<>(2); |
36 | 29 |
|
37 | | - private MessageEncoder templateEngine; |
38 | | - |
39 | | - public CompositeMessageEncoder() { |
40 | | - list.add(MessageEncoder.TO_STRING); |
41 | | - } |
| 30 | + private List<TemplateEngine> templateEngine = new ArrayList<>(2); |
42 | 31 |
|
43 | 32 | public CompositeMessageEncoder add(MessageEncoder encoder) { |
44 | 33 | if (encoder instanceof TemplateEngine) { |
45 | | - templateEngine = computeRenderer(templateEngine, encoder); |
| 34 | + templateEngine.add((TemplateEngine) encoder); |
46 | 35 | } else { |
47 | | - list.addFirst(encoder); |
| 36 | + decoders.add(encoder); |
48 | 37 | } |
49 | 38 | return this; |
50 | 39 | } |
51 | 40 |
|
52 | | - private MessageEncoder computeRenderer(MessageEncoder it, MessageEncoder next) { |
53 | | - if (it == null) { |
54 | | - return next; |
55 | | - } else if (it instanceof CompositeMessageEncoder) { |
56 | | - ((CompositeMessageEncoder) it).list.addFirst(next); |
57 | | - return it; |
58 | | - } else { |
59 | | - CompositeMessageEncoder composite = new CompositeMessageEncoder(); |
60 | | - composite.list.addFirst(it); |
61 | | - composite.list.addFirst(next); |
62 | | - return composite; |
63 | | - } |
64 | | - } |
65 | | - |
66 | 41 | @Override public byte[] encode(@Nonnull Context ctx, @Nonnull Object value) throws Exception { |
67 | 42 | if (value instanceof ModelAndView) { |
68 | | - return templateEngine.encode(ctx, value); |
| 43 | + ModelAndView modelAndView = (ModelAndView) value; |
| 44 | + for (TemplateEngine engine : templateEngine) { |
| 45 | + if (engine.supports(modelAndView)) { |
| 46 | + return engine.encode(ctx, modelAndView); |
| 47 | + } |
| 48 | + } |
| 49 | + throw new IllegalArgumentException("No template engine for: " + modelAndView.view); |
69 | 50 | } |
70 | 51 | /** InputStream: */ |
71 | 52 | if (value instanceof InputStream) { |
@@ -109,8 +90,7 @@ private MessageEncoder computeRenderer(MessageEncoder it, MessageEncoder next) { |
109 | 90 | ctx.send((ByteBuf) value); |
110 | 91 | return null; |
111 | 92 | } |
112 | | - /** Fallback to more complex encoder: */ |
113 | | - Iterator<MessageEncoder> iterator = list.iterator(); |
| 93 | + Iterator<MessageEncoder> iterator = decoders.iterator(); |
114 | 94 | /** NOTE: looks like an infinite loop but there is a default renderer at the end of iterator. */ |
115 | 95 | byte[] result = null; |
116 | 96 | while (result == null) { |
|
0 commit comments