| /* |
| * 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.netty4; |
| import java.io.File; |
| import java.io.FileOutputStream; |
| import java.util.Collection; |
| import java.util.Properties; |
| |
| import io.netty.buffer.ByteBufAllocator; |
| import io.netty.util.ResourceLeakDetector; |
| import org.apache.camel.CamelContext; |
| import org.apache.camel.component.properties.PropertiesComponent; |
| import org.apache.camel.converter.IOConverter; |
| import org.apache.camel.impl.JndiRegistry; |
| import org.apache.camel.test.AvailablePortFinder; |
| import org.apache.camel.test.junit4.CamelTestSupport; |
| import org.apache.logging.log4j.core.LogEvent; |
| import org.junit.AfterClass; |
| import org.junit.BeforeClass; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| |
| /** |
| * |
| */ |
| public class BaseNettyTest extends CamelTestSupport { |
| protected static final Logger LOG = LoggerFactory.getLogger(BaseNettyTest.class); |
| |
| private static volatile int port; |
| |
| @BeforeClass |
| public static void initPort() throws Exception { |
| File file = new File("target/nettyport.txt"); |
| |
| if (!file.exists()) { |
| // start from somewhere in the 25xxx range |
| port = AvailablePortFinder.getNextAvailable(25000); |
| } else { |
| // read port number from file |
| String s = IOConverter.toString(file, null); |
| port = Integer.parseInt(s); |
| // use next free port |
| port = AvailablePortFinder.getNextAvailable(port + 1); |
| } |
| |
| } |
| |
| @AfterClass |
| public static void savePort() throws Exception { |
| File file = new File("target/nettyport.txt"); |
| |
| // save to file, do not append |
| FileOutputStream fos = new FileOutputStream(file, false); |
| try { |
| fos.write(String.valueOf(port).getBytes()); |
| } finally { |
| fos.close(); |
| } |
| } |
| |
| @BeforeClass |
| public static void startLeakDetection() { |
| System.setProperty("io.netty.leakDetection.maxRecords", "100"); |
| System.setProperty("io.netty.leakDetection.acquireAndReleaseOnly", "true"); |
| ResourceLeakDetector.setLevel(ResourceLeakDetector.Level.PARANOID); |
| } |
| |
| @AfterClass |
| public static void verifyNoLeaks() throws Exception { |
| //Force GC to bring up leaks |
| System.gc(); |
| //Kick leak detection logging |
| ByteBufAllocator.DEFAULT.buffer(1).release(); |
| Collection<LogEvent> events = LogCaptureAppender.getEvents(); |
| if (!events.isEmpty()) { |
| String message = "Leaks detected while running tests: " + events; |
| // Just write the message into log to help debug |
| for (LogEvent event: events) { |
| LOG.info(event.getMessage().getFormattedMessage()); |
| } |
| LogCaptureAppender.reset(); |
| throw new AssertionError(message); |
| } |
| } |
| |
| @Override |
| protected CamelContext createCamelContext() throws Exception { |
| CamelContext context = super.createCamelContext(); |
| context.addComponent("properties", new PropertiesComponent("ref:prop")); |
| return context; |
| } |
| |
| @Override |
| protected JndiRegistry createRegistry() throws Exception { |
| JndiRegistry jndi = super.createRegistry(); |
| |
| Properties prop = new Properties(); |
| prop.setProperty("port", "" + getPort()); |
| jndi.bind("prop", prop); |
| |
| return jndi; |
| } |
| |
| protected int getNextPort() { |
| port = AvailablePortFinder.getNextAvailable(port + 1); |
| return port; |
| } |
| |
| protected int getPort() { |
| return port; |
| } |
| |
| protected String byteArrayToHex(byte[] bytes) { |
| StringBuilder sb = new StringBuilder(); |
| for (byte b : bytes) { |
| sb.append(String.format("%02x", b & 0xff)); |
| } |
| return sb.toString(); |
| } |
| |
| protected byte[] fromHexString(String hexstr) { |
| byte data[] = new byte[hexstr.length() / 2]; |
| int i = 0; |
| for (int n = hexstr.length(); i < n; i += 2) { |
| data[i / 2] = (Integer.decode("0x" + hexstr.charAt(i) |
| + hexstr.charAt(i + 1))).byteValue(); |
| } |
| return data; |
| } |
| |
| } |