com.alogic.kube.xscript.api.KubeClient Maven / Gradle / Ivy

Go to download
package com.alogic.kube.xscript.api;

import com.alogic.kube.KubeApiClients;
import com.alogic.xscript.ExecuteWatcher;
import com.alogic.xscript.Logiclet;
import com.alogic.xscript.LogicletContext;
import com.alogic.xscript.doc.XsObject;
import com.alogic.xscript.plugins.Segment;
import com.anysoft.util.Properties;
import com.anysoft.util.PropertiesConstants;
import io.kubernetes.client.openapi.ApiClient;
import io.kubernetes.client.util.ClientBuilder;
import io.kubernetes.client.util.KubeConfig;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.io.FileReader;

/**
 * Kube的api客户端
 *
 * @since 1.6.16.14
 */
public class KubeClient extends Segment {
    /**
     * Response对象的上下文id
     */
    protected String cid = KubeUtil.DFT_CONTEXT_ID;

    /**
     * 指定客户端的配置文件
     */
    protected String $config = KubeUtil.DFT_KUBE_CONFIG;

    /**
     * 客户端ID
     */
    protected String $kubeClientId = null;

    public KubeClient(String tag, Logiclet p) {
        super(tag, p);

        registerModule("kube-list", KubeList.class);
        registerModule("kube-get",KubeGet.class);
        registerModule("kube-delete",KubeUpdate.Delete.class);
        registerModule("kube-post", KubeUpdate.PostByJson.class);
        registerModule("kube-post-json",KubeUpdate.PostByJson.class);
        registerModule("kube-post-text",KubeUpdate.PostByText.class);
        registerModule("kube-put",KubeUpdate.PutByJson.class);
        registerModule("kube-put-json",KubeUpdate.PutByJson.class);
        registerModule("kube-put-text",KubeUpdate.PutByText.class);
        registerModule("kube-patch",KubeUpdate.PatchByJson.class);
        registerModule("kube-patch-json",KubeUpdate.PatchByJson.class);
        registerModule("kube-patch-text",KubeUpdate.PatchByText.class);
    }

    @Override
    public void configure(Properties p){
        super.configure(p);
        cid = PropertiesConstants.getString(p, "cid", cid,true);
        $kubeClientId = PropertiesConstants.getRaw(p,"kubeId",$kubeClientId);
        $config = PropertiesConstants.getRaw(p,"config",$config);
    }

    @Override
    protected void onExecute(XsObject root, XsObject current, LogicletContext ctx, ExecuteWatcher watcher) {
        try {
            ApiClient apiClient = null;

            String kubeClientId = PropertiesConstants.transform(ctx,$kubeClientId,null);
            if (StringUtils.isNotEmpty(kubeClientId)){
                apiClient = KubeApiClients.getClient(kubeClientId,false);
            }

            if (apiClient == null){
                String config = PropertiesConstants.transform(ctx, $config, KubeUtil.DFT_KUBE_CONFIG);
                File file = new File(config);
                if (file.exists() && file.isFile() && file.canRead()) {
                    apiClient = ClientBuilder.kubeconfig(
                            KubeConfig.loadKubeConfig(new FileReader(file))
                    ).build();
                    log("Load kube config from file:" + config);
                } else {
                    log("Load kube config from cluster.");
                    apiClient = ClientBuilder.oldCluster().build();
                }
                if (StringUtils.isNotEmpty(kubeClientId)){
                    KubeApiClients.registryClient(kubeClientId,apiClient);
                }
            }

            if (apiClient != null) {
                try {
                    ctx.setObject(cid, apiClient);
                    super.onExecute(root,current,ctx,watcher);
                } finally {
                    ctx.removeObject(cid);
                }
            }else{
                log("Cannot init kube api client:" + $config);
            }
        }catch (Exception ex){
            log("Cannot init kube api client:" + ex.getMessage());
        }
    }
}