| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.camel.component.netty; |
| |
| import io.netty.channel.Channel; |
| import io.netty.channel.ChannelPipeline; |
| import io.netty.handler.codec.DelimiterBasedFrameDecoder; |
| import io.netty.handler.codec.Delimiters; |
| import io.netty.handler.codec.string.StringDecoder; |
| import io.netty.handler.codec.string.StringEncoder; |
| import io.netty.util.CharsetUtil; |
| |
| import org.apache.camel.BindToRegistry; |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Processor; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.netty.handlers.ClientChannelHandler; |
| import org.apache.camel.component.netty.handlers.ServerChannelHandler; |
| import org.junit.Test; |
| |
| public class NettyCustomPipelineFactoryAsynchTest extends BaseNettyTest { |
| |
| private volatile boolean clientInvoked; |
| private volatile boolean serverInvoked; |
| |
| @BindToRegistry("cpf") |
| private TestClientChannelInitializerFactory testClientFactory = new TestClientChannelInitializerFactory(null); |
| |
| @BindToRegistry("spf") |
| private TestServerChannelPipelineFactory testServerFactory = new TestServerChannelPipelineFactory(null); |
| |
| @Override |
| protected RouteBuilder createRouteBuilder() throws Exception { |
| return new RouteBuilder() { |
| @Override |
| public void configure() throws Exception { |
| from("netty:tcp://localhost:{{port}}?serverInitializerFactory=#spf&textline=true").process(new Processor() { |
| public void process(Exchange exchange) throws Exception { |
| exchange.getOut().setBody("Forrest Gump: We was always taking long walks, and we was always looking for a guy named 'Charlie'"); |
| } |
| }); |
| } |
| }; |
| } |
| |
| @Test |
| public void testCustomClientInitializerFactory() throws Exception { |
| String response = (String)template.requestBody("netty:tcp://localhost:{{port}}?clientInitializerFactory=#cpf&textline=true", "Forest Gump describing Vietnam..."); |
| |
| assertEquals("Forrest Gump: We was always taking long walks, and we was always looking for a guy named 'Charlie'", response); |
| assertEquals(true, clientInvoked); |
| assertEquals(true, serverInvoked); |
| } |
| |
| public class TestClientChannelInitializerFactory extends ClientInitializerFactory { |
| private int maxLineSize = 1024; |
| private NettyProducer producer; |
| |
| public TestClientChannelInitializerFactory(NettyProducer producer) { |
| this.producer = producer; |
| } |
| |
| @Override |
| protected void initChannel(Channel ch) throws Exception { |
| |
| ChannelPipeline channelPipeline = ch.pipeline(); |
| clientInvoked = true; |
| channelPipeline.addLast("decoder-DELIM", new DelimiterBasedFrameDecoder(maxLineSize, true, Delimiters.lineDelimiter())); |
| channelPipeline.addLast("decoder-SD", new StringDecoder(CharsetUtil.UTF_8)); |
| channelPipeline.addLast("encoder-SD", new StringEncoder(CharsetUtil.UTF_8)); |
| channelPipeline.addLast("handler", new ClientChannelHandler(producer)); |
| } |
| |
| @Override |
| public ClientInitializerFactory createPipelineFactory(NettyProducer producer) { |
| return new TestClientChannelInitializerFactory(producer); |
| } |
| } |
| |
| public class TestServerChannelPipelineFactory extends ServerInitializerFactory { |
| private int maxLineSize = 1024; |
| private NettyConsumer consumer; |
| |
| public TestServerChannelPipelineFactory(NettyConsumer consumer) { |
| this.consumer = consumer; |
| } |
| |
| @Override |
| protected void initChannel(Channel ch) throws Exception { |
| ChannelPipeline channelPipeline = ch.pipeline(); |
| serverInvoked = true; |
| channelPipeline.addLast("encoder-SD", new StringEncoder(CharsetUtil.UTF_8)); |
| channelPipeline.addLast("decoder-DELIM", new DelimiterBasedFrameDecoder(maxLineSize, true, Delimiters.lineDelimiter())); |
| channelPipeline.addLast("decoder-SD", new StringDecoder(CharsetUtil.UTF_8)); |
| channelPipeline.addLast("handler", new ServerChannelHandler(consumer)); |
| } |
| |
| @Override |
| public ServerInitializerFactory createPipelineFactory(NettyConsumer consumer) { |
| return new TestServerChannelPipelineFactory(consumer); |
| } |
| } |
| } |