package com.yami.trading;


import com.yami.trading.common.manager.email.EmailMessage;
import com.yami.trading.common.manager.email.EmailMessageQueue;
import com.yami.trading.common.util.OffLineEventRejectExecutingHandler;
import com.yami.trading.common.util.RejectExecutionHandlerDelegator;
import com.yami.trading.common.util.ThreadUtils;
import com.yami.trading.service.InternalEmailSenderService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.task.TaskExecutor;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.ArrayList;
import java.util.Collection;
import java.util.concurrent.RejectedExecutionHandler;

/**
 * 邮件服务类，负责从短信消息队列取出短信消息并发送
 */
public class EmailServer implements Runnable , InitializingBean {

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


    @Autowired
    private InternalEmailSenderService internalEmailSenderService;
    private ThreadPoolTaskExecutor taskExecutor;


    public EmailServer(){
        taskExecutor = new ThreadPoolTaskExecutor();
        taskExecutor.setCorePoolSize(2);
        taskExecutor.setKeepAliveSeconds(60);
        taskExecutor.setMaxPoolSize(5);
        taskExecutor.setQueueCapacity(3000);
        taskExecutor.initialize();
        RejectExecutionHandlerDelegator rejectExecutionHandlerDelegator = new RejectExecutionHandlerDelegator();
        Collection<RejectedExecutionHandler> list=new ArrayList<>();
        list.add(new OffLineEventRejectExecutingHandler());
        rejectExecutionHandlerDelegator.setRejectExecutionHandlers(list);
        taskExecutor.setRejectedExecutionHandler(rejectExecutionHandlerDelegator);
    }







    
    /**
     * 服务运行：
     * 1. 从消息队列获取message
     * 2.调用currentProvider发送短信
     */
    public void run() {
        while (true) {

            try {
                EmailMessage item = EmailMessageQueue.poll();

                if (item != null) {
                	System.out.println("邮寄地址：" + item.getTomail());
                    taskExecutor.execute(new HandleRunner(item));
                }
                else {
                	/*
                	 * 限速，最多1秒20个
                	 */
                    ThreadUtils.sleep(50);
                }

            } catch (Throwable e) {
                logger.error("EmailServer taskExecutor.execute() fail", e);

            }
        }
    }

    public class HandleRunner implements Runnable {
        private EmailMessage item;

        public HandleRunner(EmailMessage item) {
            this.item = item;
        }

        public void run() {
            try {
            	internalEmailSenderService.send(item);
            } catch (Throwable t) {
            	 logger.error("EmailServer taskExecutor.execute() fail", t);
            }
           
        }

    }


    public void afterPropertiesSet() throws Exception {

        new Thread(this, "EmailServer").start();
        if (logger.isInfoEnabled()) {
            logger.info("启动邮件发送服务！");
        }

    }
	public void setInternalEmailSenderService(InternalEmailSenderService internalEmailSenderService) {
		this.internalEmailSenderService = internalEmailSenderService;
	}

}