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