com.github.fartherp.shiro.ClearCache Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2019. CK. All rights reserved.
 */

package com.github.fartherp.shiro;

import io.netty.util.HashedWheelTimer;
import io.netty.util.Timeout;
import io.netty.util.TimerTask;
import org.apache.shiro.cache.Cache;
import org.redisson.api.RScoredSortedSet;
import org.redisson.client.protocol.ScoredEntry;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.TimeUnit;

/**
 * Created by IntelliJ IDEA.
 *
 * @author: CK
 * @date: 2019/1/4
 */
public class ClearCache implements TimerTask {
    private HashedWheelTimer hashedWheelTimer = new HashedWheelTimer();

    private RedisSessionDAO redisSessionDAO;

    public ClearCache(RedisSessionDAO redisSessionDAO) {
        this.redisSessionDAO = redisSessionDAO;
    }

    public void init() {
        hashedWheelTimer.newTimeout(this, redisSessionDAO.getRedisCacheManager().getTtl(), TimeUnit.MINUTES);
    }

    public void clearSession() {
        RScoredSortedSet sessionKeys = redisSessionDAO.getSessionKeys();
        List> keys = (List>) sessionKeys.entryRange(0, false, new Date().getTime(), true);
        List destroyKeys = new ArrayList<>();
        for (ScoredEntry key : keys) {
            destroyKeys.add(key.getValue());
        }
        if (destroyKeys.size() > 0) {
            sessionKeys.removeAll(destroyKeys);
        }
    }

    public void clearCache() {
        ConcurrentMap caches = redisSessionDAO.getRedisCacheManager().getCaches();
        caches.forEach((k, v) -> {
            RedisCache redisCache = (RedisCache) v;
            RScoredSortedSet cacheKeys = redisCache.getCacheKeys();
            List keys = (List) cacheKeys.entryRange(0, false, new Date().getTime(), true);
            List