1 /*
2 * Copyright 2012 The Netty Project
3 *
4 * The Netty Project licenses this file to you under the Apache License,
5 * version 2.0 (the "License"); you may not use this file except in compliance
6 * with the License. You may obtain a copy of the License at:
7 *
8 * https://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13 * License for the specific language governing permissions and limitations
14 * under the License.
15 */
16 package io.netty5.example.factorial;
17
18 import io.netty5.buffer.api.Buffer;
19 import io.netty5.channel.ChannelHandlerContext;
20 import io.netty5.handler.codec.MessageToByteEncoder;
21
22 import java.math.BigInteger;
23
24 /**
25 * Encodes a {@link Number} into the binary representation prepended with
26 * a magic number ('F' or 0x46) and a 32-bit length prefix. For example, 42
27 * will be encoded to { 'F', 0, 0, 0, 1, 42 }.
28 */
29 public class NumberEncoder extends MessageToByteEncoder<Number> {
30
31 @Override
32 protected Buffer allocateBuffer(ChannelHandlerContext ctx, Number msg) throws Exception {
33 return ctx.bufferAllocator().allocate(16);
34 }
35
36 @Override
37 protected void encode(ChannelHandlerContext ctx, Number msg, Buffer out) {
38 // Convert to a BigInteger first for easier implementation.
39 BigInteger v;
40 if (msg instanceof BigInteger) {
41 v = (BigInteger) msg;
42 } else {
43 v = new BigInteger(String.valueOf(msg));
44 }
45
46 // Convert the number into a byte array.
47 byte[] data = v.toByteArray();
48 int dataLength = data.length;
49
50 // Write a message.
51 out.writeByte((byte) 'F'); // magic number
52 out.writeInt(dataLength); // data length
53 out.writeBytes(data); // data
54 }
55 }