com.icthh.xm.commons.tenant.internal.ThreadLocalPrivilegedTenantContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xm-commons-tenant Show documentation
xm-commons-tenant
There is a newer version: 4.0.17
package com.icthh.xm.commons.tenant.internal;

import com.icthh.xm.commons.tenant.PlainTenant;
import com.icthh.xm.commons.tenant.PrivilegedTenantContext;
import com.icthh.xm.commons.tenant.Tenant;
import com.icthh.xm.commons.tenant.TenantContext;
import com.icthh.xm.commons.tenant.TenantKey;
import com.icthh.xm.commons.tenant.XmJvmSecurityUtils;

import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Supplier;

import static com.icthh.xm.commons.tenant.TenantContextUtils.buildTenant;

/**
 * The {@link ThreadLocalPrivilegedTenantContext} class.
 */
final class ThreadLocalPrivilegedTenantContext implements PrivilegedTenantContext {

    private final TenantContextDataHolder tenantContextDataHolder;
    private final TenantContext tenantContextDelegate;

    /**
     * Creates a TenantContext using the given TenantContext holder as its backing instance.
     *
     * @param tenantContextDataHolder the TenantContext holder that backs this TenantContext object.
     * @see TenantContextDataHolder
     */
    ThreadLocalPrivilegedTenantContext(TenantContextDataHolder tenantContextDataHolder) {
        this.tenantContextDataHolder = tenantContextDataHolder;
        this.tenantContextDelegate = new ThreadLocalTenantContext(tenantContextDataHolder);
    }

    private TenantContextDataHolder getDataHolder() {
        return tenantContextDataHolder;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void setTenant(Tenant tenant) {
        getDataHolder().setTenant(tenant);
    }

    /**
     * Destroys context instance for current thread.
     * Need to be called for each thread where used {@link ThreadLocalPrivilegedTenantContext}.
     */
    @Override
    public void destroyCurrentContext() {
        XmJvmSecurityUtils.checkSecurity();
        getDataHolder().destroyForCurrentThread();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public  V execute(Tenant tenant, Supplier supplier) {
        return TenantContextDataHolder.execute(tenant, supplier);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void execute(Tenant tenant, Runnable runnable) {
        TenantContextDataHolder.execute(tenant, runnable);
    }

    @Override
    public void execute(String tenant, Runnable runnable) {
        execute(buildTenant(tenant.toUpperCase()), runnable);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public  void execute(Tenant tenant, Consumer consumer, V value) {
        TenantContextDataHolder.execute(tenant, consumer, value);
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public boolean isInitialized() {
        return tenantContextDelegate.isInitialized();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public Optional getTenant() {
        return tenantContextDelegate.getTenant();
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public Optional getTenantKey() {
        return tenantContextDelegate.getTenantKey();
    }

}