urtrator/timer/timer.go

// URTrator - Urban Terror server browser and game launcher, written in
// Go.
//
// Copyright (c) 2016-2018, Stanslav N. a.k.a pztrn (or p0z1tr0n) and
// URTrator contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject
// to the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
// IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
// CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
// TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
// OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

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()
	Eventer.AddEventHandler("taskDone", t.SetTaskNotInProgress)

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

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()
}