package org.infinispan.eviction;
import org.infinispan.commons.CacheException;
import org.infinispan.commons.marshall.StreamingMarshaller;
import org.infinispan.commons.util.Util;
import org.infinispan.configuration.cache.Configuration;
import org.infinispan.container.DataContainer;
import org.infinispan.container.entries.InternalCacheEntry;
import org.infinispan.context.impl.ImmutableContext;
import org.infinispan.factories.annotations.ComponentName;
import org.infinispan.factories.annotations.Inject;
import org.infinispan.factories.annotations.Start;
import org.infinispan.factories.annotations.Stop;
import org.infinispan.marshall.core.MarshalledEntry;
import org.infinispan.marshall.core.MarshalledEntryFactory;
import org.infinispan.persistence.CacheLoaderException;
import org.infinispan.persistence.manager.PersistenceManager;
import org.infinispan.marshall.core.MarshalledEntryImpl;
import org.infinispan.notifications.cachelistener.CacheNotifier;
import org.infinispan.util.TimeService;
import org.infinispan.util.logging.Log;
import org.infinispan.util.logging.LogFactory;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicLong;
import static org.infinispan.factories.KnownComponentNames.CACHE_MARSHALLER;
import static org.infinispan.persistence.PersistenceUtil.internalMetadata;
public class PassivationManagerImpl implements PassivationManager {
PersistenceManager persistenceManager;
CacheNotifier notifier;
Configuration cfg;
boolean statsEnabled = false;
boolean enabled = false;
private static final Log log = LogFactory.getLog(PassivationManagerImpl.class);
private final AtomicLong passivations = new AtomicLong(0);
private DataContainer container;
private TimeService timeService;
private static final boolean trace = log.isTraceEnabled();
private MarshalledEntryFactory marshalledEntryFactory;
@Inject
public void inject(PersistenceManager persistenceManager, CacheNotifier notifier, Configuration cfg, DataContainer container,
TimeService timeService, MarshalledEntryFactory marshalledEntryFactory) {
this.persistenceManager = persistenceManager;
this.notifier = notifier;
this.cfg = cfg;
this.container = container;
this.timeService = timeService;
this.marshalledEntryFactory = marshalledEntryFactory;
}
@Start(priority = 12)
public void start() {
enabled = cfg.persistence().passivation() && cfg.persistence().usingStores();
if (enabled) {
statsEnabled = cfg.jmxStatistics().enabled();
}
}
@Override
public boolean isEnabled() {
return enabled;
}
@Override
public void passivate(InternalCacheEntry entry) {
if (enabled && entry != null) {
Object key = entry.getKey();
// notify listeners that this entry is about to be passivated
notifier.notifyCacheEntryPassivated(key, entry.getValue(), true,
ImmutableContext.INSTANCE, null);
if (trace) log.tracef("Passivating entry %s", key);
try {
MarshalledEntry marshalledEntry = marshalledEntryFactory.newMarshalledEntry(entry.getKey(), entry.getValue(),
internalMetadata(entry));
persistenceManager.writeToAllStores(marshalledEntry, false);
if (statsEnabled) passivations.getAndIncrement();
} catch (CacheException e) {
log.unableToPassivateEntry(key, e);
}
notifier.notifyCacheEntryPassivated(key, null, false,
ImmutableContext.INSTANCE, null);
}
}
@Override
@Stop(priority = 9)
public void passivateAll() throws CacheLoaderException {
if (enabled) {
long start = timeService.time();
log.passivatingAllEntries();
for (InternalCacheEntry e : container) {
if (trace) log.tracef("Passivating %s", e.getKey());
persistenceManager.writeToAllStores(marshalledEntryFactory.newMarshalledEntry(e.getKey(), e.getValue(),
internalMetadata(e)), false);
}
log.passivatedEntries(container.size(),
Util.prettyPrintTime(timeService.timeDuration(start, TimeUnit.MILLISECONDS)));
}
}
@Override
public long getPassivationCount() {
return passivations.get();
}
@Override
public void resetPassivationCount() {
passivations.set(0L);
}
}