需求

小程序新版本上線需要審核，如果有接口新版本返回內(nèi)容發(fā)生了變化，后端直接上線會(huì)導(dǎo)致舊版本報(bào)錯(cuò)，不上線審核又通不過(guò)。

之前是通過(guò)寫(xiě)新接口來(lái)兼容，但是這樣會(huì)有很多兼容代碼或者冗余代碼，開(kāi)發(fā)也不容易能想到這一點(diǎn)，經(jīng)常直接修改了舊接口，于是版本控制就成了迫切的需求。

思路

所有請(qǐng)求都是走的網(wǎng)關(guān)，很自然的就能想到在網(wǎng)關(guān)層實(shí)現(xiàn)版本控制。首先想到的是在ZuulFilter過(guò)濾器中實(shí)現(xiàn)，前端所有請(qǐng)求都在請(qǐng)求頭中增加一個(gè)version的header,然后進(jìn)行匹配。但是這樣只能獲取到前端的版本，不能匹配選擇后端實(shí)例。

查詢資料后發(fā)現(xiàn)應(yīng)該在負(fù)載均衡的時(shí)候?qū)崿F(xiàn)版本控制。同樣是前端所有請(qǐng)求都在請(qǐng)求頭中增加一個(gè)version的header，后端實(shí)例都配置一個(gè)版本的tag。

實(shí)現(xiàn)

首先需要說(shuō)明的是我選擇的控制中心是consul，網(wǎng)關(guān)是zuul。

負(fù)載均衡策略被抽象為IRule接口，項(xiàng)目默認(rèn)情況下使用的IRule的子類ZoneAvoidanceRule extends PredicateBasedRule，我們需要實(shí)現(xiàn)一個(gè)PredicateBasedRule的子類來(lái)替換ZoneAvoidanceRule。

PredicateBasedRule需要實(shí)現(xiàn)一個(gè)過(guò)濾的方法我們就在這個(gè)方法里實(shí)現(xiàn)版本控制，過(guò)濾后就是默認(rèn)的負(fù)載均衡策略了，默認(rèn)是輪詢。

  /**
   * Method that provides an instance of {@link AbstractServerPredicate} to be used by this class.
   * 
   */
  public abstract AbstractServerPredicate getPredicate();

VersionPredicate

我們可以看到PredicateBasedRule的getPredicate()方法需要返回一個(gè)AbstractServerPredicate實(shí)例，這個(gè)實(shí)例具體定義了版本控制的業(yè)務(wù)邏輯。代碼如下：

private static class VersionPredicate extends AbstractServerPredicate {

    private static final String VERSION_KEY = "version";

    @Override
    public boolean apply(@NullableDecl PredicateKey predicateKey) {
      if (predicateKey == null) {
        return true;
      }
      RequestContext ctx = RequestContext.getCurrentContext();
      HttpServletRequest request = ctx.getRequest();
      String version = request.getHeader(VERSION_KEY);
      if (version == null) {
        return true;
      }
      ConsulServer consulServer = (ConsulServer) predicateKey.getServer();
      if (!consulServer.getMetadata().containsKey(VERSION_KEY)) {
        return true;
      }
      return consulServer.getMetadata().get(VERSION_KEY).equals(version);
    }
  }

首先來(lái)了解下負(fù)載均衡的過(guò)程。一個(gè)請(qǐng)求到達(dá)網(wǎng)關(guān)后會(huì)解析出對(duì)應(yīng)的服務(wù)名，然后會(huì)獲取到該服務(wù)的所有可用實(shí)例，之后就會(huì)調(diào)用我們的過(guò)濾方法過(guò)濾出該請(qǐng)求可用的所有服務(wù)實(shí)例，最后進(jìn)行輪詢負(fù)載均衡。

PredicateKey類就是上層方法將可用實(shí)例Server和loadBalancerKey封裝后的類。版本控制的業(yè)務(wù)邏輯如下：

