Compare commits
10 Commits
Author | SHA1 | Date |
---|---|---|
Alexander Andreev | 8c8c1f2617 | |
Alexander Andreev | 16e5ececf5 | |
Alexander Andreev | 141e1215a3 | |
Alexander Andreev | 5d56b56882 | |
Alexander Andreev | d272dada83 | |
Alexander Andreev | 84ff66b4f4 | |
Alexander Andreev | 9a93ad9a3a | |
Alexander Andreev | fe47f60581 | |
Alexander Andreev | 5a72fbecb4 | |
Alexander Andreev | cb623e8087 |
2
LICENSE
2
LICENSE
|
@ -1,6 +1,6 @@
|
|||
The MIT License
|
||||
|
||||
Copyright (c) 2022,2023 Alexander "Arav" Andreev <me@arav.top>
|
||||
Copyright (c) 2022,2023 Alexander "Arav" Andreev <me@arav.su>
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
|
|
|
@ -1,4 +0,0 @@
|
|||
justguestbook ver. 1.2.0
|
||||
========================
|
||||
|
||||
A library implementing simple guestbook with replies.
|
|
@ -1,305 +0,0 @@
|
|||
package sqlite
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
_ "embed"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"git.arav.su/Arav/justguestbook/guestbook"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
var (
|
||||
//go:embed queries/schema.sql
|
||||
querySchema string
|
||||
|
||||
//go:embed queries/entryGetAll.sql
|
||||
queryGetAll string
|
||||
//go:embed queries/entryCount.sql
|
||||
queryCount string
|
||||
|
||||
//go:embed queries/entryNew.sql
|
||||
queryNewEntry string
|
||||
//go:embed queries/entryUpdate.sql
|
||||
queryUpdateEntry string
|
||||
//go:embed queries/entryDelete.sql
|
||||
queryDeleteEntry string
|
||||
//go:embed queries/replyNew.sql
|
||||
queryNewReply string
|
||||
//go:embed queries/replyUpdate.sql
|
||||
queryUpdateReply string
|
||||
//go:embed queries/replyDelete.sql
|
||||
queryDeleteReply string
|
||||
)
|
||||
|
||||
var (
|
||||
stmtGetAll *sql.Stmt
|
||||
stmtCount *sql.Stmt
|
||||
stmtNewEntry *sql.Stmt
|
||||
stmtUpdateEntry *sql.Stmt
|
||||
stmtDeleteEntry *sql.Stmt
|
||||
stmtNewReply *sql.Stmt
|
||||
stmtUpdateReply *sql.Stmt
|
||||
stmtDeleteReply *sql.Stmt
|
||||
)
|
||||
|
||||
func initDBStatements(db *sql.DB) error {
|
||||
db.Exec("PRAGMA foreign_keys = ON;")
|
||||
|
||||
_, err := db.Exec(querySchema)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to init schema")
|
||||
}
|
||||
|
||||
stmtGetAll, err = db.Prepare(queryGetAll)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryGetAll")
|
||||
}
|
||||
|
||||
stmtCount, err = db.Prepare(queryCount)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryCount")
|
||||
}
|
||||
|
||||
stmtNewEntry, err = db.Prepare(queryNewEntry)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryNewEntry")
|
||||
}
|
||||
|
||||
stmtUpdateEntry, err = db.Prepare(queryUpdateEntry)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryUpdateEntry")
|
||||
}
|
||||
|
||||
stmtDeleteEntry, err = db.Prepare(queryDeleteEntry)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryDeleteEntry")
|
||||
}
|
||||
|
||||
stmtNewReply, err = db.Prepare(queryNewReply)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryNewReply")
|
||||
}
|
||||
|
||||
stmtUpdateReply, err = db.Prepare(queryUpdateReply)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryUpdateReply")
|
||||
}
|
||||
|
||||
stmtDeleteReply, err = db.Prepare(queryDeleteReply)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare queryDeleteReply")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
type SQLiteDatabase struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
func New(filePath string) (guestbook.Guestbook, 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(stmtGetAll).Query(pageSize, (page-1)*pageSize)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
for rows.Next() {
|
||||
var entry guestbook.Entry
|
||||
var entry_created int64
|
||||
var reply_created int64
|
||||
var reply_message string
|
||||
if err = rows.Scan(
|
||||
&entry.ID, &entry_created, &entry.Name,
|
||||
&entry.Message, &entry.Website, &entry.HideWebsite,
|
||||
&reply_created, &reply_message); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
entry.Created = time.Unix(entry_created, 0)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if reply_message != "" {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
entry.Reply = &guestbook.Reply{
|
||||
ID: entry.ID,
|
||||
// Created: date,
|
||||
Created: time.Unix(reply_created, 0),
|
||||
Message: reply_message}
|
||||
}
|
||||
|
||||
entries = append(entries, &entry)
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// Count returns how much entries are in an `entry` table.
|
||||
func (d *SQLiteDatabase) Count() (count int64, err error) {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
err = tx.Stmt(stmtCount).QueryRow().Scan(&count)
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return count, nil
|
||||
}
|
||||
|
||||
// NewEntry inserts a passed Entry struct and fills its ID field if successful.
|
||||
func (d *SQLiteDatabase) NewEntry(entry *guestbook.Entry) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
r, err := tx.Stmt(stmtNewEntry).Exec(entry.Created.UTC().Unix(), entry.Name, entry.Message,
|
||||
entry.Website, entry.HideWebsite)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
entry.ID, err = r.LastInsertId()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) EditEntry(entry *guestbook.Entry) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
row := tx.Stmt(stmtUpdateEntry).QueryRow(entry.Name, entry.Message, entry.Website, entry.HideWebsite, entry.ID)
|
||||
if row.Err() != nil {
|
||||
return row.Err()
|
||||
}
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) DeleteEntry(entryID int64) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
if _, err = tx.Stmt(stmtDeleteEntry).Exec(entryID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) NewReply(reply *guestbook.Reply) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
_, err = tx.Stmt(stmtNewReply).Exec(reply.ID, reply.Created.UTC().Unix(), reply.Message)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) EditReply(reply *guestbook.Reply) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
row := tx.Stmt(stmtUpdateReply).QueryRow(reply.Message, reply.ID)
|
||||
if row.Err() != nil {
|
||||
return row.Err()
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) DeleteReply(entryID int64) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
if _, err = tx.Stmt(stmtDeleteReply).Exec(entryID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) Close() error {
|
||||
stmtCount.Close()
|
||||
stmtDeleteEntry.Close()
|
||||
stmtDeleteReply.Close()
|
||||
stmtGetAll.Close()
|
||||
stmtNewEntry.Close()
|
||||
stmtNewReply.Close()
|
||||
stmtUpdateEntry.Close()
|
||||
stmtUpdateReply.Close()
|
||||
|
||||
return d.db.Close()
|
||||
}
|
||||
|
||||
func dsn(filePath string) string {
|
||||
return fmt.Sprintf("file:%s?_journal=WAL&_mutex=full", filePath)
|
||||
}
|
|
@ -1,4 +0,0 @@
|
|||
INSERT INTO `entry`
|
||||
(`created`, `name`, `message`, `website`, `hide_website`)
|
||||
VALUES
|
||||
(?, ?, ?, ?, ?);
|
|
@ -1,7 +0,0 @@
|
|||
UPDATE OR REPLACE `entry`
|
||||
SET
|
||||
`name` = ?,
|
||||
`message` = ?,
|
||||
`website` = ?,
|
||||
`hide_website` = ?
|
||||
WHERE `entry_id` = ?;
|
|
@ -1,4 +0,0 @@
|
|||
UPDATE OR REPLACE `reply`
|
||||
SET
|
||||
`message` = ?
|
||||
WHERE `entry_id` = ?;
|
|
@ -1,22 +0,0 @@
|
|||
-- SQLite3
|
||||
CREATE TABLE IF NOT EXISTS `entry` (
|
||||
`entry_id` INTEGER NOT NULL,
|
||||
`created` INTEGER NOT NULL,
|
||||
`name` TEXT NOT NULL,
|
||||
`message` TEXT NOT NULL,
|
||||
`website` TEXT NOT NULL,
|
||||
`hide_website` INTEGER NOT NULL DEFAULT TRUE,
|
||||
PRIMARY KEY (`entry_id`) );
|
||||
|
||||
CREATE INDEX IF NOT EXISTS `entry_created_idx`
|
||||
ON `entry` (`created`);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS `reply` (
|
||||
`entry_id` INTEGER NOT NULL,
|
||||
`created` INTEGER NOT NULL,
|
||||
`message` TEXT NOT NULL,
|
||||
PRIMARY KEY (`entry_id`),
|
||||
FOREIGN KEY (`entry_id`)
|
||||
REFERENCES `entry` (`entry_id`)
|
||||
ON DELETE CASCADE
|
||||
ON UPDATE CASCADE );
|
|
@ -0,0 +1,307 @@
|
|||
package justguestbook
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
_ "embed"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
var (
|
||||
//go:embed sqlite_queries/schema.sql
|
||||
sqlQuerySchema string
|
||||
|
||||
//go:embed sqlite_queries/entryGetAll.sql
|
||||
sqlQueryGetAll string
|
||||
//go:embed sqlite_queries/entryCount.sql
|
||||
sqlQueryCount string
|
||||
|
||||
//go:embed sqlite_queries/entryNew.sql
|
||||
sqlQueryNewEntry string
|
||||
//go:embed sqlite_queries/entryUpdate.sql
|
||||
sqlQueryUpdateEntry string
|
||||
//go:embed sqlite_queries/entryDelete.sql
|
||||
sqlQueryDeleteEntry string
|
||||
//go:embed sqlite_queries/replyNew.sql
|
||||
sqlQueryNewReply string
|
||||
//go:embed sqlite_queries/replyUpdate.sql
|
||||
sqlQueryUpdateReply string
|
||||
//go:embed sqlite_queries/replyDelete.sql
|
||||
sqlQueryDeleteReply string
|
||||
)
|
||||
|
||||
var (
|
||||
sqlStmtGetAll *sql.Stmt
|
||||
sqlStmtCount *sql.Stmt
|
||||
sqlStmtNewEntry *sql.Stmt
|
||||
sqlStmtUpdateEntry *sql.Stmt
|
||||
sqlStmtDeleteEntry *sql.Stmt
|
||||
sqlStmtNewReply *sql.Stmt
|
||||
sqlStmtUpdateReply *sql.Stmt
|
||||
sqlStmtDeleteReply *sql.Stmt
|
||||
)
|
||||
|
||||
func initSQLiteStatements(db *sql.DB) error {
|
||||
db.Exec("PRAGMA foreign_keys = ON;")
|
||||
|
||||
_, err := db.Exec(sqlQuerySchema)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to init schema")
|
||||
}
|
||||
|
||||
sqlStmtGetAll, err = db.Prepare(sqlQueryGetAll)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryGetAll")
|
||||
}
|
||||
|
||||
sqlStmtCount, err = db.Prepare(sqlQueryCount)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryCount")
|
||||
}
|
||||
|
||||
sqlStmtNewEntry, err = db.Prepare(sqlQueryNewEntry)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryNewEntry")
|
||||
}
|
||||
|
||||
sqlStmtUpdateEntry, err = db.Prepare(sqlQueryUpdateEntry)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryUpdateEntry")
|
||||
}
|
||||
|
||||
sqlStmtDeleteEntry, err = db.Prepare(sqlQueryDeleteEntry)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryDeleteEntry")
|
||||
}
|
||||
|
||||
sqlStmtNewReply, err = db.Prepare(sqlQueryNewReply)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryNewReply")
|
||||
}
|
||||
|
||||
sqlStmtUpdateReply, err = db.Prepare(sqlQueryUpdateReply)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryUpdateReply")
|
||||
}
|
||||
|
||||
sqlStmtDeleteReply, err = db.Prepare(sqlQueryDeleteReply)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "failed to prepare sqlQueryDeleteReply")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// SQLiteDatabase implements a Guestbook that works with a SQLite DB
|
||||
// under the hood.
|
||||
type SQLiteDatabase struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
// NewSQLiteDB returns an instance of a SQLite Guestbook implementation.
|
||||
func NewSQLiteDB(filePath string) (Guestbook, error) {
|
||||
db, err := sql.Open("sqlite3", sqliteDSN(filePath))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err := initSQLiteStatements(db); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &SQLiteDatabase{db: db}, nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) Entries(page, pageSize int64) (entries []*Entry, err error) {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
rows, err := tx.Stmt(sqlStmtGetAll).Query(pageSize, (page-1)*pageSize)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
for rows.Next() {
|
||||
var entry Entry
|
||||
var entry_created int64
|
||||
var reply_created int64
|
||||
var reply_message string
|
||||
if err = rows.Scan(
|
||||
&entry.ID, &entry_created, &entry.Name,
|
||||
&entry.Message, &entry.Website, &entry.HideWebsite,
|
||||
&reply_created, &reply_message); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
entry.Created = time.Unix(entry_created, 0)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if reply_message != "" {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
entry.Reply = &Reply{
|
||||
ID: entry.ID,
|
||||
// Created: date,
|
||||
Created: time.Unix(reply_created, 0),
|
||||
Message: reply_message}
|
||||
}
|
||||
|
||||
entries = append(entries, &entry)
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// Count returns how much entries are in an `entry` table.
|
||||
func (d *SQLiteDatabase) Count() (count int64, err error) {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
err = tx.Stmt(sqlStmtCount).QueryRow().Scan(&count)
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return count, nil
|
||||
}
|
||||
|
||||
// NewEntry inserts a passed Entry struct and fills its ID field if successful.
|
||||
func (d *SQLiteDatabase) NewEntry(entry *Entry) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
r, err := tx.Stmt(sqlStmtNewEntry).Exec(entry.Created.UTC().Unix(), entry.Name, entry.Message,
|
||||
entry.Website, entry.HideWebsite)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
entry.ID, err = r.LastInsertId()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) EditEntry(entry *Entry) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
_, err = tx.Stmt(sqlStmtUpdateEntry).Exec(entry.Name, entry.Message, entry.Website, entry.HideWebsite, entry.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) DeleteEntry(entryID int64) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
if _, err = tx.Stmt(sqlStmtDeleteEntry).Exec(entryID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) NewReply(reply *Reply) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
_, err = tx.Stmt(sqlStmtNewReply).Exec(reply.ID, reply.Created.UTC().Unix(), reply.Message)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) EditReply(reply *Reply) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
_, err = tx.Stmt(sqlStmtUpdateReply).Exec(reply.Message, reply.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) DeleteReply(entryID int64) error {
|
||||
tx, err := d.db.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer tx.Rollback()
|
||||
|
||||
if _, err = tx.Stmt(sqlStmtDeleteReply).Exec(entryID); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
tx.Commit()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (d *SQLiteDatabase) Close() error {
|
||||
sqlStmtCount.Close()
|
||||
sqlStmtDeleteEntry.Close()
|
||||
sqlStmtDeleteReply.Close()
|
||||
sqlStmtGetAll.Close()
|
||||
sqlStmtNewEntry.Close()
|
||||
sqlStmtNewReply.Close()
|
||||
sqlStmtUpdateEntry.Close()
|
||||
sqlStmtUpdateReply.Close()
|
||||
|
||||
return d.db.Close()
|
||||
}
|
||||
|
||||
func sqliteDSN(filePath string) string {
|
||||
return fmt.Sprintf("file:%s?_journal=WAL&_mutex=full", filePath)
|
||||
}
|
|
@ -1,10 +1,9 @@
|
|||
package sqlite_test
|
||||
package justguestbook_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"git.arav.su/Arav/justguestbook/database/sqlite"
|
||||
"git.arav.su/Arav/justguestbook/guestbook"
|
||||
guestbook "git.arav.su/Arav/justguestbook"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
|
@ -14,7 +13,7 @@ const (
|
|||
)
|
||||
|
||||
func genTestDB() (db guestbook.Guestbook, err error) {
|
||||
db, err = sqlite.New(":memory:")
|
||||
db, err = guestbook.NewSQLiteDB(":memory:")
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "failed to init DB")
|
||||
}
|
||||
|
@ -83,13 +82,17 @@ func TestSqliteEditEntry(t *testing.T) {
|
|||
|
||||
me := es0[0]
|
||||
|
||||
me.ID = es0[0].ID
|
||||
me.Name = "NotSoAnonymous"
|
||||
|
||||
err = db.EditEntry(me)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
es0, err = db.Entries(1, 30)
|
||||
if es0[0].Name != me.Name {
|
||||
t.Error("name wasn't changed", es0[0].Name, "!=", me.Name, es0[0].ID, me.ID)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSqliteEditReply(t *testing.T) {
|
|
@ -0,0 +1,108 @@
|
|||
package justguestbook
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
// DateFormat is a format of the date and time used in a guestbook.
|
||||
DateFormat = "2006-01-02 15:04:05"
|
||||
|
||||
// entryWebsiteMaxLength is a maximum length for a Website field of an Entry.
|
||||
entryWebsiteMaxLength = 255
|
||||
)
|
||||
|
||||
// Reply holds a reply for a guestbook's entry.
|
||||
type Reply struct {
|
||||
// ID of a guestbook's entry is used when update of a reply is performed.
|
||||
//
|
||||
// Not present in JSON because a reply comes within its corresponding entry.
|
||||
ID int64 `json:"-"`
|
||||
|
||||
// Created holds a date and time when a reply was created.
|
||||
Created time.Time `json:"created,omitempty"`
|
||||
|
||||
// Message holds a reply's message.
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
// NewReply creates a new reply and, before that, verifies that a passed message
|
||||
// is not empty.
|
||||
func NewReply(entryID int64, message string) (*Reply, error) {
|
||||
if message == "" {
|
||||
return nil, errors.New("empty message field")
|
||||
}
|
||||
|
||||
return &Reply{
|
||||
ID: entryID,
|
||||
Created: time.Now().UTC(),
|
||||
Message: message}, nil
|
||||
}
|
||||
|
||||
// Entry holds a guestbook entry.
|
||||
type Entry struct {
|
||||
// ID of an entry.
|
||||
ID int64 `json:"entry_id"`
|
||||
|
||||
// Created holds the date and time when an entry was created.
|
||||
Created time.Time `json:"created"`
|
||||
|
||||
// Name holds a nick-/name of a guest who posted an entry.
|
||||
Name string `json:"name"`
|
||||
|
||||
// Website of a guest. Can be empty.
|
||||
Website string `json:"website,omitempty"`
|
||||
|
||||
// HideWebsite tells wether to hide or show guest's website.
|
||||
HideWebsite bool `json:"hide_website,omitempty"`
|
||||
|
||||
// Message that a guest wrote.
|
||||
Message string `json:"message"`
|
||||
|
||||
// Reply holds a reply for this entry.
|
||||
Reply *Reply `json:"reply,omitempty"`
|
||||
}
|
||||
|
||||
// NewEntry creates a new Entry and, before that, verifies if the name and message
|
||||
// aren't empty and the website field do not exceed the limit.
|
||||
func NewEntry(name, message, website string, hideWebsite bool) (*Entry, error) {
|
||||
if name == "" || message == "" {
|
||||
return nil, errors.New("name and message field are required")
|
||||
}
|
||||
|
||||
if len(website) > entryWebsiteMaxLength {
|
||||
return nil, errors.New(fmt.Sprint("website field's max length is",
|
||||
entryWebsiteMaxLength, "but", len(website), "was given"))
|
||||
}
|
||||
|
||||
return &Entry{
|
||||
Created: time.Now().UTC(),
|
||||
Name: name,
|
||||
Website: website,
|
||||
HideWebsite: hideWebsite,
|
||||
Message: message}, nil
|
||||
}
|
||||
|
||||
// Guestbook is an interface that should be implemented by a DB.
|
||||
type Guestbook interface {
|
||||
// Entries returns a slice of guestbook entries.
|
||||
Entries(page, pageSize int64) ([]*Entry, error)
|
||||
// Count returns how much entries are in a DB.
|
||||
Count() (int64, error)
|
||||
// NewEntry inserts a given Entry into a DB.
|
||||
NewEntry(entry *Entry) error
|
||||
// EditEntry modifies the fields of an existing entry.
|
||||
EditEntry(entry *Entry) error
|
||||
// DeleteEntry removes an Entry with a given ID.
|
||||
DeleteEntry(entryID int64) error
|
||||
// NewReply inserts a given Reply into a DB.
|
||||
NewReply(reply *Reply) error
|
||||
// EditReply modifies a message field of a given Reply.
|
||||
EditReply(reply *Reply) error
|
||||
// DeleteReply removes a Reply with a given ID of an Entry.
|
||||
DeleteReply(entryID int64) error
|
||||
// Close closes a DB.
|
||||
Close() error
|
||||
}
|
|
@ -1,15 +0,0 @@
|
|||
package guestbook
|
||||
|
||||
const DateFormat = "2006-01-02 15:04:05"
|
||||
|
||||
type Guestbook interface {
|
||||
Entries(page, pageSize int64) ([]*Entry, error)
|
||||
Count() (int64, error)
|
||||
NewEntry(entry *Entry) error
|
||||
EditEntry(entry *Entry) error
|
||||
DeleteEntry(entryID int64) error
|
||||
NewReply(reply *Reply) error
|
||||
EditReply(reply *Reply) error
|
||||
DeleteReply(entryID int64) error
|
||||
Close() error
|
||||
}
|
|
@ -1,29 +0,0 @@
|
|||
package guestbook
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Entry struct {
|
||||
ID int64 `json:"entry_id"`
|
||||
Created time.Time `json:"created"`
|
||||
Name string `json:"name"`
|
||||
Website string `json:"website,omitempty"`
|
||||
HideWebsite bool `json:"hide_website,omitempty"`
|
||||
Message string `json:"message"`
|
||||
Reply *Reply `json:"reply,omitempty"`
|
||||
}
|
||||
|
||||
func NewEntry(name, message, website string, hideWebsite bool) (*Entry, error) {
|
||||
if name == "" || message == "" {
|
||||
return nil, errors.New("name and message field are required")
|
||||
}
|
||||
|
||||
return &Entry{
|
||||
Created: time.Now().UTC(),
|
||||
Name: name,
|
||||
Website: website,
|
||||
HideWebsite: hideWebsite,
|
||||
Message: message}, nil
|
||||
}
|
|
@ -1,23 +0,0 @@
|
|||
package guestbook
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Reply struct {
|
||||
ID int64 `json:"-"`
|
||||
Created time.Time `json:"created,omitempty"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
func NewReply(entryID int64, message string) (*Reply, error) {
|
||||
if message == "" {
|
||||
return nil, errors.New("empty reply field")
|
||||
}
|
||||
|
||||
return &Reply{
|
||||
ID: entryID,
|
||||
Created: time.Now().UTC(),
|
||||
Message: message}, nil
|
||||
}
|
|
@ -0,0 +1,4 @@
|
|||
INSERT INTO `entry`
|
||||
(`created`, `name`, `message`, `website`, `hide_website`)
|
||||
VALUES
|
||||
(?, ?, ?, ?, ?);
|
|
@ -0,0 +1,7 @@
|
|||
UPDATE OR REPLACE `entry`
|
||||
SET
|
||||
`name` = ?,
|
||||
`message` = ?,
|
||||
`website` = ?,
|
||||
`hide_website` = ?
|
||||
WHERE `entry_id` = ?;
|
|
@ -0,0 +1,4 @@
|
|||
UPDATE OR REPLACE `reply`
|
||||
SET
|
||||
`message` = ?
|
||||
WHERE `entry_id` = ?;
|
|
@ -0,0 +1,22 @@
|
|||
-- SQLite3
|
||||
CREATE TABLE IF NOT EXISTS `entry` (
|
||||
`entry_id` INTEGER NOT NULL,
|
||||
`created` INTEGER NOT NULL,
|
||||
`name` TEXT NOT NULL,
|
||||
`message` TEXT NOT NULL,
|
||||
`website` TEXT NOT NULL,
|
||||
`hide_website` INTEGER NOT NULL DEFAULT TRUE,
|
||||
PRIMARY KEY (`entry_id`) );
|
||||
|
||||
CREATE INDEX IF NOT EXISTS `entry_created_idx`
|
||||
ON `entry` (`created`);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS `reply` (
|
||||
`entry_id` INTEGER NOT NULL,
|
||||
`created` INTEGER NOT NULL,
|
||||
`message` TEXT NOT NULL,
|
||||
PRIMARY KEY (`entry_id`),
|
||||
FOREIGN KEY (`entry_id`)
|
||||
REFERENCES `entry` (`entry_id`)
|
||||
ON DELETE CASCADE
|
||||
ON UPDATE CASCADE );
|
Loading…
Reference in New Issue