Search in sources :

Example 1 with NamespacedKubernetesClient

use of io.fabric8.kubernetes.client.NamespacedKubernetesClient in project flink by apache.

the class FlinkKubeClientFactory method fromConfiguration.

/**
 * Create a Flink Kubernetes client with the given configuration.
 *
 * @param flinkConfig Flink configuration
 * @param useCase Flink Kubernetes client use case (e.g. client, resourcemanager,
 *     kubernetes-ha-services)
 * @return Return the Flink Kubernetes client with the specified configuration and dedicated IO
 *     executor.
 */
public FlinkKubeClient fromConfiguration(Configuration flinkConfig, String useCase) {
    final Config config;
    final String kubeContext = flinkConfig.getString(KubernetesConfigOptions.CONTEXT);
    if (kubeContext != null) {
        LOG.info("Configuring kubernetes client to use context {}.", kubeContext);
    }
    final String kubeConfigFile = flinkConfig.getString(KubernetesConfigOptions.KUBE_CONFIG_FILE);
    if (kubeConfigFile != null) {
        LOG.debug("Trying to load kubernetes config from file: {}.", kubeConfigFile);
        try {
            // If kubeContext is null, the default context in the kubeConfigFile will be used.
            // Note: the third parameter kubeconfigPath is optional and is set to null. It is
            // only used to rewrite
            // relative tls asset paths inside kubeconfig when a file is passed, and in the case
            // that the kubeconfig
            // references some assets via relative paths.
            config = Config.fromKubeconfig(kubeContext, FileUtils.readFileUtf8(new File(kubeConfigFile)), null);
        } catch (IOException e) {
            throw new KubernetesClientException("Load kubernetes config failed.", e);
        }
    } else {
        LOG.debug("Trying to load default kubernetes config.");
        config = Config.autoConfigure(kubeContext);
    }
    final String namespace = flinkConfig.getString(KubernetesConfigOptions.NAMESPACE);
    LOG.debug("Setting namespace of Kubernetes client to {}", namespace);
    config.setNamespace(namespace);
    final NamespacedKubernetesClient client = new DefaultKubernetesClient(config);
    final int poolSize = flinkConfig.get(KubernetesConfigOptions.KUBERNETES_CLIENT_IO_EXECUTOR_POOL_SIZE);
    return new Fabric8FlinkKubeClient(flinkConfig, client, createThreadPoolForAsyncIO(poolSize, useCase));
}
Also used : Config(io.fabric8.kubernetes.client.Config) NamespacedKubernetesClient(io.fabric8.kubernetes.client.NamespacedKubernetesClient) IOException(java.io.IOException) DefaultKubernetesClient(io.fabric8.kubernetes.client.DefaultKubernetesClient) File(java.io.File) KubernetesClientException(io.fabric8.kubernetes.client.KubernetesClientException)

Aggregations

Config (io.fabric8.kubernetes.client.Config)1 DefaultKubernetesClient (io.fabric8.kubernetes.client.DefaultKubernetesClient)1 KubernetesClientException (io.fabric8.kubernetes.client.KubernetesClientException)1 NamespacedKubernetesClient (io.fabric8.kubernetes.client.NamespacedKubernetesClient)1 File (java.io.File)1 IOException (java.io.IOException)1