scheduler.h

Go to the documentation of this file.

/* This file is part of the KDE project

   Copyright (C) 2005 Dario Massarin <nekkar@libero.it>
   Coypright (C) 2010 Matthias Fuchs <mat69@gmx.net>

   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 2 of the License, or (at your option) any later version.
*/

#ifndef SCHEDULER_H
#define SCHEDULER_H

#include <QObject>
#include <QMap>
#include <QTimerEvent>

#include "core/job.h"
#include "core/jobqueue.h"
#include "../kget_export.h"

class KGET_EXPORT Scheduler : public QObject
{
    Q_OBJECT

    friend class SchedulerTest;

    public:

        enum FailureStatus {
            None         = 0,
            AboutToStall = 1,
            Stall        = 2,
            StallTimeout = 3,
            Abort        = 4,
            AbortTimeout = 5,
            Error = 6
        };
        
        class JobFailure {
            public:
            JobFailure()
                : status(None), time(-1), count(0)
            {}
            
            bool isValid()                  {return ((status != None) && (time != -1));}
            
            FailureStatus status;
            int time;
            int count;
            
            bool operator==(JobFailure f) const {return ((status == f.status) && (time == f.time));}
            bool operator!=(JobFailure f) const {return ((status != f.status) || (time != f.time));}            
        };
        
        Scheduler(QObject * parent = 0);
        ~Scheduler();

        void start();

        void stop();

        void setIsSuspended(bool isSuspended);

        void setHasNetworkConnection(bool hasConnection);

        void addQueue(JobQueue * queue);

        void delQueue(JobQueue * queue);

        bool hasRunningJobs() const;

        int countRunningJobs() const;
        
        void settingsChanged();

        //JobQueue notifications
        virtual void jobQueueChangedEvent(JobQueue * queue, JobQueue::Status status);
        virtual void jobQueueMovedJobEvent(JobQueue * queue, Job * job);
        virtual void jobQueueAddedJobEvent(JobQueue * queue, Job * job);
        virtual void jobQueueAddedJobsEvent(JobQueue *queue, const QList<Job*> jobs);
        virtual void jobQueueRemovedJobEvent(JobQueue * queue, Job * job);
        virtual void jobQueueRemovedJobsEvent(JobQueue *queue, const QList<Job*> jobs);

        //Job notifications
        virtual void jobChangedEvent(Job * job, Job::Status status);
        virtual void jobChangedEvent(Job * job, Job::Policy status);
        virtual void jobChangedEvent(Job * job, JobFailure failure);

    protected:
        void updateQueue( JobQueue * queue );

        bool shouldBeRunning( Job * job );

    private:
        //Virtual QObject method
        void timerEvent(QTimerEvent * event);

        void updateAllQueues();

        bool shouldUpdate() const;

    private:
        QList<JobQueue *> m_queues;
        QMap<Job *, JobFailure> m_failedJobs;

        int m_failureCheckTimer;

        const int m_stallTime;
        int m_stallTimeout;
        int m_abortTimeout;
        bool m_isSuspended;
        bool m_hasConnection;
};

inline bool Scheduler::shouldUpdate() const
{
    return !m_isSuspended && m_hasConnection;
}
#endif