package sqlite import ( "database/sql" _ "embed" "fmt" "justguestbook/internal/guestbook" _ "github.com/mattn/go-sqlite3" ) var ( //go:embed queries/schema.sql queryCreateDatabase string //go:embed queries/entryGetAll.sql queryGetEntries string //go:embed queries/entryNew.sql queryNewEntry string //go:embed queries/entryDelete.sql queryDeleteEntry string //go:embed queries/replyNew.sql queryNewReply string //go:embed queries/replyDelete.sql queryDeleteReply string ) var ( stmtGetEntries *sql.Stmt stmtNewEntry *sql.Stmt stmtDeleteEntry *sql.Stmt stmtNewReply *sql.Stmt stmtDeleteReply *sql.Stmt ) func initDBStatements(db *sql.DB) error { _, err := db.Exec(queryCreateDatabase) if err != nil { return err } stmtGetEntries, err = db.Prepare(queryGetEntries) if err != nil { return err } stmtNewEntry, err = db.Prepare(queryNewEntry) if err != nil { return err } stmtDeleteEntry, err = db.Prepare(queryDeleteEntry) if err != nil { return err } stmtNewReply, err = db.Prepare(queryNewReply) if err != nil { return err } stmtDeleteReply, err = db.Prepare(queryDeleteReply) if err != nil { return err } return nil } type SQLiteDatabase struct { db *sql.DB } func New(filePath string) (*SQLiteDatabase, error) { db, err := sql.Open("sqlite3", dsn(filePath)) if err != nil { return nil, err } if err := initDBStatements(db); err != nil { return nil, err } return &SQLiteDatabase{db: db}, nil } func (d *SQLiteDatabase) Entries(page, pageSize int64) (entries []*guestbook.Entry, err error) { tx, err := d.db.Begin() if err != nil { return } defer tx.Rollback() rows, err := tx.Stmt(stmtGetEntries).Query(pageSize, (page-1)*pageSize) if err != nil { return } defer rows.Close() for rows.Next() { var entry guestbook.Entry var reply_created sql.NullString var reply_message sql.NullString if err = rows.Scan( &entry.ID, &entry.Created, &entry.Name, &entry.Website, &entry.Message, &reply_created, &reply_message); err != nil { return } if reply_message.Valid /* reply_created is also valid if reply is */ { entry.Reply = &guestbook.Reply{ Created: reply_created.String, Message: reply_message.String} } entries = append(entries, &entry) } tx.Commit() return } func (d *SQLiteDatabase) NewEntry(entry *guestbook.Entry) error { tx, err := d.db.Begin() if err != nil { return err } defer tx.Rollback() _, err = tx.Stmt(stmtNewEntry).Exec(entry.Created, entry.Name, entry.Message, entry.Website, entry.HideWebsite) if err != nil { return err } tx.Commit() return nil } func (d *SQLiteDatabase) DeleteEntry(entryID int64) error { return nil } func (d *SQLiteDatabase) NewReply(reply *guestbook.Reply) (err error) { tx, err := d.db.Begin() if err != nil { return } defer tx.Rollback() _, err = tx.Stmt(stmtNewReply).Exec(reply.Created, reply.ID, reply.Message) if err != nil { return } tx.Commit() return } func (d *SQLiteDatabase) DeleteReply(entryID int64) error { return nil } func (d *SQLiteDatabase) Close() error { return d.db.Close() } func dsn(filePath string) string { return fmt.Sprintf("file:%s?_journal=WAL&_mutex=full", filePath) }