com.epam.reportportal.calculation.BatchProcessing Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of service-jobs Show documentation

EPAM Report portal. Service jobs

There is a newer version: 5.13.0

package com.epam.reportportal.calculation;

import org.springframework.scheduling.TaskScheduler;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ScheduledFuture;

/**
 * Batch processing, grouping in batch based on amount or time unit.
 * @param 
 */
public abstract class BatchProcessing {

	private List objectList;
	private final TaskScheduler scheduler;
	private volatile ScheduledFuture scheduledTask;
	private int batchSize;
	private long timeout;

	public BatchProcessing(int batchSize, long timeout, TaskScheduler scheduler) {
		if (timeout < 0 || scheduler == null) {
			throw new IllegalArgumentException("Timeout must be greater than 0 and scheduler must be not null");
		}
		objectList = new ArrayList<>();
		this.batchSize = batchSize;
		this.timeout = timeout;
		this.scheduler = scheduler;
		this.scheduledTask = this.scheduler.schedule(this::processAndSchedule, getNextTime());
	}

	private Date getNextTime() {
		return new Date(System.currentTimeMillis() + this.timeout);
	}

	public void add(T message) {
		synchronized (this) {
			this.objectList.add(message);
			if (this.objectList.size() >= this.batchSize) {
				processAndSchedule();
			}
		}
	}

	private void processAndSchedule() {
		List copyObjectList;
		synchronized (this) {
			copyObjectList = new ArrayList<>(this.objectList);
			this.objectList.clear();
		}

		if (!copyObjectList.isEmpty()) {
			process(copyObjectList);
		}

		if (this.scheduledTask != null) {
			this.scheduledTask.cancel(true);
		}

		this.scheduledTask = this.scheduler.schedule(this::processAndSchedule, getNextTime());
	}

	protected abstract void process(List objectList);
}