argos/notification-service/notification/src/main/java/gr/cite/notification/data/tenant/TenantFilterAspect.java

package gr.cite.notification.data.tenant;

import gr.cite.notification.common.scope.tenant.TenantScope;
import jakarta.persistence.EntityManager;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.hibernate.Session;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.management.InvalidApplicationException;

@Aspect
@Component
public class TenantFilterAspect {

    private final TenantScope tenantScope;

    @Autowired
    public TenantFilterAspect(
            TenantScope tenantScope
    ) {
        this.tenantScope = tenantScope;
    }

    @AfterReturning(
            pointcut = "bean(entityManagerFactory) && execution(* createEntityManager(..))",
            returning = "retVal")
    public void getSessionAfter(JoinPoint joinPoint, Object retVal) throws InvalidApplicationException {
        if (retVal instanceof EntityManager && tenantScope.isSet()) {
            Session session = ((EntityManager) retVal).unwrap(Session.class);
            if(!tenantScope.isDefaultTenant()) {
                session
                        .enableFilter(TenantScopedBaseEntity.TENANT_FILTER)
                        .setParameter(TenantScopedBaseEntity.TENANT_FILTER_TENANT_PARAM, tenantScope.getTenant().toString());
            } else {
                session
                        .enableFilter(TenantScopedBaseEntity.DEFAULT_TENANT_FILTER);
            }
        }
    }

}