urtrator/timer/timer.go

// URTator - Urban Terror server browser and game launcher, written in
// Go.
//
// Copyright (c) 2016, Stanslav N. a.k.a pztrn (or p0z1tr0n)
// All rights reserved.
//
// Licensed under Terms and Conditions of GNU General Public License
// version 3 or any higher.
// ToDo: put full text of license here.
package timer

import (
    // stdlib
    "errors"
    "fmt"
    "strconv"
    "sync"
    "time"
)

type Timer struct {
    // Tasks.
    tasks map[string]*TimerTask
    // Tasks map mutex.
    tasksMutex sync.Mutex
}

func (t *Timer) AddTask(task *TimerTask) error {
    fmt.Println("Adding task '" + task.Name + "'...")

    _, ok := t.tasks[task.Name]
    if ok {
        error_text := "Task '" + task.Name + "' already exist! Ignoring..."
        fmt.Println(error_text)
        return errors.New(error_text)
    }

    task.InProgress = false

    curtime := time.Now()
    nextlaunch := curtime.Add(time.Duration(task.Timeout) * time.Second)
    task.NextLaunch = nextlaunch

    t.tasksMutex.Lock()
    t.tasks[task.Name] = task
    t.tasksMutex.Unlock()

    fmt.Println("Added task '" + task.Name + "' with " + strconv.Itoa(task.Timeout) + " seconds timeout")
    return nil
}

func (t *Timer) executeTasks() {
    t.tasksMutex.Lock()
    for task_name, task := range t.tasks {
        // Check if task should be run.
        curtime := time.Now()
        diff := curtime.Sub(task.NextLaunch)
        //fmt.Println(diff)
        if diff > 0 {
            fmt.Println("Checking task '" + task_name + "'...")
            // Check if task is already running.
            if task.InProgress {
                fmt.Println("Already executing, skipping...")
                continue
            }

            fmt.Println("Launching task '" + task_name + "'...")
            task.InProgress = true
            Eventer.LaunchEvent(task.Callee, map[string]string{})

            curtime = time.Now()
            nextlaunch := curtime.Add(time.Duration(task.Timeout) * time.Second)
            task.NextLaunch = nextlaunch
        }
    }
    t.tasksMutex.Unlock()
}

func (t *Timer) GetTaskStatus(task_name string) bool {
    t.tasksMutex.Lock()
    task, ok := t.tasks[task_name]
    t.tasksMutex.Unlock()
    if !ok {
        return false
    }

    return task.InProgress
}

func (t *Timer) Initialize() {
    fmt.Println("Initializing timer...")

    t.initializeStorage()

    ticker := time.NewTicker(time.Second * 1)
    go func() {
        for _ = range ticker.C {
            go t.executeTasks()
        }
    }()

    Eventer.AddEventHandler("taskDone", t.SetTaskNotInProgress)
}

func (t *Timer) initializeStorage() {
    t.tasks = make(map[string]*TimerTask)
}

func (t *Timer) RemoveTask(task_name string) {
    t.tasksMutex.Lock()
    _, ok := t.tasks[task_name]
    t.tasksMutex.Unlock()
    if !ok {
        return
    }

    t.tasksMutex.Lock()
    delete(t.tasks, task_name)
    t.tasksMutex.Unlock()
}

func (t *Timer) SetTaskNotInProgress(data map[string]string) {
    t.tasksMutex.Lock()
    _, ok := t.tasks[data["task_name"]]
    if !ok {
        return
    }

    t.tasks[data["task_name"]].InProgress = false
    t.tasksMutex.Unlock()
}