org.hibernate.ogm.datastore.ignite.IgnitePessimisticReadLockingStrategy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hibernate-ogm-ignite Show documentation
Persist objects in Apache Ignite
The newest version!
/*
 * Hibernate OGM, Domain model persistence for NoSQL datastores
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.ogm.datastore.ignite;

import java.io.Serializable;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;

import org.apache.ignite.IgniteCache;
import org.apache.ignite.binary.BinaryObject;
import org.hibernate.LockMode;
import org.hibernate.StaleObjectStateException;
import org.hibernate.dialect.lock.LockingStrategy;
import org.hibernate.dialect.lock.LockingStrategyException;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.ogm.datastore.ignite.impl.IgniteDatastoreProvider;
import org.hibernate.ogm.datastore.ignite.logging.impl.Log;
import org.hibernate.ogm.datastore.ignite.logging.impl.LoggerFactory;
import org.hibernate.ogm.model.impl.EntityKeyBuilder;
import org.hibernate.ogm.model.key.spi.EntityKey;
import org.hibernate.ogm.persister.impl.OgmEntityPersister;
import org.hibernate.ogm.type.spi.GridType;
import org.hibernate.ogm.type.spi.TypeTranslator;
import org.hibernate.persister.entity.Lockable;

public class IgnitePessimisticReadLockingStrategy implements LockingStrategy {

	private static final Log log = LoggerFactory.getLogger();

	private final Lockable lockable;
	private final LockMode lockMode;

	private IgniteDatastoreProvider provider;

	public IgnitePessimisticReadLockingStrategy(Lockable lockable, LockMode lockMode, IgniteDatastoreProvider provider) {
		this.lockable = lockable;
		this.lockMode = lockMode;
		this.provider = provider;
		// this.provider = getProvider(lockable.getFactory());
	}

	@Override
	public void lock(Serializable id, Object version, Object object, int timeout, SharedSessionContractImplementor session)
			throws StaleObjectStateException, LockingStrategyException {

		TypeTranslator typeTranslator = lockable.getFactory().getServiceRegistry().getService( TypeTranslator.class );
		GridType idGridType = typeTranslator.getType( lockable.getIdentifierType() );
		EntityKey key = EntityKeyBuilder.fromData(
								( (OgmEntityPersister) lockable ).getRootEntityKeyMetadata(),
								idGridType,
								id,
								session
						);

		IgniteCache cache = provider.getEntityCache( key.getMetadata() );
		Lock lock = cache.lock( provider.createKeyObject( key ) );
		try {
			lock.tryLock( timeout, TimeUnit.MILLISECONDS );
		}
		catch (InterruptedException e) {
			throw new IgniteLockingStrategyException( object, e.getMessage(), e );
		}
	}

	// private static IgniteDatastoreProvider getProvider(SessionFactoryImplementor factory) {
	// DatastoreProvider service = factory.getServiceRegistry().getService( DatastoreProvider.class );
	//
	// if ( service instanceof IgniteDatastoreProvider ) {
	// return IgniteDatastoreProvider.class.cast( service );
	// }
	// else {
	// throw log.unexpectedDatastoreProvider( service.getClass(), IgniteDatastoreProvider.class );
	// }
	// }

}