gofmt.
This commit is contained in:
162
timer/timer.go
162
timer/timer.go
@@ -10,121 +10,121 @@
|
||||
package timer
|
||||
|
||||
import (
|
||||
// stdlib
|
||||
"errors"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"sync"
|
||||
"time"
|
||||
// stdlib
|
||||
"errors"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Timer struct {
|
||||
// Tasks.
|
||||
tasks map[string]*TimerTask
|
||||
// Tasks map mutex.
|
||||
tasksMutex sync.Mutex
|
||||
// Tasks.
|
||||
tasks map[string]*TimerTask
|
||||
// Tasks map mutex.
|
||||
tasksMutex sync.Mutex
|
||||
}
|
||||
|
||||
func (t *Timer) AddTask(task *TimerTask) error {
|
||||
fmt.Println("Adding task '" + task.Name + "'...")
|
||||
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)
|
||||
}
|
||||
_, 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
|
||||
task.InProgress = false
|
||||
|
||||
curtime := time.Now()
|
||||
nextlaunch := curtime.Add(time.Duration(task.Timeout) * time.Second)
|
||||
task.NextLaunch = nextlaunch
|
||||
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()
|
||||
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
|
||||
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
|
||||
}
|
||||
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{})
|
||||
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()
|
||||
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
|
||||
}
|
||||
t.tasksMutex.Lock()
|
||||
task, ok := t.tasks[task_name]
|
||||
t.tasksMutex.Unlock()
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
return task.InProgress
|
||||
return task.InProgress
|
||||
}
|
||||
|
||||
func (t *Timer) Initialize() {
|
||||
fmt.Println("Initializing timer...")
|
||||
fmt.Println("Initializing timer...")
|
||||
|
||||
t.initializeStorage()
|
||||
Eventer.AddEventHandler("taskDone", t.SetTaskNotInProgress)
|
||||
t.initializeStorage()
|
||||
Eventer.AddEventHandler("taskDone", t.SetTaskNotInProgress)
|
||||
|
||||
ticker := time.NewTicker(time.Second * 1)
|
||||
go func() {
|
||||
for _ = range ticker.C {
|
||||
go t.executeTasks()
|
||||
}
|
||||
}()
|
||||
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)
|
||||
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()
|
||||
_, ok := t.tasks[task_name]
|
||||
t.tasksMutex.Unlock()
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
t.tasksMutex.Lock()
|
||||
delete(t.tasks, task_name)
|
||||
t.tasksMutex.Unlock()
|
||||
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.tasksMutex.Lock()
|
||||
_, ok := t.tasks[data["task_name"]]
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
t.tasks[data["task_name"]].InProgress = false
|
||||
t.tasksMutex.Unlock()
|
||||
t.tasks[data["task_name"]].InProgress = false
|
||||
t.tasksMutex.Unlock()
|
||||
}
|
||||
|
@@ -10,25 +10,25 @@
|
||||
package timer
|
||||
|
||||
import (
|
||||
// stdlib
|
||||
"time"
|
||||
// stdlib
|
||||
"time"
|
||||
)
|
||||
|
||||
type TimerTask struct {
|
||||
// Task name.
|
||||
Name string
|
||||
// Task timeout, in seconds.
|
||||
Timeout int
|
||||
// What we should call?
|
||||
// This should be an event name.
|
||||
Callee string
|
||||
// Task name.
|
||||
Name string
|
||||
// Task timeout, in seconds.
|
||||
Timeout int
|
||||
// What we should call?
|
||||
// This should be an event name.
|
||||
Callee string
|
||||
|
||||
// Internal variables, used by Timer.
|
||||
// These variables can be defined, but they will be most likely
|
||||
// overrided after first task launch.
|
||||
// Next task launch time.
|
||||
NextLaunch time.Time
|
||||
// Is task currently executed?
|
||||
// Kinda alternative to mutex.
|
||||
InProgress bool
|
||||
// Internal variables, used by Timer.
|
||||
// These variables can be defined, but they will be most likely
|
||||
// overrided after first task launch.
|
||||
// Next task launch time.
|
||||
NextLaunch time.Time
|
||||
// Is task currently executed?
|
||||
// Kinda alternative to mutex.
|
||||
InProgress bool
|
||||
}
|
||||
|
Reference in New Issue
Block a user