136 lines
2.8 KiB
Go
136 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/leandrofars/oktopus/internal/db"
|
|
)
|
|
|
|
type StatusCount struct {
|
|
Online int
|
|
Offline int
|
|
}
|
|
|
|
type GeneralInfo struct {
|
|
MqttRtt time.Duration
|
|
ProductClassCount []db.ProductClassCount
|
|
StatusCount StatusCount
|
|
VendorsCount []db.VendorsCount
|
|
}
|
|
|
|
// TODO: fix when mqtt broker is not set don't break api
|
|
func (a *Api) generalInfo(w http.ResponseWriter, r *http.Request) {
|
|
|
|
var result GeneralInfo
|
|
|
|
productclasscount, err := a.Db.RetrieveProductsClassInfo()
|
|
if err != nil {
|
|
log.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
vendorcount, err := a.Db.RetrieveVendorsInfo()
|
|
if err != nil {
|
|
log.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
statuscount, err := a.Db.RetrieveStatusInfo()
|
|
if err != nil {
|
|
log.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
for _, v := range statuscount {
|
|
switch db.Status(v.Status) {
|
|
case db.Online:
|
|
result.StatusCount.Online = v.Count
|
|
case db.Offline:
|
|
result.StatusCount.Offline = v.Count
|
|
}
|
|
}
|
|
|
|
result.VendorsCount = vendorcount
|
|
result.ProductClassCount = productclasscount
|
|
|
|
conn, err := net.Dial("tcp", a.Mqtt.Addr+":"+a.Mqtt.Port)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
json.NewEncoder(w).Encode("Error to connect to broker: " + err.Error())
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
info, err := tcpInfo(conn.(*net.TCPConn))
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
json.NewEncoder(w).Encode("Error to get TCP socket info")
|
|
return
|
|
}
|
|
rtt := time.Duration(info.Rtt) * time.Microsecond
|
|
|
|
result.MqttRtt = rtt / 1000
|
|
|
|
err = json.NewEncoder(w).Encode(result)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
func (a *Api) vendorsInfo(w http.ResponseWriter, r *http.Request) {
|
|
vendors, err := a.Db.RetrieveVendorsInfo()
|
|
if err != nil {
|
|
log.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
err = json.NewEncoder(w).Encode(vendors)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
func (a *Api) productClassInfo(w http.ResponseWriter, r *http.Request) {
|
|
vendors, err := a.Db.RetrieveProductsClassInfo()
|
|
if err != nil {
|
|
log.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
err = json.NewEncoder(w).Encode(vendors)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|
|
|
|
func (a *Api) statusInfo(w http.ResponseWriter, r *http.Request) {
|
|
vendors, err := a.Db.RetrieveStatusInfo()
|
|
if err != nil {
|
|
log.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
var status StatusCount
|
|
for _, v := range vendors {
|
|
switch db.Status(v.Status) {
|
|
case db.Online:
|
|
status.Online = v.Count
|
|
case db.Offline:
|
|
status.Offline = v.Count
|
|
}
|
|
}
|
|
|
|
err = json.NewEncoder(w).Encode(status)
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
}
|