com.boot.auth.starter.service.impl.DefaultCacheServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-webflux-auth-starter Show documentation
基于springboot webflux的权限验证
The newest version!
package com.boot.auth.starter.service.impl;

import com.boot.auth.starter.common.AuthConstant;
import com.boot.auth.starter.common.AuthProperties;
import com.boot.auth.starter.common.RestStatus;
import com.boot.auth.starter.exception.AuthException;
import com.boot.auth.starter.service.CacheService;
import com.boot.auth.starter.support.GuavaCacheSupport;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;

/**
 * 在使用本类时，需要注意以下问题:
 * 1、若使用 Cache 而非 LoadingCache ，可以直接使用本类，无需任何改动
 * 2、使用 LoadingCache 时，需要自行在 GuavaCacheSupport 中调用 setCacheLoader 后才能使用该类其他方法
 * 3、GuavaCacheSupport 中的 setRemovalListener 是可选的
 * 4、在 Guava 中没有直接实现 expire ，只是判断了是否存在，请自行实现
 * 5、exclude 选项不是必须的
 * 6、若本类中有任何方法不适用，请自行复写它。若先使用其他缓存，请直接复写全部方法
 */
@Component
public class DefaultCacheServiceImpl implements CacheService {
    private final static Logger log = LoggerFactory.getLogger(DefaultCacheServiceImpl.class);
    final
    GuavaCacheSupport guavaCacheSupport;
    AuthProperties authProperties;

    public DefaultCacheServiceImpl(GuavaCacheSupport guavaCacheSupport,
                                   AuthProperties authProperties) {
        this.guavaCacheSupport = guavaCacheSupport;
        this.authProperties = authProperties;
    }

    @Override
    public Long getOverdueTime() {
        return authProperties.getOverdueTime();
    }

    @Override
    public Boolean getExclude() {
        return authProperties.getEnableExclude();
    }

    @Override
    public void put(String key, String data) {
        if (authProperties.getGuavaCache().getEnableLoadingCache()) {
            guavaCacheSupport.getLoadingCache().put(key, data);
        } else {
            guavaCacheSupport.getCache().put(key, data);
        }
    }

    @Override
    public void put(String key, String data, Long overdueTime) {
        this.put(key, data);
    }

    @Override
    public String get(String key) {
        if (authProperties.getEnableExclude()
                && !this.exclude(key)) {
            return null;
        }
        Object obj = null;
        if (authProperties.getGuavaCache().getEnableLoadingCache()) {
            try {
                obj = guavaCacheSupport.getLoadingCache().get(key);
            } catch (ExecutionException e) {
                log.error("LoadingCache error key:{}", key, e);
            }
        } else {
            obj = guavaCacheSupport.getCache().getIfPresent(key);
        }
        if (obj == null) return null;
        return obj.toString();
    }

    @Override
    public String get(String key, Callable