org.b3log.latke.taskqueue.local.LocalTaskRunner Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of latke Show documentation
There is a newer version: 2.4.33
/*
 * Copyright (c) 2009-2016, b3log.org & hacpai.com
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.b3log.latke.taskqueue.local;

import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import org.b3log.latke.Latkes;
import org.b3log.latke.logging.Level;
import org.b3log.latke.logging.Logger;
import org.b3log.latke.taskqueue.Task;
import org.b3log.latke.urlfetch.HTTPRequest;
import org.b3log.latke.urlfetch.HTTPResponse;
import org.b3log.latke.urlfetch.URLFetchService;
import org.b3log.latke.urlfetch.URLFetchServiceFactory;

/**
 * run the task in queue, now using httpUrlfetch to handle the request.
 *
 * @author Love Yao
 * @author Liang Ding
 * @version 1.0.0.2, Sep 20, 2012
 */
public final class LocalTaskRunner extends Thread {

    /**
     * Logger.
     */
    private static final Logger LOGGER = Logger.getLogger(LocalTaskRunner.class.getName());

    /**
     * the task need to do .
     */
    private Task task;

    /**
     * the retry time of a task.
     */
    private Integer retryLimit;

    /**
     * default constructor.
     *
     * @param task task need to do
     * @param retryLimit retryTime
     */
    public LocalTaskRunner(final Task task, final Integer retryLimit) {
        this.task = task;
        this.retryLimit = retryLimit;
    }

    /**
     * using urlFetchService to do the TASK.
     */
    private URLFetchService urlFetchService;

    @Override
    public void run() {
        urlFetchService = URLFetchServiceFactory.getURLFetchService();

        final HTTPRequest httpRequest = new HTTPRequest();

        try {
            httpRequest.setURL(new URL(Latkes.getServer() + Latkes.getContextPath() + task.getURL()));
            httpRequest.setRequestMethod(task.getRequestMethod());
            httpRequest.setPayload(task.getPayload());
        } catch (final MalformedURLException e) {
            e.printStackTrace();
        }

        Integer retry = 0;

        while (retry < retryLimit) {

            if (!doUrlFetch(httpRequest)) {
                retry++;
                try {
                    Thread.sleep(new Integer("1000"));
                } catch (final InterruptedException e) {
                    e.printStackTrace();
                }
            } else {
                break;
            }
        }
    }

    /**
     * do task using urlfetch(method:get),if wrong return false.
     *
     * @param httpRequest {@link HTTPRequest}
     * @return isSuccess
     */
    private boolean doUrlFetch(final HTTPRequest httpRequest) {
        HTTPResponse httpResponse = null;

        try {
            httpResponse = urlFetchService.fetch(httpRequest);
        } catch (final IOException e) {
            LOGGER.log(Level.INFO, "The task[{0}] throw exception {1}", new Object[]{task.getURL(), e.getMessage()});
            return false;
        }

        /**
         * 
         * Quote GAE:" If a push task request handler returns an HTTP status code within the range 200–299, App Engine
         * considers the task to have completed successfully. If the task returns a status code outside of this range"
         * 
         */
        final Integer beginCode = 200;
        final Integer endCode = 299;

        if (httpResponse.getResponseCode() >= beginCode && httpResponse.getResponseCode() <= endCode) {
            return true;
        }

        LOGGER.log(Level.INFO, "The task[{0}] not success, the return code is [{1}]",
                new Object[]{task.getURL(), httpResponse.getResponseCode()});

        return false;
    }
}