/*
 * Copyright 2002-2019 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
 *
 *      https://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.integration.jdbc.store.channel;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 *
 * @author Gunnar Hillert
 *
 */
public class TestService {

	private static final Log log = LogFactory.getLog(TestService.class);

	private Map<String, String> seen = new ConcurrentHashMap<String, String>();

	private AtomicInteger duplicateMessagesCount = new AtomicInteger(0);

	private final CountDownLatch latch;

	private int maxNumberOfMessages = 10;
	private int threadSleep = 10;

	public TestService(int maxNumberOfMessages, int threadSleep) {
		super();
		this.maxNumberOfMessages = maxNumberOfMessages;
		latch = new CountDownLatch(maxNumberOfMessages);
		this.threadSleep = threadSleep;
	}

	public TestService() {
		super();
		this.maxNumberOfMessages = 10;
		latch = new CountDownLatch(maxNumberOfMessages);
	}

	public void process(final String message) {
		log.info("Received: " + message);

		if (seen.containsKey(message)) {
			log.error("Already seen: " + message);
			duplicateMessagesCount.addAndGet(1);
		}
		else {
			seen.put(message, message);
			log.info("Pre: " + message);

			if (this.threadSleep > 0) {
				try {
					Thread.sleep(10);
				}
				catch (InterruptedException e) {
					Thread.currentThread().interrupt();
				}
			}
			log.info("Post: " + message + "; latch: " + latch.getCount());
			latch.countDown();
		}
	}

	public boolean await(int milliseconds) throws InterruptedException {
		return latch.await(milliseconds, TimeUnit.MILLISECONDS);
	}

	public Map<String, String> getSeenMessages() {
		return seen;
	}

	public int getDuplicateMessagesCount() {
		return duplicateMessagesCount.get();
	}

}
