78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
package sqlite
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"go-mod.ewintr.nl/planner/plan/storage"
|
|
)
|
|
|
|
type LocalID struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func (l *LocalID) FindAll() (map[string]int, error) {
|
|
rows, err := l.db.Query(`
|
|
SELECT id, local_id
|
|
FROM localids
|
|
`)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("%w: %v", ErrSqliteFailure, err)
|
|
}
|
|
result := make(map[string]int)
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var id string
|
|
var localID int
|
|
if err := rows.Scan(&id, &localID); err != nil {
|
|
return nil, fmt.Errorf("%w: %v", ErrSqliteFailure, err)
|
|
}
|
|
result[id] = localID
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
func (l *LocalID) Next() (int, error) {
|
|
idMap, err := l.FindAll()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
cur := make([]int, 0, len(idMap))
|
|
for _, localID := range idMap {
|
|
cur = append(cur, localID)
|
|
}
|
|
|
|
return storage.NextLocalID(cur), nil
|
|
}
|
|
|
|
func (l *LocalID) Store(id string, localID int) error {
|
|
if _, err := l.db.Exec(`
|
|
INSERT INTO localids
|
|
(id, local_id)
|
|
VALUES
|
|
(? ,?)`, id, localID); err != nil {
|
|
return fmt.Errorf("%w: %v", ErrSqliteFailure, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *LocalID) Delete(id string) error {
|
|
result, err := l.db.Exec(`
|
|
DELETE FROM localids
|
|
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 storage.ErrNotFound
|
|
}
|
|
return nil
|
|
}
|