TcpSyslogMessageSenderLoadTest.java

/*
 * Copyright 2010-2014, CloudBees Inc.
 *
 * 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 com.cloudbees.syslog.sender;

import com.cloudbees.syslog.Facility;
import com.cloudbees.syslog.Severity;

import java.io.IOException;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author <a href="mailto:cleclerc@cloudbees.com">Cyrille Le Clerc</a>
 */
public class TcpSyslogMessageSenderLoadTest {

    public static void main(String[] args) throws Exception {
        final int THREADS_COUNT = 5;
        final int ITERATION_COUNT = 100;

        ExecutorService executorService = Executors.newFixedThreadPool(THREADS_COUNT);

        final TcpSyslogMessageSender messageSender = new TcpSyslogMessageSender();
        messageSender.setDefaultMessageHostname("mysecretkey");
        messageSender.setDefaultAppName("myapp");
        messageSender.setDefaultFacility(Facility.USER);
        messageSender.setDefaultSeverity(Severity.INFORMATIONAL);
        messageSender.setSyslogServerHostname("logs2.papertrailapp.com");
        // messageSender.setSyslogServerHostname("127.0.0.1");
        messageSender.setSyslogServerPort(46022);
        messageSender.setSsl(true);
        messageSender.setSocketFlush(TcpSyslogMessageSender.SocketFlush.onSend);

        final AtomicInteger count = new AtomicInteger();

        final Random random = new Random();

        for (int i = 0; i < THREADS_COUNT; i++) {
            final String prefix = "thread-" + i + "msg-";

            Runnable command = new Runnable() {
                @Override
                public void run() {
                    for (int j = 0; j < ITERATION_COUNT; j++) {
                        try {
                            messageSender.sendMessage(prefix + j);
                            Thread.sleep(random.nextInt(3));
                        } catch (IOException e) {
                            System.err.println("ERROR in " + prefix);
                            e.printStackTrace();
                            break;
                        } catch (InterruptedException e) {
                            System.err.println("ERROR in " + prefix);
                            e.printStackTrace();
                            break;
                        }
                    }
                }
            };

            executorService.execute(command);
        }

        executorService.shutdown();
        executorService.awaitTermination(1, TimeUnit.MINUTES);
        System.out.println("sent " + messageSender.getSendCount() + " in " + messageSender.getSendDurationInMillis() + "ms");
        System.out.println("bye");
    }
}