MQTTListener/models/db.go
Siegfried Siegert daa79b27ca Extend this program:
- add a database for devices
- read devices from database and present this data on a simple web/REST
interface

The code or model is from a blog entry from Alex Edward:
https://www.alexedwards.net/blog/organising-database-access
2020-08-30 07:16:50 +02:00

34 lines
564 B
Go

package models
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
)
type Datastore interface {
AllDevices() ([]*Device, error)
}
type DB struct {
*sql.DB
}
func NewDB(dataSourceName string) (*DB, error) {
db, err := sql.Open("sqlite3", dataSourceName)
if err != nil {
return nil, err
}
if err = db.Ping(); err != nil {
return nil, err
}
sqlStmt := "CREATE TABLE IF NOT EXISTS devices (id TEXT not null primary key, mac TEXT, sn TEXT, lastMsg TEXT);"
_, err = db.Exec(sqlStmt)
if err != nil {
return nil, err
}
return &DB{db}, nil
}