package com.kangaroohy.milo.configuration;

import com.kangaroohy.milo.pool.MiloConnectFactory;
import com.kangaroohy.milo.pool.MiloConnectPool;
import com.kangaroohy.milo.service.MiloService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.pool2.impl.GenericKeyedObjectPoolConfig;
import org.eclipse.milo.opcua.sdk.client.OpcUaClient;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.DependsOn;

import javax.annotation.PreDestroy;

/**
 * @author mes
 * @date 2020/4/25
 * @desc milo-spring-boot-starter
 * @since 0.0.1
 */
@Configuration
@EnableConfigurationProperties(MiloProperties.class)
@ConditionalOnClass({MiloService.class, MiloConnectPool.class})
@ConditionalOnProperty(prefix = MiloProperties.PREFIX, value = "enabled", havingValue = "true", matchIfMissing = true)
@Slf4j
public class MiloAutoConfiguration {
    private final MiloProperties properties;

    private MiloConnectPool connectPool;

    public MiloAutoConfiguration(MiloProperties properties) {
        this.properties = properties;
    }

    @Bean(name = "miloConnectPool")
    @ConditionalOnMissingBean({MiloConnectPool.class})
    protected MiloConnectPool miloConnectPool() {
        MiloConnectFactory objectFactory = new MiloConnectFactory(this.properties);
        //设置对象池的相关参数
        GenericKeyedObjectPoolConfig<OpcUaClient> poolConfig = new GenericKeyedObjectPoolConfig<>();

        MiloProperties.Pool pool = properties.getPool();
        // 最大空闲数
        poolConfig.setMaxIdlePerKey(pool.getMaxIdle());
        //最小空闲,设置为2表示池内至少存放2个空闲对象(当池内有2个空闲对象时调用borrowObject去对象时会立即调用创建对象的方法保证池内有2个空闲对象)
        poolConfig.setMinIdlePerKey(pool.getMinIdle());
        //最大总数 10
        poolConfig.setMaxTotal(pool.getMaxTotal());
        // 多久执行一次对象扫描，将无用的对象销毁，默认-1不扫描
        // poolConfig.setTimeBetweenEvictionRuns(Duration.ofMinutes(1));
        // 在获取对象的时候检查有效性, 默认false
        poolConfig.setTestOnBorrow(true);
        // 在归还对象的时候检查有效性, 默认false
        poolConfig.setTestOnReturn(false);
        // 在空闲时检查有效性, 默认false
        poolConfig.setTestWhileIdle(false);
        // 最大等待时间， 默认的值为-1，表示无限等待。
//        poolConfig.setMaxWait(Duration.ofSeconds(1));
        // 是否启用后进先出, 默认true
        poolConfig.setLifo(true);
        // 连接耗尽时是否阻塞, false立即抛异常,true阻塞直到超时, 默认true
        poolConfig.setBlockWhenExhausted(true);
        // 每次逐出检查时 逐出的最大数目 默认3
        poolConfig.setNumTestsPerEvictionRun(3);

        //一定要关闭jmx，不然springboot启动会报已经注册了某个jmx的错误
        poolConfig.setJmxEnabled(false);

        //新建一个对象池,传入对象工厂和配置
        connectPool = new MiloConnectPool(objectFactory, poolConfig);

        initPool(pool.getInitialSize(), pool.getMaxIdle());
        return connectPool;
    }

    @Bean
    @ConditionalOnMissingBean(MiloService.class)
    @DependsOn("miloConnectPool")
    public MiloService miloService(MiloConnectPool miloConnectPool) {
        return new MiloService(miloConnectPool, properties);
    }

    /**
     * 预先加载testObject对象到对象池中
     *
     * @param initialSize 初始化连接数
     * @param maxIdle     最大空闲连接数
     */
    private void initPool(int initialSize, int maxIdle) {
        if (initialSize <= 0) {
            return;
        }

        properties.getConfig().forEach((key, config) -> {
            for (int i = 0; i < Math.min(initialSize, maxIdle); i++) {
                try {
                    connectPool.addObject(config);
                } catch (Exception e) {
                    throw new RuntimeException(e);
                }
            }
        });

    }

    @PreDestroy
    public void destroy() {
        if (connectPool != null) {
            connectPool.close();
            log.info("all opcUaClients are closed");
        }
    }
}