View Javadoc
1   /*
2    * Copyright 2022 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.channel.kqueue;
17  
18  import io.netty5.channel.DefaultMaxMessagesRecvBufferAllocator;
19  
20  /**
21   * {@link DefaultMaxMessagesRecvBufferAllocator} which always try to allocate buffers as big as kqueue tells us
22   * in terms of pending bytes to read.
23   */
24  public final class KQueueGuessRecvBufferAllocator extends DefaultMaxMessagesRecvBufferAllocator {
25      @Override
26      public Handle newHandle() {
27          return new MaxMessageHandle() {
28  
29              @Override
30              public int guess() {
31                  // attemptedBytesRead(...) is called with what KQueue tells us.
32                  int guess = attemptedBytesRead();
33                  if (guess == 0) {
34                      // Just in case attemptedBytesRead() returns 0 let us allocate something small.
35                      return 512;
36                  }
37                  return guess;
38              }
39          };
40      }
41  }