1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.websocket.jsr356.messages;
20
21 import java.io.IOException;
22 import java.nio.ByteBuffer;
23
24 import javax.websocket.MessageHandler;
25 import javax.websocket.MessageHandler.Partial;
26
27 import org.eclipse.jetty.util.BufferUtil;
28 import org.eclipse.jetty.websocket.common.message.MessageAppender;
29 import org.eclipse.jetty.websocket.jsr356.MessageHandlerWrapper;
30
31
32
33
34 public class BinaryPartialMessage implements MessageAppender
35 {
36 private final MessageHandlerWrapper msgWrapper;
37 private final MessageHandler.Partial<Object> partialHandler;
38
39 @SuppressWarnings("unchecked")
40 public BinaryPartialMessage(MessageHandlerWrapper wrapper)
41 {
42 this.msgWrapper = wrapper;
43 this.partialHandler = (Partial<Object>)wrapper.getHandler();
44 }
45
46 @Override
47 public void appendFrame(ByteBuffer payload, boolean isLast) throws IOException
48 {
49
50
51
52 if (msgWrapper.isMessageType(ByteBuffer.class))
53 {
54 partialHandler.onMessage(payload.slice(),isLast);
55 return;
56 }
57
58
59 if (msgWrapper.isMessageType(byte[].class))
60 {
61 partialHandler.onMessage(BufferUtil.toArray(payload),isLast);
62 return;
63 }
64
65 StringBuilder err = new StringBuilder();
66 err.append(msgWrapper.getHandler().getClass());
67 err.append(" does not implement an expected ");
68 err.append(MessageHandler.Partial.class.getName());
69 err.append(" of type ");
70 err.append(ByteBuffer.class.getName());
71 err.append(" or byte[]");
72 throw new IllegalStateException(err.toString());
73 }
74
75 @Override
76 public void messageComplete()
77 {
78
79 }
80 }