EntityContext

1.java

public interface EntityContext {

  Object get();
  void set(Object entity);
  void remove();

}

2.java

public class EntityContextImpl implements EntityContext {

  private final ThreadLocal<Object> context = new ThreadLocal<Object>();

  public Object get() {
    return context.get();
  }

  public void set(Object entity) {
    context.set(entity);
  }

  public void remove() {
    context.remove();
  }

}