1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
package main
import (
"errors"
"time"
)
type Database interface {
GetTracking() (*Task, error)
GetTasks() ([]*Task, error)
QueryTask(id *int) (*Task, error)
StartNewEntry(now time.Time, task *Task) (*Task, error)
StopEntry(now time.Time) error
SaveEntry(task *Task) (*Task, error)
}
type InMemDb struct {
tasks []*Task
current *Task
}
var (
ErrNotFound = errors.New("Not found")
ErrNotRunning = errors.New("Not running")
)
func NewInMemDb() Database {
return &InMemDb {
tasks: []*Task{},
current: nil,
}
}
func (db *InMemDb) GetTracking() (*Task, error) {
return db.current, nil
}
func (db *InMemDb) GetTasks() ([]*Task, error) {
res := make([]*Task, 0, len(db.tasks))
for _, task := range db.tasks {
res = append(res, task)
}
return res, nil
}
func (db *InMemDb) QueryTask(id *int) (*Task, error) {
for _, task := range db.tasks {
if id != nil && task.Id == *id {
return task, nil
}
}
return nil, ErrNotFound
}
func (db *InMemDb) StartNewEntry(now time.Time, task *Task) (*Task, error) {
newTask := &Task {
Id: len(db.tasks),
From: task.From,
To: nil,
Tag: task.Tag,
Comment: task.Comment,
}
if newTask.From == nil {
newTask.From = &now
}
db.tasks = append(db.tasks, newTask)
db.StopEntry(now)
db.current = newTask
return newTask, nil
}
func (db *InMemDb) StopEntry(now time.Time) error {
if db.current != nil {
db.current.To = &now
db.current = nil
} else {
return ErrNotRunning
}
return nil
}
func (db *InMemDb) SaveEntry(task *Task) (*Task, error) {
if task.Id < 0 || task.Id >= len(db.tasks) {
copyTask := *task
copyTask.Id = len(db.tasks)
db.tasks = append(db.tasks, ©Task)
return ©Task, nil
}
existent := db.tasks[task.Id]
existent.Comment = task.Comment
existent.Tag = task.Tag
existent.From = task.From
existent.To = task.To
return existent, nil
}
|