MQTTListener/models/devices.go

52 lines
1.2 KiB
Go
Raw Normal View History

package models
2020-09-02 09:02:30 +02:00
import (
"fmt"
)
type Device struct {
2020-09-13 18:37:22 +02:00
TOPIC string
CustomerID string
DeviceID string
ProjectName string
MAC string
SN string
LastMsg string
}
func (db *DB) AllDevices() ([]*Device, error) {
rows, err := db.Query("SELECT * FROM devices")
if err != nil {
return nil, err
}
defer rows.Close()
devices := make([]*Device, 0)
for rows.Next() {
device := new(Device)
2020-09-13 18:37:22 +02:00
err := rows.Scan(&device.TOPIC, device.CustomerID, device.DeviceID, device.ProjectName, &device.MAC, &device.SN, &device.LastMsg)
if err != nil {
return nil, err
}
devices = append(devices, device)
}
if err = rows.Err(); err != nil {
return nil, err
}
return devices, nil
}
func (db *DB) InsertDevice(device *Device) error {
2020-09-13 18:37:22 +02:00
sqlStmt := `INSERT OR REPLACE INTO devices (topic, customerID, deviceID, projectName, mac, sn, lastMsg)
VALUES($1, $2, $3, $4, $5, $6, $7);`
_, err := db.Exec(sqlStmt, device.TOPIC, device.CustomerID, device.DeviceID, device.ProjectName, device.MAC, device.SN, device.LastMsg)
if err != nil {
return err
}
2020-09-02 09:02:30 +02:00
// DEBUG
2020-09-13 18:37:22 +02:00
fmt.Printf("Sucessfully inserted device with ID %s to database\n", device.TOPIC)
2020-09-02 09:02:30 +02:00
return err
}