ConcurrentSkipListSet实现接口权限鉴权

package cn.silence;

import org.apache.commons.lang.StringUtils;
import org.junit.Assert;
import org.junit.Test;
import org.springframework.util.CollectionUtils;

import java.util.NavigableSet;
import java.util.concurrent.ConcurrentSkipListSet;

/**
 * @author mzq
 * @date 2022/9/29 9:48
 */
public class ConcurrentSkipListSetTest {


    @Test
    public void test() {
        ConcurrentSkipListSet<String> set = new ConcurrentSkipListSet<>();
        set.add("/aa/bb/cc/ee/list");
        set.add("/aa/bb/cc/gg/sync");
        set.add("/aa/bb/cc/gg/list");
        set.add("/aa/bb/cc/gg23/sync");
        set.add("/aa/bb/cc/ef/list");
        set.add("/aa/bb/cc/stockOut/sync");
        set.add("/aa/cc/*");
        set.add("/aa/cc/11/check");
        set.add("/aa/cc/22/*");
        set.add("/aa/cc/33/*");
        set.add("/aa/cc/44/sync");
        set.forEach(System.out::println);
        System.out.println();

        //true /aa/cc/*
        havePermission("/aa/cc/44/sync/rege",set);

        //true /aa/cc/44/sync
        havePermission("/aa/cc/44/sync",set);

        //true  /aa/cc/33/*
        havePermission("/aa/cc/33/156156",set);

        //false
        havePermission("/aa/cc222222/33/156156",set);

        //true /aa/cc/33/*
        havePermission("/aa/cc/33/15156",set);

        //true /aa/cc/11/check
        havePermission("/aa/cc/11/check",set);

        //aa/cc/*
        havePermission("/aa/cc/11/1111",set);

        //aa/cc/*
        havePermission("/aa/cc",set);

        //false
        havePermission("/aa/222",set);
    }

    private boolean havePermission(String requestUri, ConcurrentSkipListSet<String> set) {
        int index = org.apache.commons.lang3.StringUtils.ordinalIndexOf(requestUri, "/", 3);
        String pre = index == -1 ? requestUri : requestUri.substring(0, index);
        NavigableSet<String> navigableSet = set.subSet(pre, true, requestUri, true);
        if (CollectionUtils.isEmpty(navigableSet)) {
            return false;
        }
        navigableSet = navigableSet.descendingSet();
        String wildcard = "/*";
        for (String uri : navigableSet) {
            if (org.apache.commons.lang3.StringUtils.equals(uri, requestUri)) {
                return true;
            }
            if (!uri.endsWith(wildcard)) {
                continue;
            }
            uri = org.apache.commons.lang3.StringUtils.substring(uri, 0, uri.length() - wildcard.length());
            if (requestUri.startsWith(uri)) {
                return true;
            }
        }
        return false;
    }

    @Test
    public void stringTest(){
        String str="/aa/cc/44/sync";
        int index = StringUtils.ordinalIndexOf(str, "/", 3);
        System.out.println("index = " + index);
        System.out.println(str.substring(0,index));
    }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Silence丶你的名字

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值