1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package io.netty.handler.codec.dns;
17
18 import io.netty.buffer.ByteBuf;
19 import io.netty.channel.ChannelHandler;
20 import io.netty.channel.ChannelHandlerContext;
21 import io.netty.handler.codec.MessageToMessageEncoder;
22 import io.netty.util.internal.ObjectUtil;
23 import io.netty.util.internal.UnstableApi;
24
25 import java.util.List;
26
27 @UnstableApi
28 @ChannelHandler.Sharable
29 public final class TcpDnsResponseEncoder extends MessageToMessageEncoder<DnsResponse> {
30 private final DnsRecordEncoder encoder;
31
32
33
34
35 public TcpDnsResponseEncoder() {
36 this(DnsRecordEncoder.DEFAULT);
37 }
38
39
40
41
42 public TcpDnsResponseEncoder(DnsRecordEncoder encoder) {
43 this.encoder = ObjectUtil.checkNotNull(encoder, "encoder");
44 }
45
46 @Override
47 protected void encode(ChannelHandlerContext ctx, DnsResponse response, List<Object> out) throws Exception {
48 ByteBuf buf = ctx.alloc().ioBuffer(1024);
49
50 buf.writerIndex(buf.writerIndex() + 2);
51 DnsMessageUtil.encodeDnsResponse(encoder, response, buf);
52 buf.setShort(0, buf.readableBytes() - 2);
53
54 out.add(buf);
55 }
56 }