aether/gotes
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Validation is cheap so do it more

Browse Source
This commit is contained in:
DelphicOkami
2026-03-19 00:49:35 +00:00
parent de6c9d6ae5
commit 4e5b535f09
3 changed files with 47 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
internal/store/sqlite/sqlite.go
View File

@@ -85,10 +85,14 @@ func (s *SQLiteStore) SaveNote(ctx context.Context, note models.Note) (models.No
}
defer tx.Rollback()
if err := note.Validate(); err != nil {
return models.Note{}, err
}
if note.ID == 0 {
result, err := tx.ExecContext(ctx, `
INSERT INTO notes (content, last_update) VALUES (?, ?);
`, note.Content, note.LastUpdate)
INSERT INTO notes (title, content, last_update) VALUES (?, ?, ?);
`, note.Title, note.Content, note.LastUpdate)
if err != nil {
return models.Note{}, fmt.Errorf("failed to insert note: %w", err)
}
@@ -99,14 +103,14 @@ func (s *SQLiteStore) SaveNote(ctx context.Context, note models.Note) (models.No
note.ID = int(id)
} else {
_, err := s.scanNote(tx.QueryRowContext(ctx, `
SELECT id, content, last_update FROM notes WHERE id = ?;
SELECT id, title, content, last_update FROM notes WHERE id = ?;
`, note.ID))
if err != nil {
return models.Note{}, fmt.Errorf("cannot update note not found: %w", err)
}
_, err = tx.ExecContext(ctx, `
UPDATE notes SET content = ?, last_update = ? WHERE id = ?;
`, note.Content, note.LastUpdate, note.ID)
UPDATE notes SET title = ?, content = ?, last_update = ? WHERE id = ?;
`, note.Title, note.Content, note.LastUpdate, note.ID)
if err != nil {
return models.Note{}, fmt.Errorf("failed to update note: %w", err)
}
@@ -120,7 +124,7 @@ func (s *SQLiteStore) SaveNote(ctx context.Context, note models.Note) (models.No
func (s *SQLiteStore) GetNoteByID(ctx context.Context, id int) (models.Note, error) {
row := s.read.QueryRowContext(ctx, `
SELECT id, content, last_update FROM notes WHERE id = ?;
SELECT id, title, content, last_update FROM notes WHERE id = ?;
`, id)
note, err := s.scanNote(row)
if err != nil {
@@ -134,7 +138,7 @@ func (s *SQLiteStore) GetNoteByID(ctx context.Context, id int) (models.Note, err
func (s *SQLiteStore) GetAllNotes(ctx context.Context) ([]models.Note, error) {
rows, err := s.read.QueryContext(ctx, `
SELECT id, content, last_update FROM notes ORDER BY id ASC;
SELECT id, title, content, last_update FROM notes ORDER BY id ASC;
`)
if err != nil {
return nil, fmt.Errorf("failed to query all notes: %w", err)
@@ -177,8 +181,11 @@ func (s *SQLiteStore) validateSchema(ctx context.Context) error {
CREATE TABLE IF NOT EXISTS notes (
id INTEGER PRIMARY KEY AUTOINCREMENT,
last_update TIMESTAMP NOT NULL,
content TEXT NOT NULL
title TEXT NOT NULL,
content BLOB NOT NULL
);
CREATE UNIQUE INDEX idx_id ON notes(id);
CREATE INDEX idx_title ON notes(title);
`, nil)
return err
}
@@ -193,7 +200,7 @@ func (s *SQLiteStore) getWriteTransaction(ctx context.Context) (*sql.Tx, error)
func (s *SQLiteStore) scanNote(results scannable) (models.Note, error) {
var note models.Note
err := results.Scan(&note.ID, &note.Content, &note.LastUpdate)
err := results.Scan(&note.ID, &note.Title, &note.Content, &note.LastUpdate)
if err != nil {
return models.Note{}, fmt.Errorf("failed to scan note: %w", err)
}

16
internal/store/sqlite/sqlite_test.go
View File

@@ -86,11 +86,13 @@ func TestSaveNote(t *testing.T) {
name: "new note",
note: models.Note{
Content: "Test note",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedNote: models.Note{
ID: 1,
Content: "Test note",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedError: false,
@@ -98,11 +100,13 @@ func TestSaveNote(t *testing.T) {
name: "new 2 note",
note: models.Note{
Content: "Test note 2!",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedNote: models.Note{
ID: 2,
Content: "Test note 2!",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedError: false,
@@ -112,11 +116,13 @@ func TestSaveNote(t *testing.T) {
note: models.Note{
ID: 1,
Content: "Updated note",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedNote: models.Note{
ID: 1,
Content: "Updated note",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedError: false,
@@ -126,6 +132,7 @@ func TestSaveNote(t *testing.T) {
note: models.Note{
ID: 999,
Content: "This note does not exist",
Title: "No Problem",
LastUpdate: time.Now(),
},
expectedNote: models.Note{},
@@ -168,6 +175,7 @@ func TestGetNoteByID(t *testing.T) {
}()
store.SaveNote(context.Background(), models.Note{
Content: "Test note",
Title: "Saved No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
})
@@ -183,6 +191,7 @@ func TestGetNoteByID(t *testing.T) {
expectedNote: models.Note{
ID: 1,
Content: "Test note",
Title: "Saved No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
},
expectedError: false,
@@ -259,10 +268,12 @@ func TestGetAllNotes(t *testing.T) {
store.SaveNote(context.Background(), models.Note{
Content: "Test note 1",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
})
store.SaveNote(context.Background(), models.Note{
Content: "Test note 2",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 2, 0, 0, 0, 0, time.UTC),
})
@@ -277,11 +288,13 @@ func TestGetAllNotes(t *testing.T) {
{
ID: 1,
Content: "Test note 1",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
},
{
ID: 2,
Content: "Test note 2",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 2, 0, 0, 0, 0, time.UTC),
},
},
@@ -324,14 +337,17 @@ func TestDeleteNoteByID(t *testing.T) {
store.SaveNote(context.Background(), models.Note{
Content: "Test note 1",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
})
store.SaveNote(context.Background(), models.Note{
Content: "Test note 2",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
})
store.SaveNote(context.Background(), models.Note{
Content: "Test note 3",
Title: "No Problem",
LastUpdate: time.Date(2000, 1, 1, 0, 0, 0, 0, time.UTC),
})