View Javadoc
1   /*
2    * This file is part of Waarp Project (named also Waarp or GG).
3    *
4    *  Copyright (c) 2019, Waarp SAS, and individual contributors by the @author
5    *  tags. See the COPYRIGHT.txt in the distribution for a full listing of
6    * individual contributors.
7    *
8    *  All Waarp Project is free software: you can redistribute it and/or
9    * modify it under the terms of the GNU General Public License as published by
10   * the Free Software Foundation, either version 3 of the License, or (at your
11   * option) any later version.
12   *
13   * Waarp is distributed in the hope that it will be useful, but WITHOUT ANY
14   * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
15   * A PARTICULAR PURPOSE. See the GNU General Public License for more details.
16   *
17   *  You should have received a copy of the GNU General Public License along with
18   * Waarp . If not, see <http://www.gnu.org/licenses/>.
19   */
20  package org.waarp.commandexec.client;
21  
22  import io.netty.channel.Channel;
23  import io.netty.channel.ChannelInitializer;
24  import io.netty.channel.ChannelPipeline;
25  import io.netty.channel.group.ChannelGroup;
26  import io.netty.channel.group.DefaultChannelGroup;
27  import io.netty.channel.socket.SocketChannel;
28  import io.netty.handler.codec.DelimiterBasedFrameDecoder;
29  import io.netty.handler.codec.Delimiters;
30  import io.netty.handler.codec.string.StringDecoder;
31  import io.netty.handler.codec.string.StringEncoder;
32  import io.netty.util.concurrent.DefaultEventExecutor;
33  import org.waarp.common.utility.WaarpStringUtils;
34  
35  /**
36   * Creates a newly configured {@link ChannelPipeline} for a new channel for
37   * LocalExecClientTest.
38   */
39  public class LocalExecClientInitializer
40      extends ChannelInitializer<SocketChannel> {
41  
42    private final ChannelGroup channelGroup;
43  
44    public LocalExecClientInitializer() {
45      channelGroup =
46          new DefaultChannelGroup("LocalExecClient", new DefaultEventExecutor());
47    }
48  
49    @Override
50    protected void initChannel(final SocketChannel ch) {
51      // Create a default pipeline implementation.
52      final ChannelPipeline pipeline = ch.pipeline();
53  
54      // Add the text line codec combination first,
55      pipeline.addLast("framer", new DelimiterBasedFrameDecoder(8192,
56                                                                Delimiters.lineDelimiter()));
57      pipeline.addLast("decoder", new StringDecoder(WaarpStringUtils.UTF8));
58      pipeline.addLast("encoder", new StringEncoder(WaarpStringUtils.UTF8));
59  
60      // and then business logic.
61      final LocalExecClientHandler localExecClientHandler =
62          new LocalExecClientHandler(this);
63      pipeline.addLast("handler", localExecClientHandler);
64    }
65  
66    /**
67     * Add a channel to the ExecClient Group
68     *
69     * @param channel
70     */
71    public final void addChannel(final Channel channel) {
72      channelGroup.add(channel);
73    }
74  
75    /**
76     * Release internal resources
77     */
78    public final void releaseResources() {
79      channelGroup.close();
80    }
81  }