Kristian
c1feac0812
* feat(persistence): added build-flag and repo funcs * feat(init_handler): note count check to avoid unnecessary rebuilds * ref(build_flag): remove build flag is redundant * build
98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package repositories
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
|
|
"github.com/KristianJBorgwarth/dendrite.daemon/core/models"
|
|
"github.com/KristianJBorgwarth/dendrite.daemon/persistence"
|
|
)
|
|
|
|
type INoteRepository interface {
|
|
Insert(ctx context.Context, dbContext persistence.IDbContext, note *models.Note) error
|
|
InsertRange(ctx context.Context, dbContext persistence.IDbContext, note []*models.Note) error
|
|
Update(ctx context.Context, dbContext persistence.IDbContext, noteID, path, title, slug string) error
|
|
GetBySlug(ctx context.Context, slug string) (*models.Note, error)
|
|
GetNoteCount(ctx context.Context) (int, error)
|
|
}
|
|
|
|
type noteRepository struct {
|
|
readDBContext persistence.ReadContext
|
|
}
|
|
|
|
func NewNoteRepository(rdb persistence.ReadContext) INoteRepository {
|
|
return ¬eRepository{readDBContext: rdb}
|
|
}
|
|
|
|
func (r *noteRepository) Insert(ctx context.Context, dbContext persistence.IDbContext, note *models.Note) error {
|
|
query := `
|
|
INSERT INTO note (id, title, path, slug, created_at, updated_at)
|
|
VALUES (?, ?, ?, ?, datetime('now'), datetime('now'))
|
|
ON CONFLICT (slug) DO UPDATE
|
|
SET title = EXCLUDED.title,
|
|
path = EXCLUDED.path;
|
|
`
|
|
_, err := dbContext.ExecContext(ctx, query, note.ID(), note.Title(), note.Path(), note.Slug())
|
|
return err
|
|
}
|
|
|
|
func (r *noteRepository) InsertRange(ctx context.Context, dbContext persistence.IDbContext, notes []*models.Note) error {
|
|
if len(notes) == 0 {
|
|
return nil
|
|
}
|
|
noteStatement, err := dbContext.Prepare(`
|
|
INSERT INTO note (id, title, path, slug, created_at, updated_at)
|
|
VALUES (?, ?, ?, ?, datetime('now'), datetime('now'))`)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, note := range notes {
|
|
if _, err := noteStatement.ExecContext(ctx, note.ID(), note.Title(), note.Path(), note.Slug()); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *noteRepository) Update(ctx context.Context, dbContext persistence.IDbContext, noteID, path, title, slug string) error {
|
|
query := `
|
|
UPDATE note
|
|
SET path = ?, title = ?, slug = ?, updated_at = datetime('now')
|
|
WHERE id = ?;
|
|
`
|
|
_, err := dbContext.ExecContext(ctx, query, path, title, slug, noteID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *noteRepository) GetBySlug(ctx context.Context, slug string) (*models.Note, error) {
|
|
query := `SELECT id, title, path, slug, created_at, updated_at FROM note WHERE slug = ?`
|
|
row := r.readDBContext.QueryRowContext(ctx, query, slug)
|
|
|
|
var id, title, path, createdAt, updatedAt string
|
|
err := row.Scan(&id, &title, &path, &slug, &createdAt, &updatedAt)
|
|
if err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return models.NewNote(id, path, title, slug, createdAt, updatedAt), nil
|
|
}
|
|
|
|
func (r *noteRepository) GetNoteCount(ctx context.Context) (int, error) {
|
|
query := `SELECT COUNT(*) FROM note`
|
|
row := r.readDBContext.QueryRowContext(ctx, query)
|
|
|
|
var count int
|
|
err := row.Scan(&count)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return count, nil
|
|
}
|