146 lines
2.8 KiB
Go
146 lines
2.8 KiB
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.fuyu.moe/5GPowerQuality/api/app/internal/model"
|
|
"git.ultraware.nl/NiseVoid/qb"
|
|
"git.ultraware.nl/NiseVoid/qb/qc"
|
|
"git.ultraware.nl/NiseVoid/qb/qf"
|
|
)
|
|
|
|
// Count returns the number of measurements
|
|
func Count() int {
|
|
m := model.Measurement()
|
|
|
|
q := m.Select(qf.CountAll())
|
|
|
|
var count *int
|
|
err := db.QueryRow(q).Scan(&count)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if count == nil {
|
|
return 0
|
|
}
|
|
|
|
return *count
|
|
}
|
|
|
|
// Attribute represents a single attribute
|
|
type Attribute struct {
|
|
Name string
|
|
Index int
|
|
}
|
|
|
|
// GetAttributes returns a list of all known attributes
|
|
func GetAttributes() (list []Attribute) {
|
|
m := model.Measurement()
|
|
r := reflect.TypeOf(m).Elem()
|
|
|
|
for i := 0; i < r.NumField(); i++ {
|
|
f := r.Field(i)
|
|
if f.Name[len(f.Name)-1] == '1' {
|
|
list = append(list, Attribute{f.Name[:len(f.Name)-1], i})
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
type Source string
|
|
|
|
func (s *Source) Scan(v interface{}) error {
|
|
i, ok := v.(int64)
|
|
if !ok {
|
|
return fmt.Errorf(`Unsupported scan, storing driver.Value type %T into type %T`, v, s)
|
|
}
|
|
|
|
*s = Source([]string{``, `Fortop`, `Envitron`}[i])
|
|
return nil
|
|
}
|
|
|
|
// Meter represents a single meter
|
|
type Meter struct {
|
|
ID int `json:"id"`
|
|
Name string `json:"name"`
|
|
Source Source `json:"source"`
|
|
}
|
|
|
|
// GetMeters returns a list of all known meters
|
|
func GetMeters() (list []Meter) {
|
|
me := model.Meter()
|
|
|
|
q := me.Select(me.ID, me.Name, me.Source)
|
|
|
|
rows, err := db.Query(q)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for rows.Next() {
|
|
var meter Meter
|
|
err = rows.Scan(&meter.ID, &meter.Name, &meter.Source)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
list = append(list, meter)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// GetAttribute returns the specified attribute for all phases since the specified date
|
|
func GetAttribute(index, meter int, date time.Time) (data []Phases) {
|
|
m := model.Measurement()
|
|
r := reflect.ValueOf(m).Elem()
|
|
|
|
f1, f2, f3 :=
|
|
r.Field(index).Interface().(*qb.TableField),
|
|
r.Field(index+1).Interface().(*qb.TableField),
|
|
r.Field(index+2).Interface().(*qb.TableField)
|
|
|
|
rN := r.Field(index + 3)
|
|
fN := qb.Value(nil)
|
|
if rN.IsValid() && strings.HasSuffix(r.Type().Field(index+3).Name, `N`) {
|
|
fN = rN.Interface().(*qb.TableField)
|
|
}
|
|
|
|
q := m.Select(m.Time, f1, f2, f3, fN).
|
|
OrderBy(qb.Desc(m.Time)).
|
|
Where(
|
|
qc.Gte(m.Time, date),
|
|
qc.Eq(m.MeterID, meter),
|
|
)
|
|
|
|
rows, err := db.Query(q)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for rows.Next() {
|
|
var phases Phases
|
|
err := rows.Scan(&phases.Time, &phases.P1, &phases.P2, &phases.P3, &phases.N)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
data = append(data, phases)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Phases contains data for all phases
|
|
type Phases struct {
|
|
Time time.Time `json:"time"`
|
|
P1 float64 `json:"phase_1"`
|
|
P2 float64 `json:"phase_2"`
|
|
P3 float64 `json:"phase_3"`
|
|
N *float64 `json:"null"`
|
|
}
|