io.getunleash.metric.MetricsBucket Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of unleash-client-java Show documentation

A client library for Unleash

The newest version!

package io.getunleash.metric;

import io.getunleash.lang.Nullable;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

class MetricsBucket {
    private final ConcurrentMap toggles;
    private final LocalDateTime start;
    @Nullable private volatile LocalDateTime stop;

    MetricsBucket() {
        this.start = LocalDateTime.now(ZoneId.of("UTC"));
        this.toggles = new ConcurrentHashMap<>();
    }

    void registerCount(String toggleName, boolean active) {
        getOrCreate(toggleName).register(active);
    }

    void registerCount(String toggleName, String variantName) {
        getOrCreate(toggleName).register(variantName);
    }

    private ToggleCount getOrCreate(String toggleName) {
        return toggles.computeIfAbsent(toggleName, s -> new ToggleCount());
    }

    void end() {
        this.stop = LocalDateTime.now(ZoneId.of("UTC"));
    }

    public Map getToggles() {
        return toggles;
    }

    public LocalDateTime getStart() {
        return start;
    }

    public @Nullable LocalDateTime getStop() {
        return stop;
    }
}