115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"time"
|
|
|
|
"ewintr.nl/gte/internal/task"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// Memory is an in memory implementation of LocalRepository
|
|
type Memory struct {
|
|
tasks map[string]*task.LocalTask
|
|
latestSync time.Time
|
|
}
|
|
|
|
func NewMemory(initTasks ...*task.Task) *Memory {
|
|
tasks := map[string]*task.LocalTask{}
|
|
for _, t := range initTasks {
|
|
tasks[t.Id] = &task.LocalTask{
|
|
Task: *t,
|
|
LocalUpdate: &task.LocalUpdate{},
|
|
}
|
|
}
|
|
|
|
return &Memory{
|
|
tasks: tasks,
|
|
}
|
|
}
|
|
|
|
func (m *Memory) LatestSync() (time.Time, error) {
|
|
return m.latestSync, nil
|
|
}
|
|
|
|
func (m *Memory) SetTasks(tasks []*task.Task) error {
|
|
var oldTasks []*task.LocalTask
|
|
for _, ot := range m.tasks {
|
|
oldTasks = append(oldTasks, ot)
|
|
}
|
|
|
|
newTasks := MergeNewTaskSet(oldTasks, tasks)
|
|
|
|
m.tasks = map[string]*task.LocalTask{}
|
|
for _, nt := range newTasks {
|
|
m.tasks[nt.Id] = nt
|
|
}
|
|
m.latestSync = time.Now()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Memory) FindAll() ([]*task.LocalTask, error) {
|
|
tasks := []*task.LocalTask{}
|
|
for _, t := range m.tasks {
|
|
tasks = append(tasks, t)
|
|
}
|
|
|
|
return tasks, nil
|
|
}
|
|
|
|
func (m *Memory) FindById(id string) (*task.LocalTask, error) {
|
|
for _, t := range m.tasks {
|
|
if t.Id == id {
|
|
return t, nil
|
|
}
|
|
}
|
|
|
|
return &task.LocalTask{}, ErrTaskNotFound
|
|
}
|
|
|
|
func (m *Memory) FindByLocalId(localId int) (*task.LocalTask, error) {
|
|
for _, t := range m.tasks {
|
|
if t.LocalId == localId {
|
|
return t, nil
|
|
}
|
|
}
|
|
|
|
return &task.LocalTask{}, ErrTaskNotFound
|
|
}
|
|
|
|
func (m *Memory) SetLocalUpdate(id string, update *task.LocalUpdate) error {
|
|
m.tasks[id].LocalStatus = task.STATUS_UPDATED
|
|
m.tasks[id].LocalUpdate = update
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Memory) MarkDispatched(localId int) error {
|
|
t, _ := m.FindByLocalId(localId)
|
|
m.tasks[t.Id].LocalStatus = task.STATUS_DISPATCHED
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Memory) Add(update *task.LocalUpdate) (*task.LocalTask, error) {
|
|
var used []int
|
|
for _, t := range m.tasks {
|
|
used = append(used, t.LocalId)
|
|
}
|
|
|
|
tsk := &task.LocalTask{
|
|
Task: task.Task{
|
|
Id: uuid.New().String(),
|
|
Version: 0,
|
|
Folder: task.FOLDER_NEW,
|
|
},
|
|
LocalId: NextLocalId(used),
|
|
LocalStatus: task.STATUS_UPDATED,
|
|
LocalUpdate: update,
|
|
}
|
|
|
|
m.tasks[tsk.Id] = tsk
|
|
|
|
return tsk, nil
|
|
}
|