转载

聊聊nacos address的getCluster

序

本文主要研究一下nacos address的getCluster

ServerListController

nacos-1.1.3/address/src/main/java/com/alibaba/nacos/address/controller/ServerListController.java

@RestController
public class ServerListController {

    @Autowired
    private ServiceManager serviceManager;

    @Autowired
    private AddressServerGeneratorManager addressServerBuilderManager;

    /**
     * @param product will get Ip list of that products to be associated
     * @param cluster will get Ip list of that product cluster to be associated
     * @return
     */
    @RequestMapping(value = "/{product}/{cluster}", method = RequestMethod.GET)
    public ResponseEntity getCluster(@PathVariable String product,
                                     @PathVariable String cluster) {

        String productName = addressServerBuilderManager.generateProductName(product);
        String serviceName = addressServerBuilderManager.generateNacosServiceName(productName);
        Service service = serviceManager.getService(Constants.DEFAULT_NAMESPACE_ID, serviceName);
        if (service == null) {

            return ResponseEntity.status(HttpStatus.NOT_FOUND).body("product=" + product + " not found.");
        }

        if (!service.getClusterMap().containsKey(cluster)) {

            return ResponseEntity.status(HttpStatus.NOT_FOUND).body("product=" + product + ",cluster=" + cluster + " not found.");
        }

        Cluster clusterObj = service.getClusterMap().get(cluster);
        return ResponseEntity.status(HttpStatus.OK).body(addressServerBuilderManager.generateResponseIps(clusterObj.allIPs(false)));
    }
}

getCluster方法首先通过addressServerBuilderManager.generateProductName(product)获取productName，通过addressServerBuilderManager.generateNacosServiceName(productName)获取serviceName
然后通过serviceManager.getService(Constants.DEFAULT_NAMESPACE_ID, serviceName)获取service，之后判断service.getClusterMap()是否包含cluster，不包含返回404
最后通过service.getClusterMap().get(cluster)获取cluster，然后通过addressServerBuilderManager.generateResponseIps(clusterObj.allIPs(false))返回结果

AddressServerGeneratorManager

nacos-1.1.3/address/src/main/java/com/alibaba/nacos/address/component/AddressServerGeneratorManager.java

@Component
public class AddressServerGeneratorManager {

    public String generateProductName(String name) {

        if (StringUtils.isBlank(name) || AddressServerConstants.DEFAULT_PRODUCT.equals(name)) {

            return AddressServerConstants.ALIWARE_NACOS_DEFAULT_PRODUCT_NAME;
        }

        return String.format(AddressServerConstants.ALIWARE_NACOS_PRODUCT_DOM_TEMPLATE, name);
    }

    public String generateNacosServiceName(String rawServiceName) {

        if (rawServiceName.indexOf(Constants.DEFAULT_GROUP) != -1) {
            return rawServiceName;
        }

        return Constants.DEFAULT_GROUP + AddressServerConstants.GROUP_SERVICE_NAME_SEP + rawServiceName;
    }

    public String generateResponseIps(List<Instance> instanceList) {

        StringBuilder ips = new StringBuilder();
        instanceList.forEach(instance -> {
            ips.append(instance.getIp() + ":" + instance.getPort());
            ips.append("/n");
        });

        return ips.toString();
    }

    //......
}

generateProductName方法对于name为空或者是nacos的，返回nacos.as.default，否则返回nacos.as.前缀加name；generateNacosServiceName会返回DEFAULT_GROUP@@加rawServiceName，默认rawServiceName是不包含DEFAULT_GROUP的；generateResponseIps方法将instanceList转为ip及port并用换行拼接的string

小结

getCluster方法首先通过addressServerBuilderManager.generateProductName(product)获取productName，通过addressServerBuilderManager.generateNacosServiceName(productName)获取serviceName
然后通过serviceManager.getService(Constants.DEFAULT_NAMESPACE_ID, serviceName)获取service，之后判断service.getClusterMap()是否包含cluster，不包含返回404
最后通过service.getClusterMap().get(cluster)获取cluster，然后通过addressServerBuilderManager.generateResponseIps(clusterObj.allIPs(false))返回结果