cal sqlite

This commit is contained in:
Erik Winter 2024-09-24 07:44:00 +02:00
parent 540482bc52
commit 49d45970a4
2 changed files with 233 additions and 13 deletions

View File

@ -6,37 +6,63 @@ import (
"time" "time"
"go-mod.ewintr.nl/planner/item" "go-mod.ewintr.nl/planner/item"
"go-mod.ewintr.nl/planner/sync/client"
) )
func main() { func main() {
fmt.Println("cal") fmt.Println("cal")
c := client.NewClient("http://localhost:8092", "testKey") repo, err := NewSqlite("test.db")
items, err := c.Updated([]item.Kind{item.KindEvent}, time.Time{})
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
} }
fmt.Printf("%+v\n", items) one := item.Event{
ID: "a",
i := item.Item{ EventBody: item.EventBody{
ID: "id-1", Title: "title",
Kind: item.KindEvent, Start: time.Now(),
Updated: time.Now(), End: time.Now().Add(-5 * time.Second),
Body: "body", },
} }
if err := c.Update([]item.Item{i}); err != nil { if err := repo.Delete(one.ID); err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
} }
items, err = c.Updated([]item.Kind{item.KindEvent}, time.Time{}) all, err := repo.FindAll()
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
os.Exit(1) os.Exit(1)
} }
fmt.Printf("%+v\n", items) fmt.Printf("all: %+v\n", all)
// c := client.NewClient("http://localhost:8092", "testKey")
// items, err := c.Updated([]item.Kind{item.KindEvent}, time.Time{})
// if err != nil {
// fmt.Println(err)
// os.Exit(1)
// }
// fmt.Printf("%+v\n", items)
// i := item.Item{
// ID: "id-1",
// Kind: item.KindEvent,
// Updated: time.Now(),
// Body: "body",
// }
// if err := c.Update([]item.Item{i}); err != nil {
// fmt.Println(err)
// os.Exit(1)
// }
// items, err = c.Updated([]item.Kind{item.KindEvent}, time.Time{})
// if err != nil {
// fmt.Println(err)
// os.Exit(1)
// }
// fmt.Printf("%+v\n", items)
} }

194
cal/sqlite.go Normal file
View File

@ -0,0 +1,194 @@
package main
import (
"database/sql"
"errors"
"fmt"
"go-mod.ewintr.nl/planner/item"
_ "modernc.org/sqlite"
)
const (
timestampFormat = "2006-01-02 15:04:05"
)
var migrations = []string{
`CREATE TABLE events ("id" TEXT UNIQUE, "title" TEXT, "start" TIMESTAMP, "end" TIMESTAMP)`,
`PRAGMA journal_mode=WAL`,
`PRAGMA synchronous=NORMAL`,
`PRAGMA cache_size=2000`,
}
var (
ErrInvalidConfiguration = errors.New("invalid configuration")
ErrIncompatibleSQLMigration = errors.New("incompatible migration")
ErrNotEnoughSQLMigrations = errors.New("already more migrations than wanted")
ErrSqliteFailure = errors.New("sqlite returned an error")
)
type Sqlite struct {
db *sql.DB
}
func NewSqlite(dbPath string) (*Sqlite, error) {
db, err := sql.Open("sqlite", dbPath)
if err != nil {
return &Sqlite{}, fmt.Errorf("%w: %v", ErrInvalidConfiguration, err)
}
s := &Sqlite{
db: db,
}
if err := s.migrate(migrations); err != nil {
return &Sqlite{}, err
}
return s, nil
}
func (s *Sqlite) Store(event item.Event) error {
if _, err := s.db.Exec(`
INSERT INTO events
(id, title, start, end)
VALUES
(?, ?, ?, ?)
ON CONFLICT(id) DO UPDATE
SET
title=?,
start=?,
end=?`,
event.ID, event.Title, event.Start.Format(timestampFormat), event.End.Format(timestampFormat),
event.Title, event.Start.Format(timestampFormat), event.End.Format(timestampFormat)); err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
return nil
}
func (s *Sqlite) Find(id string) (item.Event, error) {
var event item.Event
err := s.db.QueryRow(`
SELECT id, title, start, end
FROM events
WHERE id = ?`, id).Scan(&event.ID, &event.Title, &event.Start, &event.End)
if err == sql.ErrNoRows {
return event, fmt.Errorf("event not found: %w", err)
}
if err != nil {
return event, fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
return event, nil
}
func (s *Sqlite) FindAll() ([]item.Event, error) {
rows, err := s.db.Query(`
SELECT id, title, start, end
FROM events`)
if err != nil {
return nil, fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
result := make([]item.Event, 0)
defer rows.Close()
for rows.Next() {
var event item.Event
if err := rows.Scan(&event.ID, &event.Title, &event.Start, &event.End); err != nil {
return nil, fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
result = append(result, event)
}
return result, nil
}
func (s *Sqlite) Delete(id string) error {
result, err := s.db.Exec(`
DELETE FROM events
WHERE id = ?`, id)
if err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
rowsAffected, err := result.RowsAffected()
if err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
if rowsAffected == 0 {
return fmt.Errorf("event not found: %s", id)
}
return nil
}
func (s *Sqlite) migrate(wanted []string) error {
// admin table
if _, err := s.db.Exec(`
CREATE TABLE IF NOT EXISTS migration
("id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, "query" TEXT)
`); err != nil {
return err
}
// find existing
rows, err := s.db.Query(`SELECT query FROM migration ORDER BY id`)
if err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
existing := []string{}
for rows.Next() {
var query string
if err := rows.Scan(&query); err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
existing = append(existing, string(query))
}
rows.Close()
// compare
missing, err := compareMigrations(wanted, existing)
if err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
// execute missing
for _, query := range missing {
if _, err := s.db.Exec(string(query)); err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
// register
if _, err := s.db.Exec(`
INSERT INTO migration
(query) VALUES (?)
`, query); err != nil {
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
}
}
return nil
}
func compareMigrations(wanted, existing []string) ([]string, error) {
needed := []string{}
if len(wanted) < len(existing) {
return []string{}, ErrNotEnoughSQLMigrations
}
for i, want := range wanted {
switch {
case i >= len(existing):
needed = append(needed, want)
case want == existing[i]:
// do nothing
case want != existing[i]:
return []string{}, fmt.Errorf("%w: %v", ErrIncompatibleSQLMigration, want)
}
}
return needed, nil
}