Liu Haoyu / viid-collector

  package com.objecteye.handle;
  

  import com.objecteye.content.ViidCollectorRedisKey;

  import com.objecteye.handle.viid.CommonHandle;

  import lombok.RequiredArgsConstructor;

  import org.springframework.beans.factory.DisposableBean;
  import org.springframework.boot.ApplicationArguments;
  import org.springframework.boot.ApplicationRunner;
  import org.springframework.data.redis.core.Cursor;
  import org.springframework.data.redis.core.ScanOptions;
  import org.springframework.data.redis.core.StringRedisTemplate;
  import org.springframework.stereotype.Component;
  
  import java.util.Map;
  import java.util.function.Consumer;
  
  /**
   * @author: liuhaoyu
   * @date: 2023/6/30
   */
  @Component
  @RequiredArgsConstructor
  public class StartUpAndShutdown implements DisposableBean, ApplicationRunner {

      private final CommonHandle commonHandle;

      private final StringRedisTemplate stringRedisTemplate;

  
      @Override
      public void destroy() throws Exception {

          allCacheHandle(commonHandle::unregister);

      }
  
      @Override
      public void run(ApplicationArguments args) throws Exception {

          allCacheHandle(commonHandle::register);

      }
  

      private void allCacheHandle(Consumer<String> func) {

          try (Cursor<Map.Entry<Object, Object>> cursor = stringRedisTemplate.boundHashOps(ViidCollectorRedisKey.DEVICE_CACHE)
                  .scan(ScanOptions.scanOptions().build())) {
              while (cursor.hasNext()) {
                  Map.Entry<Object, Object> entry = cursor.next();

                  func.accept(entry.getKey().toString());

              }
          }
      }
  }