com.redis.spring.batch.memcached.MemcachedItemWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-batch-memcached Show documentation

Spring Batch Memcached

The newest version!

package com.redis.spring.batch.memcached;

import java.time.Duration;
import java.util.List;
import java.util.function.Supplier;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.springframework.batch.item.Chunk;
import org.springframework.batch.item.ExecutionContext;
import org.springframework.batch.item.ItemStreamWriter;
import org.springframework.util.Assert;

import net.spy.memcached.MemcachedClient;
import net.spy.memcached.internal.OperationFuture;
import net.spy.memcached.transcoders.Transcoder;

public class MemcachedItemWriter implements ItemStreamWriter {

	public static final Duration DEFAULT_TIMEOUT = Duration.ofSeconds(10);

	private static final Transcoder transcoder = new ByteArrayTranscoder();

	private final Supplier clientSupplier;
	private Duration timeout = DEFAULT_TIMEOUT;

	private MemcachedClient client;

	public MemcachedItemWriter(Supplier clientSupplier) {
		this.clientSupplier = clientSupplier;
	}

	@Override
	public synchronized void open(ExecutionContext executionContext) {
		if (client == null) {
			client = clientSupplier.get();
		}
	}

	@Override
	public synchronized void close() {
		if (client != null) {
			client.shutdown();
			client = null;
		}
	}

	@Override
	public void write(Chunk chunk) throws Exception {
		Stream> stream = chunk.getItems().stream().map(this::write);
		List results = MemcachedUtils.getAll(timeout, stream.collect(Collectors.toList()));
		for (int index = 0; index < results.size(); index++) {
			Boolean result = results.get(index);
			Assert.isTrue(Boolean.TRUE.equals(result), "Could not write key " + chunk.getItems().get(index).getKey());
		}
	}

	private OperationFuture write(MemcachedEntry entry) {
		return client.set(entry.getKey(), entry.getExpiration(), entry.getValue(), transcoder);
	}

	public Duration getTimeout() {
		return timeout;
	}

	public void setTimeout(Duration timeout) {
		this.timeout = timeout;
	}
}