/*
* Copyright 2013-2014 the original author or authors.
*
* Licensed 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.springframework.xd.integration.reactor;
import static org.junit.Assert.assertTrue;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ByteChannel;
import java.nio.channels.DatagramChannel;
import java.nio.channels.SocketChannel;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.beans.factory.config.PropertyPlaceholderConfigurer;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.integration.channel.DirectChannel;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHandler;
import org.springframework.messaging.MessagingException;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.xd.integration.reactor.syslog.SyslogInboundChannelAdapter;
import reactor.core.Environment;
import reactor.core.Reactor;
import reactor.core.spec.Reactors;
import reactor.spring.context.config.EnableReactor;
/**
* @author Jon Brisbin
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration
public class SyslogInboundChannelAdapterIntegrationTests {
static int MSG_COUNT = 2000;
CountDownLatch latch;
Environment env;
long start;
long end;
double elapsed;
@Autowired
SyslogInboundChannelAdapter channelAdapter;
@Autowired
SyslogWriter syslogWriter1;
@Autowired
SyslogWriter syslogWriter2;
@Autowired
SyslogWriter syslogWriter3;
@Autowired
SyslogWriter syslogWriter4;
@Autowired
DirectChannel output;
@Before
public void setup() {
env = new Environment();
latch = new CountDownLatch(MSG_COUNT * 4);
output.subscribe(new MessageHandler() {
@Override
public void handleMessage(Message<?> message) throws MessagingException {
latch.countDown();
}
});
}
private void start(String name) {
System.out.println("Starting " + name + " test...");
start = System.currentTimeMillis();
}
private void stop() {
end = System.currentTimeMillis();
elapsed = (end - start);
long throughput = Math.round((MSG_COUNT * 4) / (elapsed / 1000));
System.out.println("Sent " + (MSG_COUNT * 4) + " msgs in " + Math.round(elapsed) + "ms. Throughput: " +
throughput +
"/sec");
}
@Test
public void testSyslogInboundChannelAdapter() throws InterruptedException {
start("syslog");
syslogWriter1.start();
syslogWriter2.start();
syslogWriter3.start();
syslogWriter4.start();
assertTrue("Latch did not time out", latch.await(120, TimeUnit.SECONDS));
stop();
}
@Configuration
@EnableReactor
static class TestConfiguration {
static @Bean
public PropertyPlaceholderConfigurer propertyPlaceholderConfigurer() {
return new PropertyPlaceholderConfigurer();
}
@Value("${transport:tcp}")
String transport;
@Bean
public Reactor reactor(Environment env) {
return Reactors.reactor().env(env).get();
}
@Bean
public DirectChannel output() {
return new DirectChannel();
}
@Bean
public SyslogInboundChannelAdapter syslogInboundChannelAdapter(Environment env, DirectChannel output) {
SyslogInboundChannelAdapter sica = new SyslogInboundChannelAdapter(env);
sica.setOutputChannel(output);
sica.setTransport(transport);
return sica;
}
@Bean
public SyslogWriter syslogWriter1() {
return new SyslogWriter(transport, MSG_COUNT);
}
@Bean
public SyslogWriter syslogWriter2() {
return new SyslogWriter(transport, MSG_COUNT);
}
@Bean
public SyslogWriter syslogWriter3() {
return new SyslogWriter(transport, MSG_COUNT);
}
@Bean
public SyslogWriter syslogWriter4() {
return new SyslogWriter(transport, MSG_COUNT);
}
}
static class SyslogWriter extends Thread {
final int runs;
final String transport;
SyslogWriter() {
this("tcp", 1);
}
SyslogWriter(String transport, int runs) {
super();
this.transport = transport;
this.runs = runs;
}
@Override
public void run() {
try {
InetSocketAddress connectAddr = new InetSocketAddress("127.0.0.1", 5140);
ByteChannel out;
if ("tcp".equals(transport)) {
SocketChannel channel = SocketChannel.open();
channel.connect(connectAddr);
out = channel;
}
else {
DatagramChannel channel = DatagramChannel.open();
channel.connect(connectAddr);
out = channel;
}
ByteBuffer buff = ByteBuffer.wrap(
("<34>Oct 11 22:14:15 mymachine su: 'su root' failed for lonvick on /dev/pts/8\n").getBytes()
);
for (int i = 0; i < runs; i++) {
out.write(buff);
buff.flip();
if ("udp".equals(transport)) {
Thread.sleep(5);
}
}
out.close();
}
catch (Exception e) {
throw new IllegalStateException(e);
}
}
}
}