|
11 | 11 | import org.java_websocket.exceptions.InvalidDataException; |
12 | 12 | import org.java_websocket.exceptions.InvalidHandshakeException; |
13 | 13 | import org.java_websocket.exceptions.LimitExedeedException; |
14 | | -import org.java_websocket.framing.CloseFrame; |
15 | | -import org.java_websocket.framing.FrameBuilder; |
16 | | -import org.java_websocket.framing.Framedata; |
| 14 | +import org.java_websocket.framing.*; |
17 | 15 | import org.java_websocket.framing.Framedata.Opcode; |
18 | | -import org.java_websocket.framing.FramedataImpl1; |
19 | 16 | import org.java_websocket.handshake.ClientHandshake; |
20 | 17 | import org.java_websocket.handshake.ClientHandshakeBuilder; |
21 | 18 | import org.java_websocket.handshake.HandshakeBuilder; |
@@ -138,20 +135,24 @@ public List<Framedata> continuousFrame( Opcode op, ByteBuffer buffer, boolean fi |
138 | 135 | if(op != Opcode.BINARY && op != Opcode.TEXT) { |
139 | 136 | throw new IllegalArgumentException( "Only Opcode.BINARY or Opcode.TEXT are allowed" ); |
140 | 137 | } |
141 | | - |
| 138 | + DataFrame bui = null; |
142 | 139 | if( continuousFrameType != null ) { |
143 | | - continuousFrameType = Opcode.CONTINUOUS; |
| 140 | + bui = new ContinuousFrame(); |
144 | 141 | } else { |
145 | 142 | continuousFrameType = op; |
| 143 | + if (op == Opcode.BINARY) { |
| 144 | + bui = new BinaryFrame(); |
| 145 | + } else if (op == Opcode.TEXT) { |
| 146 | + bui = new TextFrame(); |
| 147 | + } |
146 | 148 | } |
147 | | - |
148 | | - FrameBuilder bui = new FramedataImpl1( continuousFrameType ); |
| 149 | + bui.setPayload( buffer ); |
| 150 | + bui.setFin( fin ); |
149 | 151 | try { |
150 | | - bui.setPayload( buffer ); |
| 152 | + bui.isValid(); |
151 | 153 | } catch ( InvalidDataException e ) { |
152 | 154 | throw new RuntimeException( e ); // can only happen when one builds close frames(Opcode.Close) |
153 | 155 | } |
154 | | - bui.setFin( fin ); |
155 | 156 | if( fin ) { |
156 | 157 | continuousFrameType = null; |
157 | 158 | } else { |
|
0 commit comments