Files
universal-data-ingest/src/udi/handlers/ttn/ttn.go
Wolfgang Hottgenroth b3de6182b3
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed
emu
2023-12-11 16:50:28 +01:00

136 lines
4.1 KiB
Go

package ttn
import "log"
import "fmt"
import "encoding/json"
import "udi/config"
import "udi/handlers/handler"
import "udi/handlers/ttn/models/emuProfIILoRaCfg1"
import "udi/database"
var idSeq int = 0
type TTNHandler struct {
id int
dbh *database.DatabaseHandle
}
type uplinkMessage struct {
EndDeviceIds struct {
DeviceId string `json:"device_id"`
ApplicationIds struct {
ApplicationId string `json:"application_id"`
} `json:"application_ids"`
DevEui string `json:"dev_eui"`
JoinEui string `json:"join_eui"`
DevAddr string `json:"dev_addr"`
} `json:"end_device_ids"`
ReceivedAt string `json:"received_at"`
UplinkMessage struct {
FCnt int `json:"f_cnt"`
FPort int `json:"f_port"`
FrmPayload string `json:"frm_payload"`
DecodedPayload map[string]interface{} `json:"decoded_payload"`
RxMetadata []struct {
GatewayIds struct {
GatewayId string `json:"gateway_id"`
Eui string `json:"eui"`
} `json:"gateway_ids"`
Time string `json:"time"`
Rssi int `json:"rssi"`
ChannelRssi int `json:"channel_rssi"`
Snr float32 `json:"snr"`
ChannelIndex int `json:"channel_index"`
} `json:"rx_metadata"`
ConsumedAirtime string `json:"consumed_airtime"`
} `json:"uplink_message"`
}
type gatewayAttributes struct {
GatewayId string `json:"gateway_id"`
Rssi int `json:"rssi"`
Snr float32 `json:"snr"`
}
type attributes struct {
DeviceId string `json:"device_id"`
ApplicationId string `json:"application_id"`
FCnt int `json:"f_cnt"`
FPort int `json:"f_port"`
FrmPayload string `json:"frm_payload"`
Gateways []gatewayAttributes `json:"gateways"`
ConsumedAirtime string `json:"consumed_airtime"`
}
func NewTTNHandler(config config.HandlerConfigT) handler.Handler {
t := &TTNHandler {
id: idSeq,
}
idSeq += 1
t.dbh = database.NewDatabaseHandle(config.DatabaseConnStr)
return t
}
func (self *TTNHandler) GetId() string {
return fmt.Sprintf("TTN%d", self.id)
}
func lost(msg string, message handler.MessageT) {
log.Printf("Error: %s, message %s is lost", msg, message)
}
func (self *TTNHandler) Handle(message handler.MessageT) {
log.Printf("Handler TTN %d processing %s -> %s", self.id, message.Topic, message.Payload)
var uplinkMessage uplinkMessage
err := json.Unmarshal([]byte(message.Payload), &uplinkMessage)
if err != nil {
lost(fmt.Sprintf("Error when unmarshaling message: %s, ", err), message)
return
}
//log.Printf("Parsed message: %s", uplinkMessage)
var attributes attributes
attributes.DeviceId = uplinkMessage.EndDeviceIds.DeviceId
attributes.ApplicationId = uplinkMessage.EndDeviceIds.ApplicationIds.ApplicationId
attributes.FCnt = uplinkMessage.UplinkMessage.FCnt
attributes.FPort = uplinkMessage.UplinkMessage.FPort
attributes.FrmPayload = uplinkMessage.UplinkMessage.FrmPayload
attributes.ConsumedAirtime = uplinkMessage.UplinkMessage.ConsumedAirtime
for _, rxm := range uplinkMessage.UplinkMessage.RxMetadata {
log.Printf("RXM: %s", rxm)
g := gatewayAttributes { GatewayId: rxm.GatewayIds.GatewayId, Rssi: rxm.Rssi, Snr: rxm.Snr }
attributes.Gateways = append(attributes.Gateways, g)
}
//log.Printf("Attributes: %s", attributes)
log.Printf("ApplicationId: %s, DeviceId: %s", attributes.ApplicationId, attributes.DeviceId)
device, err2 := self.dbh.GetDeviceByLabelAndApplication(attributes.ApplicationId, attributes.DeviceId)
if err2 != nil {
lost(fmt.Sprintf("Error when loading device: %s, ", err2), message)
return
}
log.Printf("DeviceLabel: %s, DeviceType: %s", device.Label, device.DeviceType.ModelIdentifier)
var parser func(int, interface{}) (map[string]database.VariableType, error)
switch device.DeviceType.ModelIdentifier {
case "emu-prof-ii-lora-cfg1":
parser = emuProfIILoRaCfg1.Parse
default:
lost(fmt.Sprintf("No parser found for %s", device.DeviceType.ModelIdentifier), message)
return
}
_, err3 := parser(uplinkMessage.UplinkMessage.FPort, uplinkMessage.UplinkMessage.DecodedPayload)
if err3 != nil {
lost(fmt.Sprintf("Model parser failed: %s", err3), message)
return
}
}