拉取增量注册表并合并

就是拉取一下增量注册表,然后根据最近增量注册表中不同的操作,做不同的事情

  1. /**
  2. * 增量拉取注册表信息
  3. */
  4. private class FetchDeltaRegistryWorker extends Thread {
  5. @Override
  6. public void run() {
  7. //当客户端存活的时候继续循环,没有存活时就不能继续循环了
  8. while (registerClient.isRunning()) {
  9. try {
  10. //通过通信组件获取服务注册表
  11. LinkedList<RecentlyChangedServiceInstance> deltaRegistry = httpSender.fetchDeltaServiceRegistry();
  12. //分以下几种情况,一个是注册的操作,一种是删除的
  13. //如果是注册的话,就判断一下本地缓存的注册表中有没有这个实例,不在的话就要在本地缓存中放一份
  14. //如果是删除,看一下缓存中实例是否存在,存在就删掉
  15. //由于要操作同一份的缓存变量,这里要加锁的
  16. synchronized (registry) {
  17. mergeDeltaRegistry(deltaRegistry);
  18. }
  19. Thread.sleep(SERVICE_REGISTRY_FETCH_INTERVAL);
  20. } catch (InterruptedException e) {
  21. e.printStackTrace();
  22. }
  23. }
  24. }
  25. }
  26. /**
  27. * 合并增量注册表到本地缓存注册表
  28. * @param deltaRegistry
  29. */
  30. private void mergeDeltaRegistry(LinkedList<RecentlyChangedServiceInstance> deltaRegistry) {
  31. for (RecentlyChangedServiceInstance recentlyChangedItem : deltaRegistry) {
  32. //如果是注册操作
  33. if (ServiceInstanceOperation.REGISTER.equals(recentlyChangedItem.changedOperation)) {
  34. Map<String, ServiceInstance> serviceInstanceMap =
  35. registry.get(recentlyChangedItem.serviceInstance.getServiceName());
  36. //如果缓存没有该实例列表就要搞一份进去缓存中
  37. if (serviceInstanceMap == null) {
  38. serviceInstanceMap = new HashMap<String, ServiceInstance>();
  39. registry.put(recentlyChangedItem.serviceInstance.getServiceName(),serviceInstanceMap);
  40. }
  41. //如果已经有实例的列表了,看下有没有具体的那个实例,没有那个具体的实例的话就要把实例放进去
  42. ServiceInstance serviceInstance =
  43. serviceInstanceMap.get(recentlyChangedItem.serviceInstance.getServiceInstanceId());
  44. if (serviceInstance == null) {
  45. serviceInstanceMap.put(recentlyChangedItem.serviceInstance.getServiceInstanceId(),
  46. recentlyChangedItem.serviceInstance);
  47. }
  48. } else if (ServiceInstanceOperation.REMOVE.equals(recentlyChangedItem.changedOperation)) {
  49. //如果是删除操作,就把服务实例删除就行
  50. Map<String, ServiceInstance> serviceInstanceMap =
  51. registry.get(recentlyChangedItem.serviceInstance.getServiceName());
  52. if (serviceInstanceMap != null) {
  53. serviceInstanceMap.remove(recentlyChangedItem.serviceInstance.getServiceInstanceId());
  54. }
  55. }
  56. }
  57. }

合并注册表后对客户端注册表的校验

在操作的时候,可能因为某些原因导致最后客户端缓存的注册表的数据和服务端的数据是不一样的,因此就需要做一下这个校验的操作

  1. /**
  2. * 增量拉取注册表信息
  3. */
  4. private class FetchDeltaRegistryWorker extends Thread {
  5. @Override
  6. public void run() {
  7. //当客户端存活的时候继续循环,没有存活时就不能继续循环了
  8. while (registerClient.isRunning()) {
  9. try {
  10. //通过通信组件获取服务注册表
  11. DeltaRegistry deltaRegistry = httpSender.fetchDeltaServiceRegistry();
  12. //分以下几种情况,一个是注册的操作,一种是删除的
  13. //如果是注册的话,就判断一下本地缓存的注册表中有没有这个实例,不在的话就要在本地缓存中放一份
  14. //如果是删除,看一下缓存中实例是否存在,存在就删掉
  15. //由于要操作同一份的缓存变量,这里要加锁的
  16. synchronized (registry) {
  17. mergeDeltaRegistry(deltaRegistry.getRecentlyChangedQueue());
  18. }
  19. //需要检查一下,服务端注册表的服务实例数和客户端的是否一样
  20. //封装一下增量注册表的对象,拉取增量注册表的时候,一方面返回增量的实例,
  21. // 另一方面要返回当前这个实例在服务端的注册表中有多少个
  22. Long serviceSideTotalCount = deltaRegistry.getServiceInstanceTotalCount();
  23. Long clientSideTotalCount = 0L;
  24. for (Map<String, ServiceInstance> serviceInstanceMap : registry.values()) {
  25. clientSideTotalCount += serviceInstanceMap.size();
  26. }
  27. if (serviceSideTotalCount != clientSideTotalCount) {
  28. //重新拉取全量注册表进行纠正
  29. registry = httpSender.fetchFullServiceRegistry();
  30. }
  31. Thread.sleep(SERVICE_REGISTRY_FETCH_INTERVAL);
  32. } catch (InterruptedException e) {
  33. e.printStackTrace();
  34. }
  35. }
  36. }
  37. }