• 判斷predicateKey是否為null，是的話直接返回true，true代表該實(shí)例可用
• 通過(guò)RequestContext獲取當(dāng)前請(qǐng)求實(shí)例HttpServletRequest，再通過(guò)請(qǐng)求實(shí)例獲取請(qǐng)求頭里的版本號(hào)
• 判斷前端請(qǐng)求是否帶了版本號(hào)，沒(méi)帶的話就不進(jìn)行版本控制直接返回true
• 獲取服務(wù)實(shí)例并轉(zhuǎn)換成ConsulServer類，這里是因?yàn)槲矣玫淖?cè)中心是consul，選擇其他的可自行轉(zhuǎn)換成對(duì)應(yīng)的實(shí)現(xiàn)類
• 判斷服務(wù)實(shí)例是否設(shè)置了版本號(hào)（例：spring.cloud.consul.discovery.tags="version=1.0.0"），可以看到我們是用consul的tags實(shí)現(xiàn)的版本控制，可以設(shè)置不同的tag實(shí)現(xiàn)很多功能
• 同樣服務(wù)實(shí)例沒(méi)有設(shè)置版本號(hào)的話也是直接返回true
• 最后進(jìn)行版本匹配，返回匹配成功的服務(wù)實(shí)例

注意的點(diǎn)

最終實(shí)現(xiàn)如下：

/**
 * @author Yuicon
 */
@Slf4j
public class VersionRule extends PredicateBasedRule {

  private final CompositePredicate predicate;

  public VersionRule() {
    super();
    this.predicate = createCompositePredicate(new VersionPredicate(),
        new AvailabilityPredicate(this, null));
  }

  @Override
  public AbstractServerPredicate getPredicate() {
    return this.predicate;
  }

  private CompositePredicate createCompositePredicate(VersionPredicate versionPredicate,
                            AvailabilityPredicate availabilityPredicate) {
    return CompositePredicate.withPredicates(versionPredicate, availabilityPredicate)
        .build();
  }

  private static class VersionPredicate extends AbstractServerPredicate {

    private static final String VERSION_KEY = "version";

    @Override
    public boolean apply(@NullableDecl PredicateKey predicateKey) {
      if (predicateKey == null) {
        return true;
      }
      RequestContext ctx = RequestContext.getCurrentContext();
      HttpServletRequest request = ctx.getRequest();
      String version = request.getHeader(VERSION_KEY);
      if (version == null) {
        return true;
      }
      ConsulServer consulServer = (ConsulServer) predicateKey.getServer();
      if (!consulServer.getMetadata().containsKey(VERSION_KEY)) {
        return true;
      }
      log.info("id is {}, header is {}, metadata is {}, result is {}",
          consulServer.getMetaInfo().getInstanceId(),
          version, consulServer.getMetadata().get(VERSION_KEY),
          consulServer.getMetadata().get(VERSION_KEY).equals(version));
      return consulServer.getMetadata().get(VERSION_KEY).equals(version);
    }
  }
}

原本我是加上@Component注解后在本地直接測(cè)試通過(guò)了?？墒窃诟碌缴a(chǎn)服務(wù)器后卻出現(xiàn)大部分請(qǐng)求都找不到的服務(wù)實(shí)例的錯(cuò)誤，搞的我一頭霧水，趕緊回滾到原來(lái)的版本。

查詢了很多資料后才找到一篇文章，發(fā)現(xiàn)需要一個(gè)Config類來(lái)聲明替換原有的負(fù)載均衡策略類。代碼如下：

@RibbonClients(defaultConfiguration = RibbonGatewayConfig.class)
@Configuration
public class RibbonGatewayConfig {

  @Bean
  public IRule versionRule() {
    return new VersionRule();
  }
}

到此為止版本控制算是實(shí)現(xiàn)成功了。

結(jié)尾

在實(shí)際使用過(guò)程中發(fā)現(xiàn)還是有很多問(wèn)題。比如前端版本號(hào)是全局唯一的，當(dāng)其中一個(gè)服務(wù)升級(jí)了版本號(hào)，就需要將所有服務(wù)都升級(jí)到該版本號(hào)，即使代碼沒(méi)有任何更改。比較好的解決方案是前端根據(jù)不同服務(wù)傳遞不同的版本號(hào)，不過(guò)前端反饋實(shí)現(xiàn)困難。

還有個(gè)妥協(xié)的方案，就是利用配置中心來(lái)對(duì)具體服務(wù)是否開(kāi)啟版本控制進(jìn)行配置，因?yàn)楝F(xiàn)在的需求只是一小段時(shí)間里需要版本控制，小程序?qū)徍诉^(guò)后就可以把舊服務(wù)實(shí)例關(guān)了。大家如果有更好的方案歡迎討論。

到此這篇關(guān)于Springcloud實(shí)現(xiàn)服務(wù)多版本控制的示例代碼的文章就介紹到這了,更多相關(guān)Springcloud多版本控制內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！