mirror of
https://code.dumpstack.io/tools/wi.git
synced 2024-11-14 14:29:19 +02:00
125 lines
2.2 KiB
Go
125 lines
2.2 KiB
Go
/**
|
|
* @file storage.go
|
|
* @author Mikhail Klementyev jollheef<AT>riseup.net
|
|
* @license GNU GPLv3
|
|
* @date July, 2016
|
|
* @brief Database functions
|
|
*/
|
|
|
|
package storage
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func OpenDB(path string) (db *sql.DB, err error) {
|
|
db, err = sql.Open("sqlite3", path)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
_, err = db.Exec("CREATE TABLE IF NOT EXISTS `links` " +
|
|
"( `id` INTEGER PRIMARY KEY AUTOINCREMENT, `url` TEXT );")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
_, err = db.Exec("CREATE TABLE IF NOT EXISTS `history` " +
|
|
"( `id` INTEGER PRIMARY KEY AUTOINCREMENT, `url` TEXT );")
|
|
|
|
return
|
|
}
|
|
|
|
func AddLink(db *sql.DB, url string) (linkNo int64, err error) {
|
|
stmt, err := db.Prepare("INSERT INTO `links` (`url`) VALUES ($1);")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
r, err := stmt.Exec(url)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
linkNo, err = r.LastInsertId()
|
|
|
|
return
|
|
}
|
|
|
|
func GetLink(db *sql.DB, linkID int64) (url string, err error) {
|
|
stmt, err := db.Prepare("SELECT `url` FROM `links` WHERE id=$1;")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
err = stmt.QueryRow(linkID).Scan(&url)
|
|
|
|
return
|
|
}
|
|
|
|
func GetLinkID(db *sql.DB, url string) (linkID int64, err error) {
|
|
stmt, err := db.Prepare("SELECT `id` FROM `links` WHERE url=$1;")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
err = stmt.QueryRow(url).Scan(&linkID)
|
|
|
|
return
|
|
}
|
|
|
|
func AddHistoryURL(db *sql.DB, url string) (err error) {
|
|
stmt, err := db.Prepare("INSERT INTO `history` (`url`) VALUES ($1);")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
_, err = stmt.Exec(url)
|
|
|
|
return
|
|
}
|
|
|
|
type HistoryItem struct {
|
|
ID int64
|
|
URL string
|
|
}
|
|
|
|
func GetHistory(db *sql.DB) (history []HistoryItem, err error) {
|
|
rows, err := db.Query("SELECT `id`, `url` FROM `history`;")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var h HistoryItem
|
|
|
|
err = rows.Scan(&h.ID, &h.URL)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
history = append(history, h)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func GetHistoryUrl(db *sql.DB, historyID int64) (url string, err error) {
|
|
stmt, err := db.Prepare("SELECT `url` FROM `history` WHERE id=$1;")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer stmt.Close()
|
|
|
|
err = stmt.QueryRow(historyID).Scan(&url)
|
|
|
|
return
|
|
}
|