package jnpf.base.service.impl;

import jnpf.base.service.LoginSessionService;
import jnpf.config.ConfigValueUtil;
import jnpf.database.util.DynamicDataSourceUtil;
import jnpf.database.util.TenantDataSourceUtil;
import jnpf.util.DateUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import java.util.Collections;
import java.util.Set;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;

/**
 * Closes sessions after their last synchronized token expiry time.
 */
@Component
@Slf4j
@RequiredArgsConstructor
public class LoginSessionExpirationWorker implements InitializingBean, DisposableBean {

    private static final String TENANT_REGISTRY_KEY = "jnpf:login-session:tenants";

    private final LoginSessionService loginSessionService;
    private final ConfigValueUtil configValueUtil;
    private final RedisTemplate<String, Object> redisTemplate;
    private ScheduledExecutorService executor;

    @Override
    public void afterPropertiesSet() {
        ThreadFactory factory = runnable -> {
            Thread thread = new Thread(runnable, "login-session-expiration");
            thread.setDaemon(true);
            return thread;
        };
        executor = Executors.newSingleThreadScheduledExecutor(factory);
        executor.scheduleWithFixedDelay(this::closeExpired, 60, 60, TimeUnit.SECONDS);
    }

    private void closeExpired() {
        if (!configValueUtil.isMultiTenancy()) {
            closeCurrentTenant();
            return;
        }
        Set<Object> tenantIds = redisTemplate.opsForSet().members(TENANT_REGISTRY_KEY);
        for (Object tenantId : tenantIds == null ? Collections.emptySet() : tenantIds) {
            try {
                TenantDataSourceUtil.switchTenant(String.valueOf(tenantId));
                closeCurrentTenant();
            } catch (Exception e) {
                log.warn("登录会话超时回收失败，tenantId: {}", tenantId, e);
            } finally {
                DynamicDataSourceUtil.clearSwitchDataSource();
            }
        }
    }

    private void closeCurrentTenant() {
        try {
            loginSessionService.closeExpired(DateUtil.dateAddMinutes(DateUtil.getNowDate(), -2), 200);
        } catch (Exception e) {
            log.warn("登录会话超时回收失败", e);
        }
    }

    @Override
    public void destroy() {
        if (executor != null) {
            executor.shutdownNow();
        }
    }
}