Отслеживание Hazelcast через сыщика

Sleuth (на момент написания статьи) не поддерживает Hazelcast.

Решение более общее, чем просто Hazelcast: вам нужно передать brave.Span Zipkin между клиентом и сервером, но brave.Span не является сериализуемым.

Zipkin предоставляет средства, с помощью которых можно обойти это.

Учитывая brave.Span на клиенте, вы можете преобразовать его в java.util.Map:

Span span = ...
Map<String, String> map = new HashMap<>();

tracing.propagation().injector(Map<String, String>::put).inject(span.context(), map);

На сервере вы можете преобразовать java.util.Map обратно в brave.Span:

Span span = tracer.toSpan(tracing.propagation().extractor(Map<String, String>::get).extract(map).context())

Использование java.util.Map, очевидно, может быть заменено по мере необходимости, но принцип тот же.

Я не могу заставить его работать для ItemListeners. Я думаю, нам нужно иметь возможность обернуть StripedExecutor Hazelcast во что-то вроде LazyTraceThreadPoolTaskExecutor (но такое, которое принимает простой делегат Executor вместо ThreadPoolTaskExecutor).

Для EntryProcessors я взломал это вместе. Фабрика для создания EntryProcessors, передавая текущий диапазон из потока, который создает процессор. Когда процессор работает, он использует этот диапазон в качестве родительского диапазона в потоке исполнителя.

@Component
public class SleuthedEntryProcessorFactory {

    private final Tracer tracer;

    public SleuthedEntryProcessorFactory(Tracer tracer) {
        this.tracer = tracer;
    }

    /**
     * Create an entry processor that will continue the Sleuth span of the thread 
     * that invokes this method.
     * Mutate the given value as required. It will then be set on the entry.
     *
     * @param name name of the span
     * @param task task to perform on the map entry
     */
    public <K, V, R> SleuthedEntryProcessor<K, V, R> create(String name, Function<V, R> task) {
        return new SleuthedEntryProcessor<>(name, tracer.getCurrentSpan(), task);
    }
}

/**
 * Copies the MDC context (which contains Sleuth's trace ID, etc.) and the current span
 * from the thread that constructs this into the thread that runs this.

 * @param <K> key type
 * @param <V> value type
 * @param <R> return type
 */
@SpringAware
public class SleuthedEntryProcessor<K, V, R> extends AbstractEntryProcessor<K, V> {

    private final Map<String, String> copyOfContextMap;
    private final String name;
    private final Span parentSpan;
    private final Function<V, R> task;
    private transient Tracer tracer;

    public SleuthedEntryProcessor(String name, Span parentSpan, Function<V, R> task) {
        this(name, parentSpan, task, true);
    }

    public SleuthedEntryProcessor(
            String name, Span parentSpan, Function<V, R> task, boolean applyOnBackup) {
        super(applyOnBackup);
        this.name = name + "Hz";
        this.parentSpan = parentSpan;
        this.task = task;
        copyOfContextMap = MDC.getCopyOfContextMap();
    }

    @Override
    public final R process(Map.Entry<K, V> entry) {
        if (nonNull(copyOfContextMap)) {
            MDC.setContextMap(copyOfContextMap);
        }

        Span span = tracer.createSpan(toLowerHyphen(name), parentSpan);
        try {
            V value = entry.getValue();
            // The task mutates the value.
            R result = task.apply(value);
            // Set the mutated value back onto the entry.
            entry.setValue(value);
            return result;

        } finally {
            MDC.clear();
            tracer.close(span);
        }
    }

    @Autowired
    public void setTracer(Tracer tracer) {
        this.tracer = tracer;
    }
}

Затем передайте EntryProcessor в свой IMap следующим образом:

Function<V, R> process = ...;
SleuthedEntryProcessor<K, V, R> entryProcessor = sleuthedEntryProcessorFactory.create(label, process);
Map<K, R> results = iMap.executeOnEntries(entryProcessor);