parent
f506dfdbcd
commit
ec64a2b58b
@ -0,0 +1,117 @@
|
||||
package m9zApi
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sync"
|
||||
|
||||
g "tgk-touch/internal/global"
|
||||
|
||||
"github.com/towgo/towgo/lib/system"
|
||||
"github.com/towgo/towgo/towgo"
|
||||
)
|
||||
|
||||
type ScanConfigManager struct {
|
||||
mu sync.RWMutex
|
||||
loopCount int
|
||||
}
|
||||
|
||||
type ScanConfigData struct {
|
||||
LoopCount int `json:"loop_count"`
|
||||
}
|
||||
|
||||
var scanConfigManager = &ScanConfigManager{}
|
||||
|
||||
func getScanConfigFilePath() string {
|
||||
return filepath.Join(system.GetPathOfProgram(), "data", "scanConfig.json")
|
||||
}
|
||||
|
||||
func (m *ScanConfigManager) GetLoopCount() int {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
return normalizeLoopCount(m.loopCount)
|
||||
}
|
||||
|
||||
func (m *ScanConfigManager) SetLoopCount(loopCount int) error {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
m.loopCount = loopCount
|
||||
if err := m.saveToFile(); err != nil {
|
||||
return err
|
||||
}
|
||||
g.GVA_Config.DeviceInfo.LoopCount = loopCount
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *ScanConfigManager) loadFromFile() error {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
filePath := getScanConfigFilePath()
|
||||
jsonData, err := os.ReadFile(filePath)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
m.loopCount = normalizeLoopCount(g.Config().DeviceInfo.LoopCount)
|
||||
return m.saveToFile()
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
var data ScanConfigData
|
||||
if err := json.Unmarshal(jsonData, &data); err != nil {
|
||||
return err
|
||||
}
|
||||
m.loopCount = normalizeLoopCount(data.LoopCount)
|
||||
g.GVA_Config.DeviceInfo.LoopCount = m.loopCount
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *ScanConfigManager) saveToFile() error {
|
||||
filePath := getScanConfigFilePath()
|
||||
dir := filepath.Dir(filePath)
|
||||
if err := os.MkdirAll(dir, 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
jsonData, err := json.Marshal(ScanConfigData{
|
||||
LoopCount: normalizeLoopCount(m.loopCount),
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return os.WriteFile(filePath, jsonData, 0600)
|
||||
}
|
||||
|
||||
func getScanConfig(rpcConn towgo.JsonRpcConnection) {
|
||||
rpcConn.WriteResult(map[string]interface{}{
|
||||
"loop_count": scanConfigManager.GetLoopCount(),
|
||||
"max_loop_count": maxLoopCount,
|
||||
"min_loop_count": 1,
|
||||
})
|
||||
}
|
||||
|
||||
type SetScanConfigReq struct {
|
||||
LoopCount int `json:"loop_count"`
|
||||
}
|
||||
|
||||
func setScanConfig(rpcConn towgo.JsonRpcConnection) {
|
||||
var req SetScanConfigReq
|
||||
rpcConn.ReadParams(&req)
|
||||
|
||||
if req.LoopCount < 1 || req.LoopCount > maxLoopCount {
|
||||
rpcConn.WriteError(400, "loop_count must be between 1 and 10")
|
||||
return
|
||||
}
|
||||
if err := scanConfigManager.SetLoopCount(req.LoopCount); err != nil {
|
||||
rpcConn.WriteError(500, "save scan config failed: "+err.Error())
|
||||
return
|
||||
}
|
||||
rpcConn.WriteResult(map[string]interface{}{
|
||||
"success": true,
|
||||
"loop_count": scanConfigManager.GetLoopCount(),
|
||||
"max_loop_count": maxLoopCount,
|
||||
"min_loop_count": 1,
|
||||
})
|
||||
}
|
||||
@ -0,0 +1,24 @@
|
||||
package m9zApi
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestNormalizeLoopCount(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
in int
|
||||
want int
|
||||
}{
|
||||
{name: "default when empty", in: 0, want: defaultLoopCount},
|
||||
{name: "default when negative", in: -1, want: defaultLoopCount},
|
||||
{name: "keeps valid value", in: 6, want: 6},
|
||||
{name: "clamps max value", in: 12, want: maxLoopCount},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := normalizeLoopCount(tt.in); got != tt.want {
|
||||
t.Fatalf("normalizeLoopCount(%d) = %d, want %d", tt.in, got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue