gRPC學習記錄(六)--客戶端連線池

對於客戶端來說建立一個channel是昂貴的,因為建立channel需要連線,但是建立一個stub是很簡單的,就像建立一個普通物件,因此Channel就需要複用,也就是說需要實現一個連線池應用.本文使用commons-pool2來實現連線池應用. 該案例可以當成連線池模板案例.

        <!--客戶端連線池-->
        <dependency>
            <groupId>org.apache.commons</groupId>
            <artifactId>commons-pool2</artifactId>
            <version>2.4.2</version>
        </dependency>

寫法的封裝類似我的另一篇博文: ,模仿SpringJdbcTemplate模板類寫法.

首先定義一個回撥介面.

public interface WorkCallBack<S> {  void callback(S s);
}

再者定義一個產生連線池的工廠,需要繼承自BasePooledObjectFactory,其用處是產生連線池中需要的客戶端.

public class HelloWorldFactory extends BasePooledObjectFactory<HelloWorldClientSingle> {  private String host = "127.0.0.1";  private int port = 50051;  @Override
  public HelloWorldClientSingle create() throws Exception {    return new HelloWorldClientSingle(this.host,this.port);
  }  @Override
  public PooledObject<HelloWorldClientSingle> wrap(HelloWorldClientSingle helloWorldClientSingle) {    return new DefaultPooledObject<>(helloWorldClientSingle);
  }  @Override
  public void destroyObject(PooledObject<HelloWorldClientSingle> p) throws Exception {
    p.getObject().shutdown();    super.destroyObject(p);
  }
}

實現相應的連線池,注意看execute()方法,該方法是一個模板方法,進行從池中獲取客戶端和歸還客戶端給連線池,其主要邏輯都定義在WorkCallBack介面中,也因此從服務端拿回的資料要在其裡面處理完畢.

public class HelloWorldClientPool {  private static GenericObjectPool<HelloWorldClientSingle> objectPool = null;  static {    // 連線池的配置
    GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();    // 池中的最大連線數
    poolConfig.setMaxTotal(8);    // 最少的空閒連線數
    poolConfig.setMinIdle(0);    // 最多的空閒連線數
    poolConfig.setMaxIdle(8);    // 當連線池資源耗盡時,呼叫者最大阻塞的時間,超時時丟擲異常 單位:毫秒數
    poolConfig.setMaxWaitMillis(-1);    // 連線池存放池化物件方式,true放在空閒佇列最前面,false放在空閒佇列最後
    poolConfig.setLifo(true);    // 連線空閒的最小時間,達到此值後空閒連線可能會被移除,預設即為30分鐘
    poolConfig.setMinEvictableIdleTimeMillis(1000L * 60L * 30L);    // 連線耗盡時是否阻塞,預設為true
    poolConfig.setBlockWhenExhausted(true);    // 連線池建立
    objectPool = new GenericObjectPool<>(new HelloWorldFactory(), poolConfig);
  }  /**
   * 從連線池獲取物件
   */
  private static HelloWorldClientSingle borrowObject(){    try {
      HelloWorldClientSingle clientSingle = objectPool.borrowObject();
      System.out.println("總建立執行緒數"+objectPool.getCreatedCount());      return clientSingle;
    } catch (Exception e) {
      e.printStackTrace();
    }    //連線池失敗則主動建立
    return createClient();
  }  /**
   * 當連線池異常,則主動建立物件
   */
  private static HelloWorldClientSingle createClient(){    return new HelloWorldClientSingle("127.0.0.1", 55001);
  }  /**
   * 執行器
   * @param workCallBack 主要服務內容
   */
  public static Runnable execute(WorkCallBack<HelloWorldClientSingle> workCallBack){    return () -> {
      HelloWorldClientSingle client = borrowObject();      try {
       workCallBack.callback(client);
      } finally {        /** 將連線物件返回給連線池 */
        objectPool.returnObject(client);
      }
    };
  }

}

對於客戶端來說,因為存根是不能共享的,所以存根要放在呼叫函式中例項化,不要擔心效能問題,這個操作就是建立一個物件.greeterBlockingStub = GreeterGrpc.newBlockingStub(channel).withCompression("gzip");該程式碼從之前的建構函式中移位到了greet中.

public class HelloWorldClientSingle {    private final ManagedChannel channel; //一個gRPC通道
    private GreeterGrpc.GreeterBlockingStub greeterBlockingStub;//阻塞/同步 存根

   //初始化通道和存根
    public HelloWorldClientSingle(String host,int port){
      channel = ManagedChannelBuilder.forAddress(host, port).usePlaintext(true).build();
    }    public void shutdown() throws InterruptedException {
        channel.shutdown().awaitTermination(5, TimeUnit.SECONDS);
    }    //客戶端方法
    public  void greet(String name){        //需要用到存根時建立,不可複用
        greeterBlockingStub = GreeterGrpc.newBlockingStub(channel).withCompression("gzip");
        HelloRequest request = HelloRequest.newBuilder().setName(name).build();
        HelloReply response;        try {
            response = greeterBlockingStub.sayHello(request);
        } catch (StatusRuntimeException e) {
            System.out.println("RPC呼叫失敗:"+e.getMessage());            return;
        }
        System.out.println("伺服器返回資訊:"+response.getMessage());
    }
}

寫一個簡單的測試類

public class HelloWorldClientTest {

  public static void main(String[] args) {    for (int i = 0; i < 100; i++) {      new Thread( HelloWorldClientPool.execute(clientSingle -> {
          clientSingle.greet("world");
      })).start();
    }
  }
}

因為我限定的8個客戶端,所以最高也就8個channel再跑了.

Paste_Image.png

作者：此博廢棄_更新在個人部落格
連結：