53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type Storage struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func (stor *Storage) init() error {
|
|
// initialization
|
|
sql_table := `
|
|
CREATE TABLE IF NOT EXISTS geopoints (
|
|
device VARCHAR(256), timestamp INTEGER, lon REAL, lat REAL, alt REAL, acc REAL, velocity REAL, bearing REAL, PRIMARY KEY(owner, timestamp)
|
|
);
|
|
`
|
|
if _, err := stor.db.Exec(sql_table); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func CreateDatabase(filename string) (Storage, error) {
|
|
var stor Storage
|
|
var err error
|
|
stor.db, err = sql.Open("sqlite3", filename)
|
|
if err != nil {
|
|
return stor, err
|
|
}
|
|
|
|
return stor, stor.init()
|
|
}
|
|
|
|
func (stor *Storage) InsertLocation(device string, loc Location) error {
|
|
tx, err := stor.db.Begin()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
sql := "INSERT INTO geopoints (device, timestamp, lon, lat, alt, acc, velocity, bearing) VALUES (?,?,?,?,?,?,?)"
|
|
stmt, err := tx.Prepare(sql)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer stmt.Close()
|
|
stmt.Exec(device, loc.Timestamp, loc.Lon, loc.Lat, loc.Alt, loc.Velocity, loc.Bearing)
|
|
|
|
return tx.Commit()
|
|
}
|