Java本地緩存工具之LoadingCache的使用詳解

        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>30.1.1-jre</version>
        </dependency>
        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>5.5.2</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>

package ai.guiji.csdn.tools;
 
import cn.hutool.core.thread.ThreadUtil;
import com.google.common.cache.*;
import lombok.extern.slf4j.Slf4j;
 
import java.text.MessageFormat;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.stream.LongStream;
 
/** @Author 劍客阿良_ALiang @Date 2021/12/30 17:57 @Description: 緩存工具 */
@Slf4j
public class CacheUtils {
 
  private static LoadingCache<Long, String> cache;
 
  /**
   * 初始化緩存方法
   *
   * @param totleCount 緩存池上限
   * @param overtime 超時時間
   * @param unit 時間單位
   * @param handleNotExist 處理不存在key方法
   * @param handleRemove 移除主鍵消費
   */
  private static void initCache(
      Integer totleCount,
      Integer overtime,
      TimeUnit unit,
      Function<Long, String> handleNotExist,
      Consumer<Long> handleRemove) {
    cache =
        CacheBuilder.newBuilder()
            // 緩存池大小
            .maximumSize(totleCount)
            // 設置時間對象沒有被讀/寫訪問則對象從內存中刪除
            .expireAfterWrite(overtime, unit)
            // 移除監聽器
            .removalListener(
                new RemovalListener<Long, String>() {
                  @Override
                  public void onRemoval(RemovalNotification<Long, String> rn) {
                    handleRemove.accept(rn.getKey());
                  }
                })
            .recordStats()
            .build(
                new CacheLoader<Long, String>() {
                  @Override
                  public String load(Long aLong) throws Exception {
                    return handleNotExist.apply(aLong);
                  }
                });
    log.info("初始化緩存");
  }
 
  /**
   * 存入緩存
   *
   * @param key 鍵
   * @param value 值
   */
  public static void put(Long key, String value) {
    try {
      log.info("緩存存入:[{}]-[{}]", key, value);
      cache.put(key, value);
    } catch (Exception exception) {
      log.error("存入緩存異常", exception);
    }
  }
 
  /**
   * 批量存入緩存
   *
   * @param map 映射
   */
  public static void putMap(Map<Long, String> map) {
    try {
      log.info("批量緩存存入:[{}]", map);
      cache.putAll(map);
    } catch (Exception exception) {
      log.error("批量存入緩存異常", exception);
    }
  }
 
  /**
   * 獲取緩存
   *
   * @param key 鍵
   */
  public static String get(Long key) {
    try {
      return cache.get(key);
    } catch (Exception exception) {
      log.error("獲取緩存異常", exception);
      return null;
    }
  }
 
  /**
   * 刪除緩存
   *
   * @param key 鍵
   */
  public static void removeKey(Long key) {
    try {
      cache.invalidate(key);
    } catch (Exception exception) {
      log.error("刪除緩存異常", exception);
    }
  }
 
  /**
   * 批量刪除緩存
   *
   * @param keys 鍵
   */
  public static void removeAll(Iterable<Long> keys) {
    try {
      cache.invalidateAll(keys);
    } catch (Exception exception) {
      log.error("批量刪除緩存異常", exception);
    }
  }
 
  /** 清理緩存 */
  public static void clear() {
    try {
      cache.invalidateAll();
    } catch (Exception exception) {
      log.error("清理緩存異常", exception);
    }
  }
 
  /**
   * 獲取緩存大小
   *
   * @return 長度
   */
  public static long size() {
    return cache.size();
  }
 
  public static void main(String[] args) {
    initCache(
        Integer.MAX_VALUE,
        10,
        TimeUnit.SECONDS,
        k -> {
          log.info("緩存:[{}]，不存在", k);
          return "";
        },
        x -> log.info("緩存:[{}]，已經移除", x));
    System.out.println(size());
    LongStream.range(0, 10).forEach(a -> put(a, MessageFormat.format("tt-{0}", a)));
    System.out.println(cache.asMap());
    ThreadUtil.sleep(5000);
    LongStream.range(0, 10)
        .forEach(
            a -> {
              System.out.println(get(a));
              ThreadUtil.sleep(1000);
            });
    System.out.println(cache.asMap());
    ThreadUtil.sleep(10000);
    System.out.println(cache.asMap());
  }
}