View Javadoc
1   /*
2    * Copyright 2023 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.netty.handler.codec.compression;
17  
18  import io.netty.channel.ChannelFuture;
19  import io.netty.channel.ChannelHandlerContext;
20  import io.netty.channel.ChannelPromise;
21  import io.netty.util.concurrent.Future;
22  
23  import java.util.concurrent.TimeUnit;
24  
25  final class EncoderUtil {
26      private static final int THREAD_POOL_DELAY_SECONDS = 10;
27  
28      static void closeAfterFinishEncode(final ChannelHandlerContext ctx, final ChannelFuture finishFuture,
29                                         final ChannelPromise promise) {
30          if (!finishFuture.isDone()) {
31              // Ensure the channel is closed even if the write operation completes in time.
32              final Future<?> future = ctx.executor().schedule(new Runnable() {
33                  @Override
34                  public void run() {
35                      ctx.close(promise);
36                  }
37              }, THREAD_POOL_DELAY_SECONDS, TimeUnit.SECONDS);
38  
39              finishFuture.addListener(f -> {
40                  // Cancel the scheduled timeout.
41                  future.cancel(true);
42                  if (!promise.isDone()) {
43                      ctx.close(promise);
44                  }
45              });
46          } else {
47              ctx.close(promise);
48          }
49      }
50  
51      private EncoderUtil() { }
52  }
53