org.zalando.tracer.spring.DefaultGeneratorResolver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracer-spring-boot-starter Show documentation

Tracing requests through a distributed system.

There is a newer version: 2.0.0-RC.3

Show newest version

package org.zalando.tracer.spring;

import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.stereotype.Component;
import org.zalando.tracer.FlowIDGenerator;
import org.zalando.tracer.Generator;
import org.zalando.tracer.PhraseGenerator;
import org.zalando.tracer.UUIDGenerator;

import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Locale;
import java.util.Map;
import java.util.Optional;

@Component
@ConditionalOnMissingBean(GeneratorResolver.class)
class DefaultGeneratorResolver implements GeneratorResolver {

    private final Map generators;

    DefaultGeneratorResolver() {
        final Map map = new LinkedHashMap<>();
        map.put("flow-id", new FlowIDGenerator());
        map.put("uuid", new UUIDGenerator());
        map.put("phrase", new PhraseGenerator());
        this.generators = Collections.unmodifiableMap(map);
    }

    @Override
    public Generator resolve(final String name) {
        return Optional.ofNullable(generators.get(name.toLowerCase(Locale.ROOT)))
                .orElseThrow(UnsupportedOperationException::new);
    }

}