| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.ignite.cache.hibernate; |
| |
| import java.util.LinkedHashMap; |
| import java.util.Map; |
| import java.util.Set; |
| import org.apache.ignite.Ignite; |
| import org.apache.ignite.IgniteCheckedException; |
| import org.apache.ignite.internal.util.GridLeanSet; |
| import org.apache.ignite.internal.util.tostring.GridToStringInclude; |
| import org.apache.ignite.internal.util.typedef.F; |
| import org.apache.ignite.internal.util.typedef.internal.S; |
| |
| /** |
| * Implementation of NONSTRICT_READ_WRITE cache access strategy. |
| * <p> |
| * Configuration of L2 cache and per-entity cache access strategy can be set in the |
| * Hibernate configuration file: |
| * <pre name="code" class="xml"> |
| * <hibernate-configuration> |
| * <!-- Enable L2 cache. --> |
| * <property name="cache.use_second_level_cache">true</property> |
| * |
| * <!-- Use Ignite as L2 cache provider. --> |
| * <property name="cache.region.factory_class">org.apache.ignite.cache.hibernate.HibernateRegionFactory</property> |
| * |
| * <!-- Specify entity. --> |
| * <mapping class="com.example.Entity"/> |
| * |
| * <!-- Enable L2 cache with nonstrict-read-write access strategy for entity. --> |
| * <class-cache class="com.example.Entity" usage="nonstrict-read-write"/> |
| * </hibernate-configuration> |
| * </pre> |
| * Also cache access strategy can be set using annotations: |
| * <pre name="code" class="java"> |
| * @javax.persistence.Entity |
| * @javax.persistence.Cacheable |
| * @org.hibernate.annotations.Cache(usage = CacheConcurrencyStrategy.NONSTRICT_READ_WRITE) |
| * public class Entity { ... } |
| * </pre> |
| */ |
| public class HibernateNonStrictAccessStrategy extends HibernateAccessStrategyAdapter { |
| /** */ |
| private final ThreadLocal<WriteContext> writeCtx; |
| |
| /** |
| * @param ignite Grid. |
| * @param cache Cache. |
| * @param writeCtx Thread local instance used to track updates done during one Hibernate transaction. |
| * @param eConverter Exception converter. |
| */ |
| HibernateNonStrictAccessStrategy(Ignite ignite, |
| HibernateCacheProxy cache, |
| ThreadLocal writeCtx, |
| HibernateExceptionConverter eConverter) { |
| super(ignite, cache, eConverter); |
| |
| this.writeCtx = (ThreadLocal<WriteContext>)writeCtx; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public void lock(Object key) { |
| WriteContext ctx = writeCtx.get(); |
| |
| if (ctx == null) |
| writeCtx.set(ctx = new WriteContext()); |
| |
| ctx.locked(key); |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public void unlock(Object key) { |
| try { |
| WriteContext ctx = writeCtx.get(); |
| |
| if (ctx != null && ctx.unlocked(key)) { |
| writeCtx.remove(); |
| |
| ctx.updateCache(cache); |
| } |
| } |
| catch (IgniteCheckedException e) { |
| throw convertException(e); |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public boolean update(Object key, Object val) { |
| return false; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public boolean afterUpdate(Object key, Object val) { |
| WriteContext ctx = writeCtx.get(); |
| |
| if (log.isDebugEnabled()) |
| log.debug("Put after update [cache=" + cache.name() + ", key=" + key + ", val=" + val + ']'); |
| |
| if (ctx != null) { |
| ctx.updated(key, val); |
| |
| unlock(key); |
| |
| return true; |
| } |
| |
| return false; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public boolean insert(Object key, Object val) { |
| return false; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public boolean afterInsert(Object key, Object val) { |
| if (log.isDebugEnabled()) |
| log.debug("Put after insert [cache=" + cache.name() + ", key=" + key + ", val=" + val + ']'); |
| |
| try { |
| cache.put(key, val); |
| |
| return true; |
| } |
| catch (IgniteCheckedException e) { |
| throw convertException(e); |
| } |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public void remove(Object key) { |
| WriteContext ctx = writeCtx.get(); |
| |
| if (ctx != null) |
| ctx.removed(key); |
| } |
| |
| /** |
| * Information about updates done during single database transaction. |
| */ |
| @SuppressWarnings("TypeMayBeWeakened") |
| private static class WriteContext { |
| /** */ |
| @GridToStringInclude |
| private Map<Object, Object> updates; |
| |
| /** */ |
| @GridToStringInclude |
| private Set<Object> rmvs; |
| |
| /** */ |
| @GridToStringInclude |
| private Set<Object> locked = new GridLeanSet<>(); |
| |
| /** |
| * Marks key as locked. |
| * |
| * @param key Key. |
| */ |
| void locked(Object key) { |
| locked.add(key); |
| } |
| |
| /** |
| * Marks key as unlocked. |
| * |
| * @param key Key. |
| * @return {@code True} if last locked key was unlocked. |
| */ |
| boolean unlocked(Object key) { |
| locked.remove(key); |
| |
| return locked.isEmpty(); |
| } |
| |
| /** |
| * Marks key as updated. |
| * |
| * @param key Key. |
| * @param val Value. |
| */ |
| void updated(Object key, Object val) { |
| if (updates == null) |
| updates = new LinkedHashMap<>(); |
| |
| updates.put(key, val); |
| } |
| |
| /** |
| * Marks key as removed. |
| * |
| * @param key Key. |
| */ |
| void removed(Object key) { |
| if (rmvs == null) |
| rmvs = new GridLeanSet<>(); |
| |
| rmvs.add(key); |
| } |
| |
| /** |
| * Updates cache. |
| * |
| * @param cache Cache. |
| * @throws IgniteCheckedException If failed. |
| */ |
| void updateCache(HibernateCacheProxy cache) throws IgniteCheckedException { |
| if (!F.isEmpty(rmvs)) |
| cache.removeAll(rmvs); |
| |
| if (!F.isEmpty(updates)) |
| cache.putAll(updates); |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public String toString() { |
| return S.toString(WriteContext.class, this); |
| } |
| } |
| } |