com.alibaba.dashscope.protocol.ClientProviders Maven / Gradle / Ivy

Go to download
// Copyright (c) Alibaba, Inc. and its affiliates.

package com.alibaba.dashscope.protocol;

import com.alibaba.dashscope.protocol.okhttp.OkHttpClientFactory;
import com.alibaba.dashscope.protocol.okhttp.OkHttpHttpClient;
import com.alibaba.dashscope.protocol.okhttp.OkHttpWebSocketClient;

public class ClientProviders {
  public static HalfDuplexClient getHalfDuplexClient() {
    return getHalfDuplexClient(null);
  }

  public static FullDuplexClient getFullDuplexClient() {
    return getFullDuplexClient(null);
  }

  /**
   * Create a dashscope half duplex client. only okhttp http and websocket is supported.
   *
   * @param options The client options.
   * @return The half duplex client.
   */
  public static HalfDuplexClient getHalfDuplexClient(ClientOptions options) {
    if (options == null) {
      // create default config client, create default http client.
      return new OkHttpHttpClient(OkHttpClientFactory.getOkHttpClient());
    } else {
      if (options.getProtocol().toLowerCase().startsWith("http")) {
        return new OkHttpHttpClient(OkHttpClientFactory.getNewOkHttpClient(options));
      } else {
        return new OkHttpWebSocketClient(OkHttpClientFactory.getNewOkHttpClient(options));
      }
    }
  }

  /**
   * Create a dashscope full duplex client. only okhttp http and websocket is supported.
   *
   * @param options The client options.
   * @return The full duplex client.
   */
  public static FullDuplexClient getFullDuplexClient(ClientOptions options) {
    if (options == null) {
      // create default config client, create default http client.
      return new OkHttpWebSocketClient(OkHttpClientFactory.getOkHttpClient());
    } else {
      return new OkHttpWebSocketClient(OkHttpClientFactory.getNewOkHttpClient(options));
    }
  }
}