|
|
|
@@ -1,16 +1,17 @@
|
|
|
|
package svej
|
|
|
|
package svej
|
|
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
import (
|
|
|
|
|
|
|
|
"encoding/json"
|
|
|
|
|
|
|
|
"fmt"
|
|
|
|
"log"
|
|
|
|
"log"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"strconv"
|
|
|
|
"strconv"
|
|
|
|
"strings"
|
|
|
|
"strings"
|
|
|
|
"fmt"
|
|
|
|
"time"
|
|
|
|
"github.com/oliveagle/jsonpath"
|
|
|
|
|
|
|
|
"encoding/json"
|
|
|
|
|
|
|
|
"udi/config"
|
|
|
|
"udi/config"
|
|
|
|
"udi/handlers/handler"
|
|
|
|
|
|
|
|
"udi/database"
|
|
|
|
"udi/database"
|
|
|
|
|
|
|
|
"udi/handlers/handler"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/oliveagle/jsonpath"
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
type SingleValueExtractorJsonpathHandler struct {
|
|
|
|
type SingleValueExtractorJsonpathHandler struct {
|
|
|
|
@@ -33,7 +34,6 @@ T:TopicPartIndex
|
|
|
|
C:ConstantValue
|
|
|
|
C:ConstantValue
|
|
|
|
*/
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func New(id string, config config.HandlerConfigT) handler.Handler {
|
|
|
|
func New(id string, config config.HandlerConfigT) handler.Handler {
|
|
|
|
t := &SingleValueExtractorJsonpathHandler{
|
|
|
|
t := &SingleValueExtractorJsonpathHandler{
|
|
|
|
ready: false,
|
|
|
|
ready: false,
|
|
|
|
@@ -46,7 +46,7 @@ func New(id string, config config.HandlerConfigT) handler.Handler {
|
|
|
|
t.application = config.Attributes["application"]
|
|
|
|
t.application = config.Attributes["application"]
|
|
|
|
|
|
|
|
|
|
|
|
t.deviceSelector = config.Attributes["deviceSelector"]
|
|
|
|
t.deviceSelector = config.Attributes["deviceSelector"]
|
|
|
|
if t.deviceSelector[:2] == "J:" {
|
|
|
|
if t.deviceSelector[:2] == "J:" || t.deviceSelector[:2] == "j:" {
|
|
|
|
jp, err := jsonpath.Compile(t.deviceSelector[2:])
|
|
|
|
jp, err := jsonpath.Compile(t.deviceSelector[2:])
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("Unable to compile deviceJsonpath: %s, %s", t.deviceSelector[2:], err)
|
|
|
|
log.Printf("Unable to compile deviceJsonpath: %s, %s", t.deviceSelector[2:], err)
|
|
|
|
@@ -55,7 +55,7 @@ func New(id string, config config.HandlerConfigT) handler.Handler {
|
|
|
|
t.deviceJsonpath = jp
|
|
|
|
t.deviceJsonpath = jp
|
|
|
|
}
|
|
|
|
}
|
|
|
|
t.valueSelector = config.Attributes["valueSelector"]
|
|
|
|
t.valueSelector = config.Attributes["valueSelector"]
|
|
|
|
if t.valueSelector[:2] == "J:" {
|
|
|
|
if t.valueSelector[:2] == "J:" || t.valueSelector[:2] == "j:" {
|
|
|
|
jp, err := jsonpath.Compile(t.valueSelector[2:])
|
|
|
|
jp, err := jsonpath.Compile(t.valueSelector[2:])
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("Unable to compile valueJsonpath: %s, %s", t.valueSelector[2:], err)
|
|
|
|
log.Printf("Unable to compile valueJsonpath: %s, %s", t.valueSelector[2:], err)
|
|
|
|
@@ -64,7 +64,7 @@ func New(id string, config config.HandlerConfigT) handler.Handler {
|
|
|
|
t.valueJsonpath = jp
|
|
|
|
t.valueJsonpath = jp
|
|
|
|
}
|
|
|
|
}
|
|
|
|
t.unitSelector = config.Attributes["unitSelector"]
|
|
|
|
t.unitSelector = config.Attributes["unitSelector"]
|
|
|
|
if t.unitSelector[:2] == "J:" {
|
|
|
|
if t.unitSelector[:2] == "J:" || t.unitSelector[:2] == "j:" {
|
|
|
|
jp, err := jsonpath.Compile(t.unitSelector[2:])
|
|
|
|
jp, err := jsonpath.Compile(t.unitSelector[2:])
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("Unable to compile unitJsonpath: %s, %s", t.unitSelector[2:], err)
|
|
|
|
log.Printf("Unable to compile unitJsonpath: %s, %s", t.unitSelector[2:], err)
|
|
|
|
@@ -76,20 +76,24 @@ func New(id string, config config.HandlerConfigT) handler.Handler {
|
|
|
|
t.Id = id
|
|
|
|
t.Id = id
|
|
|
|
t.ready = true
|
|
|
|
t.ready = true
|
|
|
|
t.dbh = database.NewDatabaseHandle()
|
|
|
|
t.dbh = database.NewDatabaseHandle()
|
|
|
|
log.Printf("Handler SVEJ %d initialized", id)
|
|
|
|
log.Printf("Handler SVEJ %s initialized", id)
|
|
|
|
return t
|
|
|
|
return t
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func (self *SingleValueExtractorJsonpathHandler) ExtractionHelper(subTopics []string, jPayload interface{}, selector string, jp *jsonpath.Compiled) (string, error) {
|
|
|
|
func (self *SingleValueExtractorJsonpathHandler) ExtractionHelper(subTopics []string, jPayload interface{}, selector string, jp *jsonpath.Compiled) (interface{}, error) {
|
|
|
|
var res string
|
|
|
|
var res interface{}
|
|
|
|
switch selector[:2] {
|
|
|
|
switch selector[:2] {
|
|
|
|
case "J:":
|
|
|
|
case "J:", "j:":
|
|
|
|
// extract using jsonpath from payload
|
|
|
|
// extract using jsonpath from payload
|
|
|
|
r, e := jp.Lookup(jPayload)
|
|
|
|
r, e := jp.Lookup(jPayload)
|
|
|
|
if e != nil {
|
|
|
|
if e != nil {
|
|
|
|
return "", fmt.Errorf("jp.Lookup failed with %s", e)
|
|
|
|
return "", fmt.Errorf("jp.Lookup failed with %s", e)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if selector[:2] == "j:" {
|
|
|
|
|
|
|
|
res = r.(float64)
|
|
|
|
|
|
|
|
} else {
|
|
|
|
res = fmt.Sprint(r)
|
|
|
|
res = fmt.Sprint(r)
|
|
|
|
|
|
|
|
}
|
|
|
|
case "T:":
|
|
|
|
case "T:":
|
|
|
|
// T: extract from topic
|
|
|
|
// T: extract from topic
|
|
|
|
i, e := strconv.Atoi(selector[2:])
|
|
|
|
i, e := strconv.Atoi(selector[2:])
|
|
|
|
@@ -109,7 +113,6 @@ func (self *SingleValueExtractorJsonpathHandler) ExtractionHelper(subTopics []st
|
|
|
|
return res, nil
|
|
|
|
return res, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func (self *SingleValueExtractorJsonpathHandler) Handle(message handler.MessageT) {
|
|
|
|
func (self *SingleValueExtractorJsonpathHandler) Handle(message handler.MessageT) {
|
|
|
|
if !self.ready {
|
|
|
|
if !self.ready {
|
|
|
|
self.Lost("Handler is not marked as ready", nil, message)
|
|
|
|
self.Lost("Handler is not marked as ready", nil, message)
|
|
|
|
@@ -149,12 +152,12 @@ func (self *SingleValueExtractorJsonpathHandler) Handle(message handler.MessageT
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
measurement.Device = device
|
|
|
|
measurement.Device = device.(string)
|
|
|
|
|
|
|
|
|
|
|
|
var variable database.VariableType
|
|
|
|
var variable database.VariableType
|
|
|
|
variable.Label = ""
|
|
|
|
variable.Label = ""
|
|
|
|
variable.Variable = ""
|
|
|
|
variable.Variable = ""
|
|
|
|
variable.Unit = unit
|
|
|
|
variable.Unit = unit.(string)
|
|
|
|
variable.Value = value
|
|
|
|
variable.Value = value
|
|
|
|
measurement.Values = make(map[string]database.VariableType)
|
|
|
|
measurement.Values = make(map[string]database.VariableType)
|
|
|
|
measurement.Values["Value"] = variable
|
|
|
|
measurement.Values["Value"] = variable
|
|
|
|
@@ -163,4 +166,3 @@ func (self *SingleValueExtractorJsonpathHandler) Handle(message handler.MessageT
|
|
|
|
self.dbh.StoreMeasurement(&measurement)
|
|
|
|
self.dbh.StoreMeasurement(&measurement)
|
|
|
|
self.S()
|
|
|
|
self.S()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|