package com.objecteye.handle;

import com.alibaba.nacos.common.utils.ExceptionUtil;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.objecteye.content.ViidCollectorRedisKey;
import com.objecteye.handle.viid.VIIDCommonHandle;
import com.objecteye.pojo.platform.DeviceRequestParams;
import lombok.RequiredArgsConstructor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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 VIIDCommonHandle viidCommonHandle;
    private final StringRedisTemplate stringRedisTemplate;
    private final ObjectMapper objectMapper;

    private final Logger logger = LoggerFactory.getLogger(StartUpAndShutdown.class);

    @Override
    public void destroy() throws Exception {
        allCacheHandle(viidCommonHandle::unregister);
    }

    @Override
    public void run(ApplicationArguments args) throws Exception {
        allCacheHandle(viidCommonHandle::register);
    }

    private void allCacheHandle(Consumer<DeviceRequestParams> 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();
                try {
                    func.accept(objectMapper.readValue(entry.getValue().toString(), DeviceRequestParams.class));
                } catch (Exception e) {
                    logger.error(ExceptionUtil.getAllExceptionMsg(e));
                }
            }
        }
    }
}