import cn.hutool.core.collection.CollUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.cloud.client.ServiceInstance;
import org.springframework.cloud.client.loadbalancer.Request;
import org.springframework.cloud.client.loadbalancer.RequestDataContext;
import org.springframework.cloud.loadbalancer.core.DelegatingServiceInstanceListSupplier;
import org.springframework.cloud.loadbalancer.core.ServiceInstanceListSupplier;
import org.springframework.http.HttpHeaders;
import org.springframework.util.MultiValueMap;
import reactor.core.publisher.Flux;

import java.util.List;
import java.util.stream.Collectors;

@Slf4j
public class VersionServiceInstanceListSupplier extends DelegatingServiceInstanceListSupplier {
    private static final String VERSION = "version";

    public VersionServiceInstanceListSupplier(ServiceInstanceListSupplier delegate) {
        super(delegate);
    }


    @Override
    public Flux<List<ServiceInstance>> get() {
        return delegate.get();
    }

    @Override
    public Flux<List<ServiceInstance>> get(Request request) {
        return delegate.get(request).map(instances -> filteredByVersion(instances, getVersion(request.getContext())));
    }


    private List<ServiceInstance> filteredByVersion(List<ServiceInstance> instances, String requestVersion) {
        if(StringUtils.isEmpty(requestVersion)){
            return instances.stream()
                    .filter(instance -> StringUtils.isBlank(instance.getMetadata().get(VERSION)))
                    .collect(Collectors.toList());
        }

        log.info("request version is {}", requestVersion);
        List<ServiceInstance> filteredInstances = instances.stream()
                .filter(instance -> requestVersion.equalsIgnoreCase(instance.getMetadata().get(VERSION)))
                .collect(Collectors.toList());

        if (CollUtil.isEmpty(filteredInstances)) {
            filteredInstances = instances.stream()
                    .filter(instance -> StringUtils.isBlank(instance.getMetadata().get(VERSION)))
                    .collect(Collectors.toList());
        }
        return filteredInstances;

    }

    private String getVersion(Object requestContext) {
        if (requestContext == null) {
            return null;
        }
        String version = null;
        if (requestContext instanceof RequestDataContext) {
            version = getVersionFromHeader((RequestDataContext) requestContext);
        }
        return version;
    }

    private String getVersionFromHeader(RequestDataContext context) {
        if (context.getClientRequest() != null) {
            String version = null;
            HttpHeaders headers = context.getClientRequest().getHeaders();
            if (headers != null) {
                version = headers.getFirst(VERSION);
            }
            if (StringUtils.isBlank(version)) {
                MultiValueMap<String, String> cookies = context.getClientRequest().getCookies();
                if (cookies != null) {
                    List<String> cookieVersions = cookies.get(VERSION);
                    if (CollUtil.isNotEmpty(cookieVersions)) {
                        version = cookieVersions.get(0);
                    }
                }
            }
            return version;
        }
        return null;
    }

}

import org.springframework.cloud.loadbalancer.core.ServiceInstanceListSupplier;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class VersionServiceInstanceListSupplierConfiguration {

    @Bean
    ServiceInstanceListSupplier serviceInstanceListSupplier(ConfigurableApplicationContext context) {
        ServiceInstanceListSupplier delegate = ServiceInstanceListSupplier.builder()
                .withDiscoveryClient()
                .withCaching()
                .build(context);
        return new VersionServiceInstanceListSupplier(delegate);
    }

}

@LoadBalancerClients(defaultConfiguration = VersionServiceInstanceListSupplierConfiguration.class)

spring:
  cloud:
    nacos:
      discovery:
        # 本地調(diào)試時(shí)修改以下信息
        metadata:
          ## 版本號(hào),唯一  避免與其他服務(wù)沖突
          version: V123