com.nimbusds.infinispan.persistence.sql.ExpiredEntryReaper Maven / Gradle / Ivy

Go to download
package com.nimbusds.infinispan.persistence.sql;


import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

import static org.jooq.impl.DSL.table;

import com.nimbusds.infinispan.persistence.common.InfinispanEntry;
import net.jcip.annotations.ThreadSafe;
import org.infinispan.metadata.InternalMetadata;
import org.infinispan.persistence.spi.AdvancedCacheWriter;
import org.jooq.DSLContext;


/**
 * Expired entry reaper.
 */
@ThreadSafe
class ExpiredEntryReaper {


	/**
	 * The SQL record transformer.
	 */
	private final SQLRecordTransformer recordTransformer;


	/**
	 * Creates a new reaper for expired entries.
	 *
	 * @param recordTransformer The SQL record transformer. Must not be
	 *                          {@code null}.
	 */
	public ExpiredEntryReaper(final SQLRecordTransformer recordTransformer) {

		assert recordTransformer != null;
		this.recordTransformer = recordTransformer;
	}


	/**
	 * Purges the expired persisted entries according to their metadata
	 * timestamps (if set / persisted).
	 *
	 * @param dsl           The DSL context. Must not be {@code null}.
	 * @param purgeListener The purge listener. Must not be {@code null}.
	 *
	 * @return The number of purged entries.
	 */
	public int purge(final DSLContext dsl, final AdvancedCacheWriter.PurgeListener purgeListener) {

		final long now = new Date().getTime();

		// The keys for deletion
		List forDeletion = new LinkedList<>();
		
		dsl.select()
			.from(table(recordTransformer.getTableName()))
			.stream()
			.forEach(record -> {
				
				InfinispanEntry infinispanEntry = recordTransformer.toInfinispanEntry(record);
				
				InternalMetadata metadata = infinispanEntry.getMetadata();
				
				if (metadata == null) {
					return; // no metadata found
				}
				
				if (metadata.isExpired(now)) {
					
					// Mark record for deletion
					forDeletion.add(infinispanEntry.getKey());
				}
			});


		final AtomicInteger deleteCounter = new AtomicInteger(0);

		for (K key: forDeletion) {

			// Delete SQL record
			int numDeleted = dsl.deleteFrom(table(recordTransformer.getTableName()))
				.where(recordTransformer.resolveSelectionConditions(key))
				.execute();
			
			if (numDeleted == 1) {
				// Notify listener, interested in the Infinispan entry key
				purgeListener.entryPurged(key);
				deleteCounter.incrementAndGet();
			}
		}

		return deleteCounter.get();
	}
}