package plc_address
|
|
import (
|
"apsClient/constvar"
|
"apsClient/pkg/logx"
|
"apsClient/utils/file"
|
"fmt"
|
"github.com/spf13/cast"
|
"os"
|
"strings"
|
"sync"
|
)
|
|
type addressMap struct {
|
store sync.Map
|
}
|
|
var defaultAddressMap *addressMap
|
|
func newAddressMap() *addressMap {
|
return &addressMap{store: sync.Map{}}
|
}
|
|
func init() {
|
defaultAddressMap = newAddressMap()
|
}
|
|
func Set(key string, value int) {
|
defaultAddressMap.store.Store(key, value)
|
}
|
|
func Get(key string) (value int, ok bool) {
|
if v, ok := defaultAddressMap.store.Load(key); ok {
|
return v.(int), ok
|
}
|
return 0, false
|
}
|
|
var LoadOk bool
|
|
func IsLoad() bool {
|
return LoadOk
|
}
|
|
func LoadAddressFromFile() (loadOk bool) {
|
keyFileName := fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataKeyFileName)
|
addressFileName := fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataValueFileName)
|
if !file.Exists(keyFileName) || !file.Exists(addressFileName) {
|
return
|
}
|
keyData, err := os.ReadFile(keyFileName)
|
if err != nil {
|
logx.Errorf("LoadAddressFromFile ReadFile err: %v", err.Error())
|
return
|
}
|
addressData, err := os.ReadFile(addressFileName)
|
if err != nil {
|
logx.Errorf("LoadAddressFromFile ReadFile err: %v", err.Error())
|
return
|
}
|
keyString := string(keyData)
|
addressString := string(addressData)
|
keys := strings.Split(keyString, "\n")
|
addresses := strings.Split(addressString, "\n")
|
if len(keys) != len(addresses) {
|
logx.Error("plc address message error: key length not equal address length")
|
return
|
}
|
for i := 0; i < len(keys); i++ {
|
key := strings.ReplaceAll(keys[i], "\r", "")
|
address := cast.ToInt(strings.ReplaceAll(addresses[i], "\r", ""))
|
Set(key, address)
|
}
|
LoadOk = true
|
return true
|
}
|
