org.zodiac.actuate.kubernetes.leader.LeaderInfoContributor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zodiac-actuator Show documentation

Zodiac Actuator

There is a newer version: 1.6.8

package org.zodiac.actuate.kubernetes.leader;

import java.util.HashMap;
import java.util.Map;
import java.util.Optional;

import org.springframework.boot.actuate.info.Info.Builder;
import org.springframework.boot.actuate.info.InfoContributor;
import org.springframework.integration.leader.Candidate;
import org.zodic.kubernetes.leader.Leader;
import org.zodic.kubernetes.leader.LeadershipController;

public class LeaderInfoContributor implements InfoContributor {

    private final LeadershipController leadershipController;

    private final Candidate candidate;

    public LeaderInfoContributor(LeadershipController leadershipController, Candidate candidate) {
        this.leadershipController = leadershipController;
        this.candidate = candidate;
    }

    @Override
    public void contribute(Builder builder) {
        Map details = new HashMap<>();
        Optional leader = leadershipController.getLocalLeader();
        if (leader.isPresent()) {
            Leader l = leader.get();
            details.put("leaderId", l.getId());
            details.put("role", l.getRole());
            details.put("isLeader", l.isCandidate(candidate));
        } else {
            details.put("leaderId", "Unknown");
        }
        builder.withDetail("leaderElection", details);
    }

}