com.voa.weixin.task.TaskManager.java Source code

Introduction

Here is the source code for com.voa.weixin.task.TaskManager.java

Source

/**
 This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
    
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
    
You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.voa.weixin.task;

import java.util.Stack;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.voa.weixin.Carp;

/**
 * ???
 * @author zhaiyuxin
 *
 */
public class TaskManager {

    private static final Log logger = LogFactory.getLog(TaskManager.class);

    /** ?token **/
    private Boolean isGetToken;
    private static TaskManager taskManager;
    /** ???? **/
    private final ExecutorService pool;
    /** ???? **/
    private final Stack<Task> pauseTasks;

    private TaskManager() {
        pool = Executors.newFixedThreadPool(30);
        pauseTasks = new Stack<Task>();
        isGetToken = false;
    }

    public static TaskManager getInstance() {
        if (taskManager == null)
            taskManager = new TaskManager();

        return taskManager;
    }

    /**
     * ?
     * @param task ??
     */
    public void addTask(Task task) {
        addTask(task, false);
    }

    /**
     * ????
     * @param task ??
     * @param isSynchronized ?false
     */
    public void addTask(Task task, boolean isSynchronized) {
        synchronized (isGetToken) {
            if (task.getName().equals("token")) {
                if (isGetToken == false) {
                    isGetToken = true;
                    executeTask(task, isSynchronized);
                }
            } else {
                if (isGetToken || Carp.token == null) {
                    pauseTasks.add(task);
                } else {
                    executeTask(task, isSynchronized);
                }
            }
        }
    }

    private void executeTask(Task task, boolean isSynchronized) {
        if (isSynchronized)
            task.run();
        else
            pool.submit(task);
    }

    /**
     * access_token
     * @param status
     */
    public void setTokenStatus(boolean status) {
        synchronized (isGetToken) {
            isGetToken = status;
            logger.debug("modify token status  " + status);
            if (isGetToken == false && pauseTasks.size() > 0) {
                logger.debug("commit task");
                while (!pauseTasks.isEmpty()) {
                    Task task = pauseTasks.pop();
                    pool.submit(task);
                }
            }
        }
    }

    public void destroy() {
        pool.shutdown();
    }
}