com.icthh.xm.commons.tenant.internal.ThreadLocalAccessorRegistryInterceptor 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.20

package com.icthh.xm.commons.tenant.internal;

import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Method;

@Slf4j
public class ThreadLocalAccessorRegistryInterceptor {

    /**
     * This method registers ThreadLocalAccessor in micrometer ContextRegistry. Ignored if micrometer is missing.
     * ContextRegistry is used for thread local transfer between threads in Project Reactor.
     * Please read context-propagation library for details.
     *
     * @param key the key to associate with the ThreadLocal value
     * @param threadLocal the underlying ThreadLocal
     */
    static void register(String key, ThreadLocal threadLocal) {
        try {
            Class contextRegistry = Class.forName("io.micrometer.context.ContextRegistry");
            Method getInstance = contextRegistry.getMethod("getInstance");
            Method registerThreadLocalAccessor = contextRegistry.getMethod("registerThreadLocalAccessor", String.class, ThreadLocal.class);

            Object instance = getInstance.invoke(null);
            registerThreadLocalAccessor.invoke(instance, key, threadLocal);
            log.info("Successfully registered ThreadLocal in micrometer ContextRegistry by key '{}'", key);

        } catch (ClassNotFoundException | NoSuchMethodException | LinkageError ignored) {
            log.error("Unexpected reflective exception while registering ThreadLocal in micrometer " +
                "ContextRegistry: {}", ignored.getMessage());
        } catch (Throwable err) {
            log.error("Unexpected exception while registering ThreadLocal in micrometer ContextRegistry. " +
                "The feature is considered disabled due to this:", err);
        }
    }
}