io.github.wycst.wast.clients.http.provider.DefaultServiceProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wast Show documentation

Wast is a high-performance Java toolset library package that includes JSON, YAML, CSV, HttpClient, JDBC and EL engines

There is a newer version: 0.0.25

package io.github.wycst.wast.clients.http.provider;

import io.github.wycst.wast.clients.http.definition.HttpClientException;
import io.github.wycst.wast.clients.http.definition.HttpClientRequest;
import io.github.wycst.wast.clients.http.loadbalance.LoadBalanceStrategy;
import io.github.wycst.wast.clients.http.loadbalance.PollingLoadBalanceStrategy;

import java.net.MalformedURLException;
import java.net.URL;
import java.util.*;

/**
 * @Author: wangy
 * @Date: 2020/7/10 15:14
 * @Description:
 */
public class DefaultServiceProvider implements ServiceProvider {

    private final Map servers;

    // 负载规则
    private LoadBalanceStrategy loadBalanceStrategy;

    public DefaultServiceProvider() {
        this(new LinkedHashMap());
    }

    public DefaultServiceProvider(Map servers) {
        if (servers == null) {
            throw new HttpClientException(" servers is null ");
        }
        this.servers = servers;
        this.loadBalanceStrategy = new PollingLoadBalanceStrategy();
    }

    public final void setLoadBalanceStrategy(LoadBalanceStrategy loadBalanceStrategy) {
        this.loadBalanceStrategy = loadBalanceStrategy;
    }

    @Override
    public final void registerServer(ServerZone serverZone) {
        servers.put(serverZone.getServerName(), serverZone);
    }

    @Override
    public final ServiceInstance getServiceInstance(HttpClientRequest httpRequest) throws MalformedURLException {
        URL url = httpRequest.getURL();
        if (!httpRequest.isUseDefaultPort()) {
            return null;
        }
        String hostname = url.getHost();
        if (ifExist(hostname)) {
            ServiceInstance serviceInstance = loadBalanceStrategy.select(getServer(hostname));
            return serviceInstance;
        }
        return null;
    }

    @Override
    public final ServerZone getServer(String serviceName) {
        return servers.get(serviceName);
    }

    @Override
    public final void clear() {
        servers.clear();
    }

    @Override
    public final void clearIfNotExist(Collection doms) {
        List serviceList = new ArrayList(servers.keySet());
        for (String serviceName : serviceList) {
            if (!doms.contains(serviceName)) {
                ServerZone serverZone = servers.get(serviceName);
                if (!serverZone.isStaticServer()) {
                    servers.remove(serviceName);
                }
            }
        }
    }

    private boolean ifExist(String serverName) {
        return servers.containsKey(serverName);
    }

    public void destroy() {
        this.clear();
    }
}