com.github.lianjiatech.retrofit.spring.boot.degrade.resilience4j.CircuitBreakerConfigRegistry Maven / Gradle / Ivy

Go to download

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

retrofit-spring-boot-starter

There is a newer version: 3.1.3

Show newest version

package com.github.lianjiatech.retrofit.spring.boot.degrade.resilience4j;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.PostConstruct;

import org.springframework.util.Assert;

import com.github.lianjiatech.retrofit.spring.boot.core.Constants;

import io.github.resilience4j.circuitbreaker.CircuitBreakerConfig;

/**
 * @author 陈添明
 * @since 2022/5/24 9:37 下午
 */
public class CircuitBreakerConfigRegistry {

    private final Map circuitBreakerConfigMap;

    private final List registrars;

    public CircuitBreakerConfigRegistry(List registrars) {
        this.circuitBreakerConfigMap = new HashMap<>(8);
        circuitBreakerConfigMap.put(Constants.DEFAULT_CIRCUIT_BREAKER_CONFIG, CircuitBreakerConfig.ofDefaults());
        this.registrars = registrars;
    }

    @PostConstruct
    public void init() {
        if (registrars == null) {
            return;
        }
        registrars.forEach(registrar -> registrar.register(this));
    }

    public void register(String name, CircuitBreakerConfig circuitBreakerConfig) {
        circuitBreakerConfigMap.put(name, circuitBreakerConfig);
    }

    public CircuitBreakerConfig get(String name) {
        CircuitBreakerConfig circuitBreakerConfig = circuitBreakerConfigMap.get(name);
        Assert.notNull(circuitBreakerConfig, "Specified CircuitBreakerConfig not found! name=" + name);
        return circuitBreakerConfig;
    }
}