parent
3e6cf9d277
commit
3b62ad9654
@ -0,0 +1,123 @@
|
||||
package m9zApi
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"tgk-touch/internal/library/m9z"
|
||||
)
|
||||
|
||||
const automaticModeUseDemoAPI = false
|
||||
|
||||
type automaticModeTaskReq struct {
|
||||
CommUid string `json:"comm_uid"`
|
||||
Mode uint8 `json:"mode"`
|
||||
}
|
||||
|
||||
type automaticModeDemoStore struct {
|
||||
mu sync.RWMutex
|
||||
currentMode uint8
|
||||
startTasks [3]m9z.ControlTask
|
||||
stopTasks [3]m9z.ControlTask
|
||||
middleTasks [3]m9z.MiddleControlTask
|
||||
}
|
||||
|
||||
var automaticModeDemo = newAutomaticModeDemoStore()
|
||||
|
||||
func newAutomaticModeDemoStore() *automaticModeDemoStore {
|
||||
store := &automaticModeDemoStore{}
|
||||
for mode := uint8(0); mode < 3; mode++ {
|
||||
store.startTasks[mode] = buildAutomaticModeDemoTask(6+int(mode), 30, true, uint8(60+mode*10))
|
||||
store.stopTasks[mode] = buildAutomaticModeDemoTask(22, int(mode)*10, false, 0)
|
||||
store.middleTasks[mode] = m9z.MiddleControlTask{
|
||||
ControlTaskList: []m9z.ControlTask{
|
||||
buildAutomaticModeDemoTask(12, int(mode)*10, true, uint8(45+mode*5)),
|
||||
},
|
||||
}
|
||||
}
|
||||
return store
|
||||
}
|
||||
|
||||
func buildAutomaticModeDemoTask(hour int, minute int, turnOn bool, pwm uint8) m9z.ControlTask {
|
||||
loops := make([]m9z.TaskLoop, 10)
|
||||
for i := range loops {
|
||||
loops[i] = m9z.TaskLoop{
|
||||
Index: uint(i),
|
||||
TurnOn: turnOn,
|
||||
Pwm: pwm,
|
||||
}
|
||||
}
|
||||
return m9z.ControlTask{
|
||||
ExecTime: time.Date(2025, 6, 7, hour, minute, 0, 0, time.Local),
|
||||
TimeType: 0,
|
||||
Loops: loops,
|
||||
}
|
||||
}
|
||||
|
||||
func normalizeAutomaticMode(mode uint8) uint8 {
|
||||
if mode > 2 {
|
||||
return 0
|
||||
}
|
||||
return mode
|
||||
}
|
||||
|
||||
func cloneAutomaticControlTask(task m9z.ControlTask) m9z.ControlTask {
|
||||
task.Loops = append([]m9z.TaskLoop(nil), task.Loops...)
|
||||
return task
|
||||
}
|
||||
|
||||
func cloneAutomaticMiddleTask(task m9z.MiddleControlTask) m9z.MiddleControlTask {
|
||||
task.ControlTaskList = append([]m9z.ControlTask(nil), task.ControlTaskList...)
|
||||
for i := range task.ControlTaskList {
|
||||
task.ControlTaskList[i] = cloneAutomaticControlTask(task.ControlTaskList[i])
|
||||
}
|
||||
return task
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) getMode() uint8 {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return s.currentMode
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) setMode(mode uint8) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.currentMode = normalizeAutomaticMode(mode)
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) getStartTask(mode uint8) m9z.ControlTask {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return cloneAutomaticControlTask(s.startTasks[normalizeAutomaticMode(mode)])
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) setStartTask(mode uint8, task m9z.ControlTask) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.startTasks[normalizeAutomaticMode(mode)] = cloneAutomaticControlTask(task)
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) getStopTask(mode uint8) m9z.ControlTask {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return cloneAutomaticControlTask(s.stopTasks[normalizeAutomaticMode(mode)])
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) setStopTask(mode uint8, task m9z.ControlTask) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.stopTasks[normalizeAutomaticMode(mode)] = cloneAutomaticControlTask(task)
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) getMiddleTask(mode uint8) m9z.MiddleControlTask {
|
||||
s.mu.RLock()
|
||||
defer s.mu.RUnlock()
|
||||
return cloneAutomaticMiddleTask(s.middleTasks[normalizeAutomaticMode(mode)])
|
||||
}
|
||||
|
||||
func (s *automaticModeDemoStore) setMiddleTask(mode uint8, task m9z.MiddleControlTask) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.middleTasks[normalizeAutomaticMode(mode)] = cloneAutomaticMiddleTask(task)
|
||||
}
|
||||
@ -0,0 +1,39 @@
|
||||
package m9zApi
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestAutomaticModeDemoStoreReturnsCopies(t *testing.T) {
|
||||
store := newAutomaticModeDemoStore()
|
||||
|
||||
start := store.getStartTask(0)
|
||||
if len(start.Loops) == 0 {
|
||||
t.Fatal("demo start task should include loops")
|
||||
}
|
||||
start.Loops[0].TurnOn = !start.Loops[0].TurnOn
|
||||
start.Loops[0].Pwm = 1
|
||||
|
||||
got := store.getStartTask(0)
|
||||
if got.Loops[0].Pwm == 1 {
|
||||
t.Fatal("demo start task loop was mutated through returned value")
|
||||
}
|
||||
|
||||
middle := store.getMiddleTask(0)
|
||||
if len(middle.ControlTaskList) == 0 || len(middle.ControlTaskList[0].Loops) == 0 {
|
||||
t.Fatal("demo middle task should include control tasks and loops")
|
||||
}
|
||||
middle.ControlTaskList[0].Loops[0].Pwm = 1
|
||||
|
||||
gotMiddle := store.getMiddleTask(0)
|
||||
if gotMiddle.ControlTaskList[0].Loops[0].Pwm == 1 {
|
||||
t.Fatal("demo middle task loop was mutated through returned value")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNormalizeAutomaticMode(t *testing.T) {
|
||||
if got := normalizeAutomaticMode(2); got != 2 {
|
||||
t.Fatalf("normalizeAutomaticMode(2) = %d, want 2", got)
|
||||
}
|
||||
if got := normalizeAutomaticMode(3); got != 0 {
|
||||
t.Fatalf("normalizeAutomaticMode(3) = %d, want 0", got)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue