73 lines
1.8 KiB
Go
73 lines
1.8 KiB
Go
package usp_handler
|
|
|
|
import (
|
|
"log"
|
|
"strconv"
|
|
|
|
"github.com/OktopUSP/oktopus/backend/services/mtp/adapter/internal/db"
|
|
"github.com/OktopUSP/oktopus/backend/services/mtp/adapter/internal/usp"
|
|
"github.com/OktopUSP/oktopus/backend/services/mtp/adapter/internal/usp/usp_msg"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func (h *Handler) HandleDeviceStatus(device, subject string, data []byte, mtp string, ack func()) {
|
|
defer ack()
|
|
payload, err := strconv.Atoi(string(data))
|
|
if err != nil {
|
|
log.Printf("Status subject payload message error %q", err)
|
|
}
|
|
|
|
switch payload {
|
|
case ONLINE:
|
|
h.deviceOnline(device, mtp)
|
|
case OFFLINE:
|
|
h.deviceOffline(device, mtp)
|
|
default:
|
|
ignoreMsg(subject, "status", data)
|
|
}
|
|
}
|
|
|
|
func (h *Handler) deviceOnline(device, mtp string) {
|
|
|
|
log.Printf("Device %s is online", device)
|
|
|
|
msg := usp.NewGetMsg(usp_msg.Get{
|
|
ParamPaths: []string{
|
|
"Device.DeviceInfo.Manufacturer",
|
|
"Device.DeviceInfo.ModelName",
|
|
"Device.DeviceInfo.SoftwareVersion",
|
|
"Device.DeviceInfo.SerialNumber",
|
|
"Device.DeviceInfo.ProductClass",
|
|
},
|
|
MaxDepth: 1,
|
|
})
|
|
|
|
payload, _ := proto.Marshal(&msg)
|
|
record := usp.NewUspRecord(payload, device, h.cid)
|
|
|
|
tr369Message, err := proto.Marshal(&record)
|
|
if err != nil {
|
|
log.Fatalln("Failed to encode tr369 record:", err)
|
|
}
|
|
|
|
err = h.nc.Publish(mtp+"-adapter.usp.v1."+device+".info", tr369Message)
|
|
if err != nil {
|
|
log.Printf("Failed to publish online device message: %v", err)
|
|
}
|
|
}
|
|
|
|
func (h *Handler) deviceOffline(device, mtp string) {
|
|
log.Printf("Device %s is offline", device)
|
|
|
|
mtpLayer := getMtp(mtp)
|
|
|
|
err := h.db.UpdateStatus(device, db.Offline, mtpLayer)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func ignoreMsg(subject, ctx string, data []byte) {
|
|
log.Printf("Unknown message of %s received, subject: %s, payload: %s. Ignored...", ctx, subject, string(data))
|
|
}
|