org.apache.hadoop.fs.cosn.ranger.security.token.DelegationTokenSelector Maven / Gradle / Ivy

package org.apache.hadoop.fs.cosn.ranger.security.token;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.security.token.Token;
import org.apache.hadoop.security.token.TokenIdentifier;
import org.apache.hadoop.security.token.TokenSelector;

import java.util.Collection;
import java.util.Iterator;

public class DelegationTokenSelector implements TokenSelector {

    public DelegationTokenSelector() {
    }

    public Token selectToken(Text service,
            Collection> tokens) {
        if (service == null) {
            return null;
        }

        Iterator> tokenIter = tokens.iterator();

        Token token;
        while (true) {
            if (!tokenIter.hasNext()) {
                return null;
            }

            token = (Token) tokenIter.next();
            if (DelegationTokenIdentifier.QCLOUD_OBJECT_STORAGE_DELEGATION_KIND.equals(token.getKind())) {
                return token;
            }
        }
    }
}