Yarn节点标签功能的java接口封装

最新推荐文章于 2024-07-24 17:27:32 发布

今天写代码了没

最新推荐文章于 2024-07-24 17:27:32 发布

阅读量150

点赞数

文章标签： java hadoop apache

本文链接：https://blog.csdn.net/newer1516/article/details/129924804

版权

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.yarn.api.records.*;
import org.apache.hadoop.yarn.client.ClientRMProxy;
import org.apache.hadoop.yarn.client.api.YarnClient;
import org.apache.hadoop.yarn.conf.YarnConfiguration;
import org.apache.hadoop.yarn.exceptions.YarnException;
import org.apache.hadoop.yarn.nodelabels.CommonNodeLabelsManager;
import org.apache.hadoop.yarn.server.api.ResourceManagerAdministrationProtocol;
import org.apache.hadoop.yarn.server.api.protocolrecords.AddToClusterNodeLabelsRequest;
import org.apache.hadoop.yarn.server.api.protocolrecords.RemoveFromClusterNodeLabelsRequest;
import org.apache.hadoop.yarn.server.api.protocolrecords.ReplaceLabelsOnNodeRequest;

import java.io.IOException;
import java.util.*;

/**
 * for node135、node136、node137集群
 */
public class YarnLabelTest {

    public static void main(String[] args) throws Exception {
        //创建所有标签
        createLables();
        //标签绑定节点 FIXME 节点名需要使用hostname，使用ip会失败
        replaceLabelsOnNode();
        //移除没有与队列关联的标签
        removeFromClusterNodeLabels();

        listDetailedClusterNodes();
        queueRelatedLabels();
    }

    public static int removeFromClusterNodeLabels() throws IOException, YarnException {
        Set<String> labels = new HashSet<>();
        labels.add("cpu")